solana-agent 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

solana_agent/ai.py

@@ -16,6 +16,7 @@ import requests
 from zep_python.client import AsyncZep
 from zep_python.client import Zep
 from zep_python.types import Message, RoleType
+import pandas as pd
 
 
 def adapt_datetime(ts):
@@ -173,6 +174,7 @@ class AI:
         zep_base_url: str = None,
         perplexity_api_key: str = None,
         grok_api_key: str = None,
+        gemini_api_key: str = None,
         code_interpreter: bool = True,
         model: Literal["gpt-4o-mini", "gpt-4o"] = "gpt-4o-mini",
     ):
@@ -195,6 +197,7 @@ class AI:
         )
         self.perplexity_api_key = perplexity_api_key
         self.grok_api_key = grok_api_key
+        self.gemini_api_key = gemini_api_key
 
     async def __aenter__(self):
         assistants = openai.beta.assistants.list()
@@ -250,6 +253,37 @@ class AI:
             thread_id=thread_id, run_id=run_id)
         return run.status
 
+    # converter tool - has to be sync
+    def csv_to_json(self, file_path: str) -> str:
+        df = pd.read_csv(file_path)
+        records = df.to_dict(orient="records")
+        return json.dumps(records)
+
+    # summarize tool - has to be sync
+    def summarize(
+        self, text: str, model: Literal["gemini-2.0-flash"] = "gemini-2.0-flash"
+    ) -> str:
+        try:
+            client = OpenAI(
+                api_key=self.gemini_api_key,
+                base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
+            )
+
+            completion = client.chat.completions.create(
+                model=model,
+                messages=[
+                    {
+                        "role": "system",
+                        "content": "You summarize the text.",
+                    },
+                    {"role": "user", "content": text},
+                ],
+            )
+
+            return completion.choices[0].message.content
+        except Exception as e:
+            return f"Failed to summarize text. Error: {e}"
+
     # search facts tool - has to be sync
     def search_facts(
         self,
@@ -317,6 +351,9 @@ class AI:
         self,
         user_id: str,
         query: str,
+        use_perplexity: bool = True,
+        use_grok: bool = True,
+        use_facts: bool = True,
         perplexity_model: Literal[
             "sonar", "sonar-pro", "sonar-reasoning-pro", "sonar-reasoning"
         ] = "sonar",
@@ -324,11 +361,20 @@ class AI:
         grok_model: Literal["grok-beta"] = "grok-beta",
     ) -> str:
         try:
-            facts = self.search_facts(user_id, query)
-            if not facts:
+            if use_facts:
+                facts = self.search_facts(user_id, query)
+                if not facts:
+                    facts = ""
+            else:
                 facts = ""
-            search_results = self.search_internet(query, perplexity_model)
-            x_search_results = self.search_x(query, grok_model)
+            if use_perplexity:
+                search_results = self.search_internet(query, perplexity_model)
+            else:
+                search_results = ""
+            if use_grok:
+                x_search_results = self.search_x(query, grok_model)
+            else:
+                x_search_results = ""
 
             response = self.client.chat.completions.create(
                 model=openai_model,

{solana_agent-0.0.3.dist-info → solana_agent-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 0.0.3
+Version: 0.0.5
 Summary: The Best AI Agent Framework
 License: MIT
 Keywords: ai,openai,ai agents
@@ -19,6 +19,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: aiosqlite (>=0.21.0,<0.22.0)
 Requires-Dist: motor (>=3.7.0,<4.0.0)
 Requires-Dist: openai (>=1.61.1,<2.0.0)
+Requires-Dist: pandas (>=2.2.3,<3.0.0)
 Requires-Dist: pydantic (>=2.10.6,<3.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: zep-python (>=2.0.2,<3.0.0)
@@ -46,6 +47,8 @@ Solana Agent is the best AI Agent framework.
 - Search Zep facts tool
 - Search X with Grok tool
 - Reasoning tool that combines OpenAI model reasoning, Zep facts, Internet search, and X search.
+- CSV to JSON tool
+- Summarize text tool using Gemini
 - Solana tools upcoming...
 
 ## Installation

solana_agent-0.0.5.dist-info/RECORD

@@ -0,0 +1,6 @@
+solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
+solana_agent/ai.py,sha256=a82WmVPwsQEKMojS9DM4-rSHqdoF3XwKdn92I96UQwA,22001
+solana_agent-0.0.5.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-0.0.5.dist-info/METADATA,sha256=IKT-sJnHh6_jOK7TsSfMH9Fe1NxNvXUONWF7Kr53qn0,2970
+solana_agent-0.0.5.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+solana_agent-0.0.5.dist-info/RECORD,,

solana_agent-0.0.3.dist-info/RECORD

@@ -1,6 +0,0 @@
-solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
-solana_agent/ai.py,sha256=JHpIAGMKy4WxTs2ofA7MQXn-y82F35rAa_Bb71FGzBo,20498
-solana_agent-0.0.3.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-0.0.3.dist-info/METADATA,sha256=pqGWounnqb9zTud-1UPFqV5u1gpBFiFDP1iN5xUAd6o,2877
-solana_agent-0.0.3.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-solana_agent-0.0.3.dist-info/RECORD,,