fix: update deprecated APIs for chonkie v1.4.0 and qdrant-client v1.16.0 (#705)

* fix: update deprecated APIs and dependency versions

* feat: add chonkie API version compatibility

* chore: update poetry.lock for chonkie version compatibility

---------

Co-authored-by: chunyu li <[email protected]>

Author: zhixiangxue

poetry.lock

@@ -119,7 +119,7 @@ all = [
     # We kindof don't want users to install them.
     "torch (>=2.7.1,<3.0.0)",
     "sentence-transformers (>=4.1.0,<5.0.0)",
-    "qdrant-client (>=1.14.2,<2.0.0)",
+    "qdrant-client (>=1.16.0,<2.0.0)",
     "volcengine-python-sdk (>=4.0.4,<5.0.0)",
     "nltk (>=3.9.1,<4.0.0)",
     "rake-nltk (>=1.0.6,<1.1.0)",

pyproject.toml

@@ -20,12 +20,25 @@ def __init__(self, config: SentenceChunkerConfig):
         from chonkie import SentenceChunker as ChonkieSentenceChunker
 
         self.config = config
-        self.chunker = ChonkieSentenceChunker(
-            tokenizer_or_token_counter=config.tokenizer_or_token_counter,
-            chunk_size=config.chunk_size,
-            chunk_overlap=config.chunk_overlap,
-            min_sentences_per_chunk=config.min_sentences_per_chunk,
-        )
+        
+        # Try new API first (v1.4.0+)
+        try:
+            self.chunker = ChonkieSentenceChunker(
+                tokenizer=config.tokenizer_or_token_counter,
+                chunk_size=config.chunk_size,
+                chunk_overlap=config.chunk_overlap,
+                min_sentences_per_chunk=config.min_sentences_per_chunk,
+            )
+        except (TypeError, AttributeError) as e:
+            # Fallback to old API (<v1.4.0)
+            logger.debug(f"Falling back to old chonkie API: {e}")
+            self.chunker = ChonkieSentenceChunker(
+                tokenizer_or_token_counter=config.tokenizer_or_token_counter,
+                chunk_size=config.chunk_size,
+                chunk_overlap=config.chunk_overlap,
+                min_sentences_per_chunk=config.min_sentences_per_chunk,
+            )
+        
         logger.info(f"Initialized SentenceChunker with config: {config}")
 
     def chunk(self, text: str) -> list[str] | list[Chunk]:

src/memos/chunkers/sentence_chunker.py

@@ -2,7 +2,7 @@
 import os
 import time
 
-from datetime import datetime
+from datetime import datetime, timezone
 from pathlib import Path
 from threading import Lock
 from typing import Any, Literal
@@ -192,7 +192,7 @@ def _register_chat_history(
         self.chat_history_manager[user_id] = ChatHistory(
             user_id=user_id if user_id is not None else self.user_id,
             session_id=session_id if session_id is not None else self.session_id,
-            created_at=datetime.utcnow(),
+            created_at=datetime.now(timezone.utc),
             total_messages=0,
             chat_history=[],
         )

src/memos/mem_os/core.py

@@ -138,14 +138,14 @@ def search(
             List of search results with distance scores and payloads.
         """
         qdrant_filter = self._dict_to_filter(filter) if filter else None
-        response = self.client.search(
+        response = self.client.query_points(
             collection_name=self.config.collection_name,
-            query_vector=query_vector,
+            query=query_vector,
             limit=top_k,
             query_filter=qdrant_filter,
             with_vectors=True,
             with_payload=True,
-        )
+        ).points
         logger.info(f"Qdrant search completed with {len(response)} results.")
         return [
             VecDBItem(