10 lines
299 B
Python
10 lines
299 B
Python
import pytest
|
|
from hellocomputer.models import Prompts
|
|
from langchain.prompts import PromptTemplate
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_general_prompt():
|
|
general: PromptTemplate = await Prompts.general()
|
|
assert general.format(query="whatever").startswith("You're a helpful assistant")
|