gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: updates the acreate_structured_output

Browse source
This commit is contained in:
hajdul88 2024-12-20 16:19:50 +01:00
parent de2394c392
commit c8a1f04b4c
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cognee/infrastructure/llm/openai/adapter.py
View file

@ -12,7 +12,8 @@ from cognee.infrastructure.llm.llm_interface import LLMInterface
from cognee.infrastructure.llm.prompts import read_query_prompt
from cognee.base_config import get_base_config
if MonitoringTool.LANGFUSE:
monitoring = get_base_config().monitoring_tool
if monitoring == MonitoringTool.LANGFUSE:
from langfuse.decorators import observe
class OpenAIAdapter(LLMInterface):
@ -43,7 +44,7 @@ class OpenAIAdapter(LLMInterface):
base_config = get_base_config()
@observe()
@observe(as_type='generation')
async def acreate_structured_output(self, text_input: str, system_prompt: str,
response_model: Type[BaseModel]) -> BaseModel: