blob: 98e27e3976a47f1aa238643c29d5e1f2f7f90cb1 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
from src.knowledge.llm.lemmatize import LLMTermLemmatizer, DEVELOPER_PROMPT_SHORT, EXAMPLES
from src.llm import create_completion_openai
class OpenAILemmatizer(LLMTermLemmatizer):
async def get_llm_response(self, term: str) -> str:
messages = [
("system", f"{DEVELOPER_PROMPT_SHORT}"),
*EXAMPLES,
# ("user", example_user),
# ("assistant", output_assistant),
("user", term)
]
return await create_completion_openai(
messages=messages,
)
|