about summary refs log tree commit diff stats
path: root/archive/2025/summer/bsc_gerg/src/knowledge/openai/lemmatize.py
diff options
context:
space:
mode:
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.py17
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,
+        )