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()
|
2024-07-27 22:05:29 +02:00
|
|
|
assert general.format(query="whatever").startswith(
|
|
|
|
|
"You've been asked to do a task you can't do"
|
|
|
|
|
)
|
