hellocomputer/test/test_prompts.py
Guillem Borrell f4b9c30a17
Some checks failed
ci/woodpecker/push/woodpecker Pipeline failed
Kind of refactored everything
2024-07-25 00:10:09 +02:00

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")