vanna 2.0.0__py3-none-any.whl → 2.0.1__py3-none-any.whl

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/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/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"],

{vanna-2.0.0.dist-info → vanna-2.0.1.dist-info}/METADATA

@@ -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.dist-info → vanna-2.0.1.dist-info}/RECORD

@@ -128,13 +128,13 @@ vanna/integrations/azuresearch/agent_memory.py,sha256=zzGQmPOgrkLy5PA63joDa5Eprr
 vanna/integrations/bigquery/__init__.py,sha256=Sw-krShPv3nOLQIfVQSflCQWXqIQ-gF_eCuE6GHatNY,108
 vanna/integrations/bigquery/sql_runner.py,sha256=65CcGdX6kxOS5wUGSQ558w4_0oH412tV0f5jVwDczx0,2691
 vanna/integrations/chromadb/__init__.py,sha256=sPumYrf9hMVgEHh6rzXdWu9OrVkOucA5_HkQ4LXfIUY,3642
-vanna/integrations/chromadb/agent_memory.py,sha256=B7cx1vnR1VVrb_gWlxXFVbWXUirHpCSZnTANJejEW20,14683
+vanna/integrations/chromadb/agent_memory.py,sha256=3s06jDc4v1m6MHSMLM3PKza9Em0Pm2hE6NMbGqD8HmU,14833
 vanna/integrations/clickhouse/__init__.py,sha256=Ke7P4HrHtgJZTfCnKv2P8flAS6IY-luZLtJCZmDlCig,114
 vanna/integrations/clickhouse/sql_runner.py,sha256=195z-Uij69AwyPOW3nO6ipjgGfA0oJ1TgxLcsGa7VtE,2350
 vanna/integrations/duckdb/__init__.py,sha256=kQiTv0Z5ozcIpL8FaZpZeYfxgbE6eteqIwjlbaqV4Ig,102
 vanna/integrations/duckdb/sql_runner.py,sha256=YOSp8YjoLBiyQ9RU_SKx8p1K4pIfg7d-ET3JT0siXUc,2088
 vanna/integrations/faiss/__init__.py,sha256=NYCQIJFd-vGocpC9tV6pSvNa7_-1aBiYXBJfimKPgdc,120
-vanna/integrations/faiss/agent_memory.py,sha256=yvv_lzqhUODZXZ8qZJS6u0NM6gdsGAt8fMlYcU3xBOU,14223
+vanna/integrations/faiss/agent_memory.py,sha256=e8fVMRXLutViI8SRst4OccQmDYEpeGF8VCsOPVkxdmo,14370
 vanna/integrations/google/__init__.py,sha256=NaQhnwHA9u7gY4Ohw7jxldMOueoFT3eiRFQaS6kukmY,159
 vanna/integrations/google/gemini.py,sha256=mOI_9vpyZLRpuxy6aCXbuLgxAtpPkgneBGf8ovJJdxg,13091
 vanna/integrations/hive/__init__.py,sha256=SXCrTNqxCZ-RtJkENp9etw6l_XuPIPULGRUiX6tZgrM,96
@@ -176,7 +176,7 @@ vanna/integrations/premium/agent_memory/premium.py,sha256=CbHm1leR4h8cgotErbTMZF
 vanna/integrations/presto/__init__.py,sha256=OAwhistqxKtAtC7o3S9t8ko2q7zd-wo3hcNTsNvNGEg,102
 vanna/integrations/presto/sql_runner.py,sha256=fE-VP8cWI1x0PILxb0YodlBGvl2FWDLy7o_z6O0IaeU,3556
 vanna/integrations/qdrant/__init__.py,sha256=1y-5knDI3LHrnDr-SCfBVYIXNXQKTgumaQDZkQ0Hp28,123
-vanna/integrations/qdrant/agent_memory.py,sha256=SvArrYYMonsPqTdJddkCHPXMOM2ks2HZ-dBYQZ3eBc0,15602
+vanna/integrations/qdrant/agent_memory.py,sha256=uqTmq0u1RFmh9_TG2TtcbwRCosZW5E3u6v92yAzylBQ,15752
 vanna/integrations/snowflake/__init__.py,sha256=7OPjDlyB-v61NYeAos-xAUwj1F-q6evwT9DUw3LUZw0,111
 vanna/integrations/snowflake/sql_runner.py,sha256=qFMWkuMfKt0pCuSut2VVcNvFLxp07RhVT2QTGsp_L7c,5303
 vanna/integrations/sqlite/__init__.py,sha256=1iM-djlwwGBpLiIAV-s5jQ5QZNybrFaW--pEPkp3H4Y,146
@@ -282,8 +282,8 @@ vanna/tools/run_sql.py,sha256=RGptT_cRfD6FP-3MnTS_pgYahWJbDSb3gv3yZ-5sR3g,6951
 vanna/tools/visualize_data.py,sha256=qgPMTxEAmbtJjb2j-qtAG1ZwtzvBvd8gdkzGMki_YGE,7551
 vanna/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vanna/web_components/__init__.py,sha256=QLzEJU3ersllv4pgZpaKdws3S3u5vDyTyUivawMc4A0,1040
-vanna-2.0.0.dist-info/entry_points.txt,sha256=lWU4rLAVNmrWnY6E66z_2TQW3QHU2Kzhwu2qvC6EBrs,62
-vanna-2.0.0.dist-info/licenses/LICENSE,sha256=VYiPMMDqj9BcxUkAYqrAzJpn5tCFXCrnglfRqS5l9Rk,1065
-vanna-2.0.0.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-vanna-2.0.0.dist-info/METADATA,sha256=QFnzdsOREK8EnDUTKOqruqNRt6nuL1IZBX0vTeI3ysI,16519
-vanna-2.0.0.dist-info/RECORD,,
+vanna-2.0.1.dist-info/entry_points.txt,sha256=lWU4rLAVNmrWnY6E66z_2TQW3QHU2Kzhwu2qvC6EBrs,62
+vanna-2.0.1.dist-info/licenses/LICENSE,sha256=VYiPMMDqj9BcxUkAYqrAzJpn5tCFXCrnglfRqS5l9Rk,1065
+vanna-2.0.1.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+vanna-2.0.1.dist-info/METADATA,sha256=6eTDvS_ohU3WNtJwlBvGg1doY2hqDZmt7xnPjuV5Hrs,16519
+vanna-2.0.1.dist-info/RECORD,,