vanna 2.0.0__tar.gz → 2.0.1__tar.gz

{vanna-2.0.0 → vanna-2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vanna
-Version: 2.0.0
+Version: 2.0.1
 Summary: Generate SQL queries from natural language
 Author-email: Zain Hoda <zain@vanna.ai>
 Requires-Python: >=3.9

{vanna-2.0.0 → vanna-2.0.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "vanna"
-version = "2.0.0"
+version = "2.0.1"
 authors = [
   { name="Zain Hoda", email="zain@vanna.ai" },
 ]

{vanna-2.0.0 → vanna-2.0.1}/src/vanna/integrations/chromadb/agent_memory.py

@@ -215,6 +215,10 @@ class ChromaAgentMemory(AgentMemory):
             for i, (doc_id, metadata) in enumerate(
                 zip(results["ids"], results["metadatas"])
             ):
+                # Skip text memories - they have is_text_memory flag
+                if metadata.get("is_text_memory"):
+                    continue
+
                 args = json.loads(metadata.get("args_json", "{}"))
                 metadata_dict = json.loads(metadata.get("metadata_json", "{}"))
 

{vanna-2.0.0 → vanna-2.0.1}/src/vanna/integrations/faiss/agent_memory.py

@@ -218,6 +218,10 @@ class FAISSAgentMemory(AgentMemory):
 
             memories = []
             for entry in sorted_entries[:limit]:
+                # Skip text memories - they have is_text_memory flag
+                if entry.get("is_text_memory"):
+                    continue
+
                 memory = ToolMemory(
                     memory_id=entry["memory_id"],
                     question=entry["question"],

{vanna-2.0.0 → vanna-2.0.1}/src/vanna/integrations/qdrant/agent_memory.py

@@ -215,6 +215,11 @@ class QdrantAgentMemory(AgentMemory):
             memories = []
             for point in sorted_points[:limit]:
                 payload = point.payload
+
+                # Skip text memories - they have is_text_memory flag
+                if payload.get("is_text_memory"):
+                    continue
+
                 memory = ToolMemory(
                     memory_id=str(point.id),
                     question=payload["question"],