diff options
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.py | 15 |
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 |