hellocomputer/test/test_prompts.py

10 lines
300 B
Python
Raw Normal View History

2024-07-17 09:04:08 +02:00
import pytest
2024-07-25 23:42:19 +02:00
from hellocomputer.prompts import Prompts
2024-07-17 09:04:08 +02:00
from langchain.prompts import PromptTemplate
@pytest.mark.asyncio
async def test_get_general_prompt():
2024-07-25 00:10:09 +02:00
general: PromptTemplate = await Prompts.general()
assert general.format(query="whatever").startswith("You're a helpful assistant")