solana-agent 1.4.1__tar.gz → 1.4.3__tar.gz

{solana_agent-1.4.1 → solana_agent-1.4.3}/PKG-INFO

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

{solana_agent-1.4.1 → solana_agent-1.4.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "1.4.1"
+version = "1.4.3"
 description = "Build self-learning AI Agents"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"

{solana_agent-1.4.1 → solana_agent-1.4.3}/solana_agent/ai.py

@@ -380,7 +380,7 @@ class AI:
             filename (str): The name of the CSV file
             prompt (str, optional): Custom prompt for summarization. Defaults to "Summarize the markdown table into a report, include important metrics and totals."
             namespace (str, optional): Knowledge base namespace. Defaults to "global".
-            model (Literal["gemini-2.0-flash", "gemini-1.5-pro"], optional): 
+            model (Literal["gemini-2.0-flash", "gemini-1.5-pro"], optional):
                 Gemini model for summarization. Defaults to "gemini-1.5-pro"
 
         Example:
@@ -766,6 +766,7 @@ class AI:
         self,
         user_id: str,
         query: str,
+        prompt: str = "You combine the data with your reasoning to answer the query.",
         use_perplexity: bool = True,
         use_grok: bool = True,
         use_facts: bool = True,
@@ -774,7 +775,7 @@ class AI:
             "sonar", "sonar-pro", "sonar-reasoning-pro", "sonar-reasoning"
         ] = "sonar",
         openai_model: Literal["o1", "o3-mini"] = "o3-mini",
-        grok_model: Literal["grok-beta"] = "grok-beta",
+        grok_model: Literal["grok-2-latest"] = "grok-2-latest",
         namespace: str = "global",
     ) -> str:
         """Combine multiple data sources with AI reasoning to answer queries.
@@ -782,6 +783,7 @@ class AI:
         Args:
             user_id (str): Unique identifier for the user
             query (str): The question or query to reason about
+            prompt (str, optional): Prompt for reasoning. Defaults to "You combine the data with your reasoning to answer the query."
             use_perplexity (bool, optional): Include Perplexity search results. Defaults to True
             use_grok (bool, optional): Include X/Twitter search results. Defaults to True
             use_facts (bool, optional): Include stored conversation facts. Defaults to True
@@ -844,7 +846,7 @@ class AI:
                 messages=[
                     {
                         "role": "system",
-                        "content": "You combine the data with your reasoning to answer the query.",
+                        "content": prompt,
                     },
                     {
                         "role": "user",
@@ -857,13 +859,15 @@ class AI:
             return f"Failed to reason. Error: {e}"
 
     # x search tool - has to be sync
-    def search_x(self, query: str, model: Literal["grok-beta"] = "grok-beta") -> str:
+    def search_x(
+        self, query: str, model: Literal["grok-2-latest"] = "grok-2-latest"
+    ) -> str:
         try:
             """Search X (formerly Twitter) using Grok API integration.
 
             Args:
                 query (str): Search query to find relevant X posts
-                model (Literal["grok-beta"], optional): Grok model to use. Defaults to "grok-beta"
+                model (Literal["grok-2-latest"], optional): Grok model to use. Defaults to "grok-2-latest"
 
             Returns:
                 str: Search results from X or error message if search fails