From bacea87186dea501e6c8ec653de02e95c5672cfd Mon Sep 17 00:00:00 2001 From: Dmitrii Galkin Date: Sat, 3 May 2025 14:47:40 +0400 Subject: [PATCH] Switch OpenAI to AsyncOpenAI and make text field required in default tools schema --- cognee/api/v1/responses/default_tools.py | 1 + cognee/api/v1/responses/routers/get_responses_router.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/cognee/api/v1/responses/default_tools.py b/cognee/api/v1/responses/default_tools.py index 1ac589c68..0866140bd 100644 --- a/cognee/api/v1/responses/default_tools.py +++ b/cognee/api/v1/responses/default_tools.py @@ -55,6 +55,7 @@ DEFAULT_TOOLS = [ "description": "Path to a custom graph model file", }, }, + "required": ["text"], }, }, # Commented as dangerous diff --git a/cognee/api/v1/responses/routers/get_responses_router.py b/cognee/api/v1/responses/routers/get_responses_router.py index 24ffecb40..c3117a230 100644 --- a/cognee/api/v1/responses/routers/get_responses_router.py +++ b/cognee/api/v1/responses/routers/get_responses_router.py @@ -38,7 +38,7 @@ def get_responses_router() -> APIRouter: Get appropriate client based on model name """ llm_config = get_llm_config() - return openai.OpenAI(api_key=llm_config.llm_api_key) + return openai.AsyncOpenAI(api_key=llm_config.llm_api_key) async def call_openai_api_for_model( input_text: str, @@ -58,7 +58,7 @@ def get_responses_router() -> APIRouter: logger.debug(f"Using model: {model}") - response = client.responses.create( + response = await client.responses.create( model=model, input=input_text, temperature=temperature,