solana-agent 0.0.20__py3-none-any.whl → 0.0.22__py3-none-any.whl

solana_agent/ai.py

@@ -141,7 +141,7 @@ class AI:
             Pinecone(api_key=pinecone_api_key) if pinecone_api_key else None
         )
         self._pinecone_index_name = pinecone_index_name if pinecone_index_name else None
-        self._pinecone_index = (
+        self.kb = (
             self._pinecone.Index(
                 self._pinecone_index_name) if self._pinecone else None
         )
@@ -261,7 +261,7 @@ class AI:
                 input=query,
                 model=self._openai_embedding_model,
             )
-            search_results = self._pinecone_index.query(
+            search_results = self.kb.query(
                 vector=response.data[0].embedding,
                 top_k=limit,
                 include_metadata=True,
@@ -302,7 +302,7 @@ class AI:
             input=values,
             model=self._openai_embedding_model,
         )
-        self._pinecone_index.upsert(
+        self.kb.upsert(
             vectors=[
                 {
                     "id": uuid.uuid4().hex,
@@ -377,15 +377,15 @@ class AI:
             limit (int, optional): Maximum number of facts to return. Defaults to 10.
 
         Returns:
-            List[str] | None: List of found facts or None if Zep is not configured
+            str: JSON string of matched facts or error message
 
         Example:
             ```python
             facts = ai.search_facts(
                 user_id="user123",
-                query="project requirements"
+                query="How many cats do I have?"
             )
-            # Returns: ["Fact 1", "Fact 2", ...]
+            # Returns: [{"fact": "user123 has 4 cats", ...}]
             ```
 
         Note:
@@ -393,18 +393,22 @@ class AI:
             This is a synchronous tool method required for OpenAI function calling.
         """
         if self._sync_zep:
-            facts = []
-            results = self._sync_zep.memory.search_sessions(
-                user_id=user_id,
-                session_ids=[user_id],
-                text=query,
-                limit=limit,
-            )
-            for result in results.results:
-                fact = result.fact.fact
-                facts.append(fact)
-            return json.dumps(facts)
-        return None
+            try:
+                facts = []
+                results = self._sync_zep.memory.search_sessions(
+                    user_id=user_id,
+                    session_ids=[user_id],
+                    text=query,
+                    limit=limit,
+                )
+                for result in results.results:
+                    fact = result.fact
+                    facts.append(fact)
+                return json.dumps(facts)
+            except Exception as e:
+                return f"Failed to search facts. Error: {e}"
+        else:
+            return "Zep integration not configured."
 
     # search internet tool - has to be sync
     def search_internet(

{solana_agent-0.0.20.dist-info → solana_agent-0.0.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 0.0.20
+Version: 0.0.22
 Summary: Build self-learning AI Agents
 License: MIT
 Keywords: ai,openai,ai agents

solana_agent-0.0.22.dist-info/RECORD

@@ -0,0 +1,6 @@
+solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
+solana_agent/ai.py,sha256=TMyTxHiFudcYDGLMub74TZV1ZRiWa4kOJ3dp4hjXUyg,33828
+solana_agent-0.0.22.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-0.0.22.dist-info/METADATA,sha256=VhkFrYnj87JRe9hDhIOSDm-uVCm6uDDHFq0FTpCGM84,5577
+solana_agent-0.0.22.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+solana_agent-0.0.22.dist-info/RECORD,,

solana_agent-0.0.20.dist-info/RECORD

@@ -1,6 +0,0 @@
-solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
-solana_agent/ai.py,sha256=UsGmhg710HrujIEsyXykhLYVvGMAT-fuHORMrtoAZ28,33672
-solana_agent-0.0.20.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-0.0.20.dist-info/METADATA,sha256=wOdfWqoaZHVooYnutm1fGbPTaOfszhBWN-wHeBHus-8,5577
-solana_agent-0.0.20.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-solana_agent-0.0.20.dist-info/RECORD,,