about summary refs log tree commit diff stats
path: root/archive/2025/summer/bsc_gerg/src/knowledge/openai/extract.py
diff options
context:
space:
mode:
Diffstat (limited to 'archive/2025/summer/bsc_gerg/src/knowledge/openai/extract.py')
-rw-r--r--archive/2025/summer/bsc_gerg/src/knowledge/openai/extract.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/archive/2025/summer/bsc_gerg/src/knowledge/openai/extract.py b/archive/2025/summer/bsc_gerg/src/knowledge/openai/extract.py
new file mode 100644
index 000000000..236476acf
--- /dev/null
+++ b/archive/2025/summer/bsc_gerg/src/knowledge/openai/extract.py
@@ -0,0 +1,15 @@
+from src.knowledge.llm.extract import LLMTermExtractor, DEVELOPER_PROMPT, EXAMPLE_USER, OUTPUT_ASSISTANT
+from src.llm import create_completion_openai
+
+
+class OpenAIExtractor(LLMTermExtractor):
+
+    async def get_llm_response(self, text: str) -> str:
+        return await create_completion_openai(
+            messages=[
+                ("developer", f"{DEVELOPER_PROMPT}"),
+                ("user", EXAMPLE_USER),
+                ("assistant", OUTPUT_ASSISTANT),
+                ("user", "Input: \n" + text)
+            ]
+        )
\ No newline at end of file