solana-agent 1.0.2__py3-none-any.whl → 1.1.1__py3-none-any.whl

solana_agent/ai.py

@@ -42,9 +42,9 @@ class ToolConfig(BaseModel):
 class MongoDatabase:
     def __init__(self, db_url: str, db_name: str):
         self._client = AsyncIOMotorClient(db_url)
-        self._db = self._client[db_name]
-        self._threads = self._db["threads"]
-        self.messages = self._db["messages"]
+        self.db = self._client[db_name]
+        self._threads = self.db["threads"]
+        self.messages = self.db["messages"]
 
     async def save_thread_id(self, user_id: str, thread_id: str):
         await self._threads.insert_one({"thread_id": thread_id, "user_id": user_id})
@@ -192,6 +192,25 @@ class AI:
         )
         return run.status
 
+    # check time tool - has to be sync
+    def check_time(self) -> str:
+        """Get current UTC time formatted as a string.
+
+        Returns:
+            str: Current UTC time in format 'YYYY-MM-DD HH:MM:SS UTC'
+
+        Example:
+            ```python
+            time = ai.check_time()
+            # Returns: "2024-02-13 15:30:45 UTC"
+            ```
+
+        Note:
+            This is a synchronous tool method required for OpenAI function calling.
+            Always returns time in UTC timezone for consistency.
+        """
+        return datetime.datetime.now(datetime.timezone.utc).strftime("%Y-%m-%d %H:%M:%S %Z")
+
     # search facts tool - has to be sync
     def search_facts(
         self,
@@ -317,7 +336,6 @@ class AI:
         use_perplexity: bool = True,
         use_grok: bool = True,
         use_facts: bool = True,
-        use_kb=True,
         perplexity_model: Literal[
             "sonar", "sonar-pro", "sonar-reasoning-pro", "sonar-reasoning"
         ] = "sonar",
@@ -367,10 +385,6 @@ class AI:
                 x_search_results = self.search_x(query, grok_model)
             else:
                 x_search_results = ""
-            if use_kb:
-                kb_results = self.search_kb(query)
-            else:
-                kb_results = ""
 
             response = self._client.chat.completions.create(
                 model=openai_model,
@@ -381,7 +395,7 @@ class AI:
                     },
                     {
                         "role": "user",
-                        "content": f"Query: {query}, Facts: {facts}, KB Results: {kb_results}, Internet Search Results: {search_results}, X Search Results: {x_search_results}",
+                        "content": f"Query: {query}, Facts: {facts}, Internet Search Results: {search_results}, X Search Results: {x_search_results}",
                     },
                 ],
             )

{solana_agent-1.0.2.dist-info → solana_agent-1.1.1.dist-info}/METADATA

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

solana_agent-1.1.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
+solana_agent/ai.py,sha256=u-CfvihRgQUVSxAJVxFs29byrZxQRAPRqMm1l4mlLLs,29680
+solana_agent-1.1.1.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-1.1.1.dist-info/METADATA,sha256=YorL6wHra39TE0pmhtfyEuiH4mnTiP1P3IZO_J5Fim0,4276
+solana_agent-1.1.1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+solana_agent-1.1.1.dist-info/RECORD,,

solana_agent-1.0.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
-solana_agent/ai.py,sha256=4hAb_iyHr-IkVqi8DFI3-Wu2QiCWSLFV5ji61X05aTk,29229
-solana_agent-1.0.2.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-1.0.2.dist-info/METADATA,sha256=MXo_LkL1ovr54qiEZ_sUJ1yhLIrSPKskhBkn2uQFYlQ,4276
-solana_agent-1.0.2.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-solana_agent-1.0.2.dist-info/RECORD,,