From 6f3fbd087dc9abcc75d61ed43a54f23dc540b2f6 Mon Sep 17 00:00:00 2001 From: Ryan Rozich Date: Wed, 25 Jun 2025 11:32:26 -0500 Subject: [PATCH] fix: Fix memory categorization by updating dependencies and correcting API usage (#3005) Co-authored-by: Claude Co-authored-by: Saket Aryan <94069182+whysosaket@users.noreply.github.com> --- openmemory/api/app/utils/categorization.py | 5 ++--- openmemory/api/requirements.txt | 1 + 2 files changed, 3 insertions(+), 3 deletions(-) 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