diff --git a/openmemory/api/app/utils/categorization.py b/openmemory/api/app/utils/categorization.py index 60ab7602..d3554691 100644 --- a/openmemory/api/app/utils/categorization.py +++ b/openmemory/api/app/utils/categorization.py @@ -24,11 +24,10 @@ def get_categories_for_memory(memory: str) -> List[str]: ] # Let OpenAI handle the pydantic parsing directly - completion = openai_client.chat.completions.with_response_format( - response_format=MemoryCategories - ).create( + completion = openai_client.beta.chat.completions.parse( model="gpt-4o-mini", messages=messages, + response_format=MemoryCategories, temperature=0 ) diff --git a/openmemory/api/requirements.txt b/openmemory/api/requirements.txt index 286643ef..239c59b8 100644 --- a/openmemory/api/requirements.txt +++ b/openmemory/api/requirements.txt @@ -7,6 +7,7 @@ psycopg2-binary>=2.9.0 python-multipart>=0.0.5 fastapi-pagination>=0.12.0 mem0ai>=0.1.92 +openai>=1.40.0 mcp[cli]>=1.3.0 pytest>=7.0.0 pytest-asyncio>=0.21.0