diff options
Diffstat (limited to 'archive/2025/summer/bsc_gerg/src/knowledge/openai/lemmatize.py')
| -rw-r--r-- | archive/2025/summer/bsc_gerg/src/knowledge/openai/lemmatize.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/archive/2025/summer/bsc_gerg/src/knowledge/openai/lemmatize.py b/archive/2025/summer/bsc_gerg/src/knowledge/openai/lemmatize.py new file mode 100644 index 000000000..98e27e397 --- /dev/null +++ b/archive/2025/summer/bsc_gerg/src/knowledge/openai/lemmatize.py @@ -0,0 +1,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, + ) |