solana-agent 2.1.2__py3-none-any.whl → 3.0.0__py3-none-any.whl

solana_agent/ai.py

@@ -9,6 +9,7 @@ from pydantic import BaseModel
 from pymongo import MongoClient
 from openai import OpenAI
 import inspect
+import pytz
 import requests
 from zep_cloud.client import AsyncZep as AsyncZepCloud
 from zep_cloud.client import Zep as ZepCloud
@@ -69,7 +70,9 @@ class AI:
         pinecone_embed_model: Literal["llama-text-embed-v2"] = "llama-text-embed-v2",
         gemini_api_key: str = None,
         openai_base_url: str = None,
-        openai_model: str = "gpt-4o-mini",
+        main_model: str = "o3-mini",
+        tool_formatting_model: str = "gpt-4o-mini",
+        tool_formatting_instructions: str = None,
     ):
         """Initialize a new AI assistant instance.
 
@@ -86,8 +89,9 @@ class AI:
             pinecone_embed_model (Literal["llama-text-embed-v2"], optional): Pinecone embedding model. Defaults to "llama-text-embed-v2"
             gemini_api_key (str, optional): API key for Gemini search. Defaults to None
             openai_base_url (str, optional): Base URL for OpenAI API. Defaults to None
-            openai_model (str, optional): OpenAI model to use. Defaults to "gpt-4o-mini"
-
+            main_model (str, optional): Main OpenAI model for conversation. Defaults to "o3-mini"
+            tool_formatting_model (str, optional): OpenAI model for tool formatting. Defaults to "gpt-4o-mini"
+            tool_formatting_instructions (str, optional): Instructions for tool formatting
         Example:
             ```python
             ai = AI(
@@ -116,6 +120,7 @@ class AI:
             Always be concise and ensure that your response maintains coherence across the conversation while respecting the user's context and previous data.
         """
         self._instructions = instructions + " " + memory_instructions
+        self._tool_formatting_instructions = tool_formatting_instructions
         self._database: MongoDatabase = database
         self._accumulated_value_queue = asyncio.Queue()
         if zep_api_key and not zep_base_url:
@@ -140,7 +145,8 @@ class AI:
                 self._pinecone_index_name) if self._pinecone else None
         )
         self._openai_base_url = openai_base_url
-        self._openai_model = openai_model
+        self._main_model = main_model
+        self._tool_formatting_model = tool_formatting_model
         self._tools = []
 
     async def __aenter__(self):
@@ -431,16 +437,19 @@ class AI:
         self.kb.delete(ids=[id], namespace=user_id)
         self._database.kb.delete_one({"reference": id})
 
-    def check_time(self) -> str:
+    def check_time(self, timezone: str) -> str:
         """Get current UTC time formatted as a string via Cloudflare's NTP service.
 
+        Args:
+            timezone (str): Timezone to convert the time to (e.g., "America/New_York")
+
         Returns:
-            str: Current UTC time in format 'YYYY-MM-DD HH:MM:SS UTC'
+            str: Current time in the requested timezone in format 'YYYY-MM-DD HH:MM:SS'
 
         Example:
             ```python
-            time = ai.check_time()
-            # Returns: "2025-02-26 15:30:45 UTC"
+            time = ai.check_time("America/New_York")
+            # Returns: "The current time in America/New_York is 2025-02-26 10:30:45"
             ```
 
         Note:
@@ -448,14 +457,25 @@ class AI:
             Fetches time over NTP from Cloudflare's time server (time.cloudflare.com).
         """
         try:
+            # Request time from Cloudflare's NTP server
             client = ntplib.NTPClient()
-            # Request time from Cloudflare's NTP server.
             response = client.request("time.cloudflare.com", version=3)
-            dt = datetime.datetime.fromtimestamp(
+
+            # Get UTC time from NTP response
+            utc_dt = datetime.datetime.fromtimestamp(
                 response.tx_time, datetime.timezone.utc)
-            return dt.strftime("%Y-%m-%d %H:%M:%S UTC")
+
+            # Convert to requested timezone
+            try:
+                tz = pytz.timezone(timezone)
+                local_dt = utc_dt.astimezone(tz)
+                formatted_time = local_dt.strftime("%Y-%m-%d %H:%M:%S")
+                return f"The current time in {timezone} is {formatted_time}"
+            except pytz.exceptions.UnknownTimeZoneError:
+                return f"Error: Unknown timezone '{timezone}'. Please use a valid timezone like 'America/New_York'."
+
         except Exception as e:
-            return f"Error: {e}"
+            return f"Error getting the current time: {e}"
 
     # has to be sync for tool
     def get_memory_context(
@@ -788,7 +808,7 @@ class AI:
         async def stream_processor():
             memory = self.get_memory_context(user_id)
             response = self._client.chat.completions.create(
-                model=self._openai_model,
+                model=self._main_model,
                 messages=[
                     {
                         "role": "system",
@@ -828,11 +848,11 @@ class AI:
                                 # Execute the tool call (synchronously; adjust if async is needed)
                                 result = func(**args)
                                 response = self._client.chat.completions.create(
-                                    model=self._openai_model,
+                                    model=self._tool_formatting_model,
                                     messages=[
                                         {
                                             "role": "system",
-                                            "content": self._instructions,
+                                            "content": self._tool_formatting_instructions,
                                         },
                                         {
                                             "role": "user",
@@ -988,11 +1008,11 @@ class AI:
                                 # Execute the tool call (synchronously; adjust if async is needed)
                                 result = func(**args)
                                 response = self._client.chat.completions.create(
-                                    model=self._openai_model,
+                                    model=self._tool_formatting_model,
                                     messages=[
                                         {
                                             "role": "system",
-                                            "content": self._instructions,
+                                            "content": self._tool_formatting_instructions,
                                         },
                                         {
                                             "role": "user",

{solana_agent-2.1.2.dist-info → solana_agent-3.0.0.dist-info}/METADATA

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

solana_agent-3.0.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
+solana_agent/ai.py,sha256=nPUsnlaTH6lkXmk0lBay3XHbQu2znBuFNvmbmGvU2gE,45399
+solana_agent-3.0.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-3.0.0.dist-info/METADATA,sha256=fRspzSg3YAhHY0f52SgrDAfGG9j66E_KLFktbt8praM,4808
+solana_agent-3.0.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+solana_agent-3.0.0.dist-info/RECORD,,

solana_agent-2.1.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
-solana_agent/ai.py,sha256=UdAsO1t3JBeW8UQBYSJjUcT4NUyrMJuAHf-yI2nzT0I,44216
-solana_agent-2.1.2.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-2.1.2.dist-info/METADATA,sha256=wbK_aB2WBqNvGAoVt1eRn6dLWO3DYlqS7pvBGP0j0Qw,4808
-solana_agent-2.1.2.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-solana_agent-2.1.2.dist-info/RECORD,,