Merge pull request #10 from walamana/main

Add bsc_gerg

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