solana-agent 1.4.2__tar.gz → 1.4.4__tar.gz

{solana_agent-1.4.2 → solana_agent-1.4.4}/PKG-INFO

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

{solana_agent-1.4.2 → solana_agent-1.4.4}/pyproject.toml

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

{solana_agent-1.4.2 → solana_agent-1.4.4}/solana_agent/ai.py

@@ -111,8 +111,8 @@ class AI:
         cohere_api_key: str = None,
         cohere_model: Literal["rerank-v3.5"] = "rerank-v3.5",
         gemini_api_key: str = None,
-        code_interpreter: bool = True,
-        file_search: bool = True,
+        code_interpreter: bool = False,
+        file_search: bool = False,
         openai_assistant_model: Literal["gpt-4o-mini",
                                         "gpt-4o"] = "gpt-4o-mini",
         openai_embedding_model: Literal[
@@ -134,8 +134,8 @@ class AI:
             cohere_api_key (str, optional): API key for Cohere search. Defaults to None
             cohere_model (Literal["rerank-v3.5"], optional): Cohere model for reranking. Defaults to "rerank-v3.5"
             gemini_api_key (str, optional): API key for Gemini search. Defaults to None
-            code_interpreter (bool, optional): Enable code interpretation. Defaults to True
-            file_search (bool, optional): Enable file search tool. Defaults to True
+            code_interpreter (bool, optional): Enable code interpretation. Defaults to False
+            file_search (bool, optional): Enable file search tool. Defaults to False
             openai_assistant_model (Literal["gpt-4o-mini", "gpt-4o"], optional): OpenAI model for assistant. Defaults to "gpt-4o-mini"
             openai_embedding_model (Literal["text-embedding-3-small", "text-embedding-3-large"], optional): OpenAI model for text embedding. Defaults to "text-embedding-3-large"
 
@@ -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,
@@ -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,7 +859,9 @@ 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-2-latest"] = "grok-2-latest") -> 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.