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:
authorJonas Gerg <joniogerg@gmail.com>2025-09-09 20:06:52 +0200
committerJonas Gerg <joniogerg@gmail.com>2025-09-09 20:06:52 +0200
commit3e5d3ca82193e8e8561beb9ceac9982f376d84e2 (patch)
tree76e4c260123b68b93da2417482024ba11f9838ee /archive/2025/summer/bsc_gerg/src/knowledge/openai/lemmatize.py
parenta910d0a3e57f4de47cf2387ac239ae8d0eaca507 (diff)
downloadresearch-work-archive-artifacts-3e5d3ca82193e8e8561beb9ceac9982f376d84e2.tar.gz
research-work-archive-artifacts-3e5d3ca82193e8e8561beb9ceac9982f376d84e2.zip
Add bsc_gerg
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,
+        )