solana-agent 28.0.0__py3-none-any.whl → 28.1.1__py3-none-any.whl

solana_agent/adapters/openai_adapter.py

@@ -10,6 +10,7 @@ from openai import AsyncOpenAI
 from pydantic import BaseModel
 import instructor
 from instructor import Mode
+import logfire
 
 from solana_agent.interfaces.providers.llm import LLMProvider
 
@@ -26,8 +27,21 @@ DEFAULT_TTS_MODEL = "tts-1"
 class OpenAIAdapter(LLMProvider):
     """OpenAI implementation of LLMProvider with web search capabilities."""
 
-    def __init__(self, api_key: str):
+    def __init__(self, api_key: str, logfire_api_key: Optional[str] = None):
         self.client = AsyncOpenAI(api_key=api_key)
+
+        self.logfire = False
+        if logfire_api_key:
+            try:
+                logfire.configure(token=logfire_api_key)
+                self.logfire = True
+                print("Logfire configured successfully.")  # Optional: confirmation log
+            except Exception as e:
+                print(
+                    f"Failed to configure Logfire: {e}"
+                )  # Log error if configuration fails
+                self.logfire = False  # Ensure logfire is False if config fails
+
         self.parse_model = DEFAULT_PARSE_MODEL
         self.text_model = DEFAULT_CHAT_MODEL
         self.transcription_model = DEFAULT_TRANSCRIPTION_MODEL
@@ -65,6 +79,7 @@ class OpenAIAdapter(LLMProvider):
             Audio bytes as they become available
         """
         try:
+            logfire.instrument_openai(self.client)
             async with self.client.audio.speech.with_streaming_response.create(
                 model=self.tts_model,
                 voice=voice,
@@ -106,6 +121,7 @@ class OpenAIAdapter(LLMProvider):
             Transcript text chunks as they become available
         """
         try:
+            logfire.instrument_openai(self.client)
             async with self.client.audio.transcriptions.with_streaming_response.create(
                 model=self.transcription_model,
                 file=(f"file.{input_format}", audio_bytes),
@@ -149,6 +165,9 @@ class OpenAIAdapter(LLMProvider):
         else:
             client = self.client
 
+        if self.logfire:
+            logfire.instrument_openai(client)
+
         try:
             # Make the non-streaming API call
             response = await client.chat.completions.create(**request_params)
@@ -186,6 +205,9 @@ class OpenAIAdapter(LLMProvider):
             else:
                 client = self.client
 
+            if self.logfire:
+                logfire.instrument_openai(client)
+
             if model:
                 self.parse_model = model
 
@@ -229,6 +251,9 @@ class OpenAIAdapter(LLMProvider):
                     else:
                         client = self.client
 
+                    if self.logfire:
+                        logfire.instrument_openai(client)
+
                     if model:
                         self.parse_model = model
 
@@ -288,6 +313,9 @@ class OpenAIAdapter(LLMProvider):
             # Replace newlines with spaces as recommended by OpenAI
             text = text.replace("\n", " ")
 
+            if self.logfire:
+                logfire.instrument_openai(self.client)
+
             response = await self.client.embeddings.create(
                 input=[text], model=embedding_model, dimensions=embedding_dimensions
             )

solana_agent/factories/agent_factory.py

@@ -78,6 +78,10 @@ class SolanaAgentFactory:
         # Create adapters
 
         if "mongo" in config:
+            if "connection_string" not in config["mongo"]:
+                raise ValueError("MongoDB connection string is required.")
+            if "database" not in config["mongo"]:
+                raise ValueError("MongoDB database name is required.")
             db_adapter = MongoDBAdapter(
                 connection_string=config["mongo"]["connection_string"],
                 database_name=config["mongo"]["database"],
@@ -85,9 +89,21 @@ class SolanaAgentFactory:
         else:
             db_adapter = None
 
-        llm_adapter = OpenAIAdapter(
-            api_key=config["openai"]["api_key"],
-        )
+        if "logfire" in config:
+            if "api_key" not in config["logfire"]:
+                raise ValueError("Pydantic Logfire API key is required.")
+            if "openai" not in config or "api_key" not in config["openai"]:
+                raise ValueError("OpenAI API key is required.")
+            llm_adapter = OpenAIAdapter(
+                api_key=config["openai"]["api_key"],
+                logfire_api_key=config["logfire"].get("api_key"),
+            )
+        else:
+            if "openai" not in config or "api_key" not in config["openai"]:
+                raise ValueError("OpenAI API key is required.")
+            llm_adapter = OpenAIAdapter(
+                api_key=config["openai"].get("api_key"),
+            )
 
         # Create business mission if specified in config
         business_mission = None

solana_agent/services/agent.py

@@ -782,10 +782,6 @@ class AgentService(AgentServiceInterface):
 
         tools_json = json.dumps(simplified_tools, indent=2)
 
-        logger.info(
-            f"Generated tool usage prompt for agent '{agent_name}': {tools_json}"
-        )
-
         return f"""
         AVAILABLE TOOLS:
         {tools_json}

{solana_agent-28.0.0.dist-info → solana_agent-28.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 28.0.0
+Version: 28.1.1
 Summary: AI Agents for Solana
 License: MIT
 Keywords: solana,solana ai,solana agent,ai,ai agent,ai agents
@@ -17,6 +17,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: instructor (>=1.7.9,<2.0.0)
 Requires-Dist: llama-index-core (>=0.12.30,<0.13.0)
 Requires-Dist: llama-index-embeddings-openai (>=0.3.1,<0.4.0)
+Requires-Dist: logfire (>=3.14.0,<4.0.0)
 Requires-Dist: openai (>=1.75.0,<2.0.0)
 Requires-Dist: pinecone (>=6.0.2,<7.0.0)
 Requires-Dist: pydantic (>=2)
@@ -63,6 +64,7 @@ Build your AI agents in three lines of code!
 * Knowledge Base
 * MCP Support
 * Guardrails
+* Pydantic Logfire
 * Tested & Secure
 * Built in Python
 * Powers [CometHeart](https://cometheart.com)
@@ -74,6 +76,7 @@ Build your AI agents in three lines of code!
 * Fast AI responses
 * Solana Ecosystem Integration via [AgentiPy](https://github.com/niceberginc/agentipy)
 * MCP tool usage with first-class support for [Zapier](https://zapier.com/mcp)
+* Integrated observability and tracing via [Pydantic Logfire](https://pydantic.dev/logfire)
 * Designed for a multi-agent swarm 
 * Seamless text and audio streaming with real-time multi-modal processing
 * Persistent memory that preserves context across all agent interactions
@@ -92,10 +95,13 @@ Build your AI agents in three lines of code!
 ### Tech
 
 * [Python](https://python.org) - Programming Language
-* [OpenAI](https://openai.com) - AI Provider
+* [OpenAI](https://openai.com) - AI Model Provider
 * [MongoDB](https://mongodb.com) - Conversational History (optional)
 * [Zep Cloud](https://getzep.com) - Conversational Memory (optional)
 * [Pinecone](https://pinecone.io) - Knowledge Base (optional)
+* [AgentiPy](https://agentipy.fun) - Solana Ecosystem (optional)
+* [Zapier](https://zapier.com) - App Integrations (optional)
+* [Pydantic Logfire](https://pydantic.dev/logfire) - Observability and Tracing (optional)
 
 ### AI Models Used
 
@@ -319,6 +325,16 @@ config = {
 }
 ```
 
+### Observability and Tracing
+
+```python
+config = {
+    "logfire": {
+        "api_key": "your-logfire-write-token",
+    },
+}
+```
+
 ### Knowledge Base
 
 The Knowledge Base (KB) is meant to store text values and/or small PDFs.

{solana_agent-28.0.0.dist-info → solana_agent-28.1.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 solana_agent/__init__.py,sha256=g83qhMOCwcWL19V4CYbQwl0Ykpb0xn49OUh05i-pu3g,1001
 solana_agent/adapters/__init__.py,sha256=tiEEuuy0NF3ngc_tGEcRTt71zVI58v3dYY9RvMrF2Cg,204
 solana_agent/adapters/mongodb_adapter.py,sha256=0KWIa6kaFbUFvtKUzuV_0p0RFlPPGKrDVIEU2McVY3k,2734
-solana_agent/adapters/openai_adapter.py,sha256=QwNMWT2JKChb6RoHzlHWZrhujsi1cn8V4HDPWjUclIs,11105
+solana_agent/adapters/openai_adapter.py,sha256=iDAOKZkN4x2yNarDIrS71w6ufjKnC-wlT3iZFOXCoY4,12099
 solana_agent/adapters/pinecone_adapter.py,sha256=SDbf_XJMuFDKhNfF25_VXaYG3vrmYyPIo2SyhaniEwg,23048
 solana_agent/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 solana_agent/client/solana_agent.py,sha256=jUGWxYJL9ZWxGsVX9C6FrRQyX7r6Cep0ijcfm7cbkJI,10098
@@ -9,7 +9,7 @@ solana_agent/domains/__init__.py,sha256=HiC94wVPRy-QDJSSRywCRrhrFfTBeHjfi5z-QfZv
 solana_agent/domains/agent.py,sha256=3Q1wg4eIul0CPpaYBOjEthKTfcdhf1SAiWc2R-IMGO8,2561
 solana_agent/domains/routing.py,sha256=1yR4IswGcmREGgbOOI6TKCfuM7gYGOhQjLkBqnZ-rNo,582
 solana_agent/factories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-solana_agent/factories/agent_factory.py,sha256=WdXvVE4301vi_5QTDesQG3dCoSnMi4rp7LWvBQQD0bk,11377
+solana_agent/factories/agent_factory.py,sha256=_tBCzx-QIAkGg0v47GiGaQIH7xjlF_5zAt7VFEyqu6k,12273
 solana_agent/guardrails/pii.py,sha256=FCz1IC3mmkr41QFFf5NaC0fwJrVkwFsxgyOCS2POO5I,4428
 solana_agent/interfaces/__init__.py,sha256=IQs1WIM1FeKP1-kY2FEfyhol_dB-I-VAe2rD6jrVF6k,355
 solana_agent/interfaces/client/client.py,sha256=hsvaQiQdz3MLMNc77oD6ocvvnyl7Ez2n087ptFDA19M,3687
@@ -31,11 +31,11 @@ solana_agent/plugins/tools/auto_tool.py,sha256=uihijtlc9CCqCIaRcwPuuN7o1SHIpWL2G
 solana_agent/repositories/__init__.py,sha256=fP83w83CGzXLnSdq-C5wbw9EhWTYtqE2lQTgp46-X_4,163
 solana_agent/repositories/memory.py,sha256=YYpCyiDVi3a5ZOFYFkzBS6MDjo9g2TnwbEZ5KKfKbII,7204
 solana_agent/services/__init__.py,sha256=iko0c2MlF8b_SA_nuBGFllr2E3g_JowOrOzGcnU9tkA,162
-solana_agent/services/agent.py,sha256=6CwTqIzPykx4yGI53BM1UbYhQN5Va1DJu0RzNvzuh6U,41894
+solana_agent/services/agent.py,sha256=9FB1Tj7v8JwJVVmZwK8IOSrBbgbV4iFZOtFHzw3gcEs,41780
 solana_agent/services/knowledge_base.py,sha256=J9V8dNoCCcko3EasiGwK2JJ_A_oG_e-Ni9pgNg0T6wA,33486
 solana_agent/services/query.py,sha256=bAoUfe_2EBVEVeh99-2E9KZ0zaHUzf7Lqel3rlHyNX8,17459
 solana_agent/services/routing.py,sha256=-0fNIKDtCn0-TLUYDFYAE4jPLMeI_jCXIpgtgWDpdf8,6986
-solana_agent-28.0.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-28.0.0.dist-info/METADATA,sha256=UdgCn1QADdEba_j5vwdZmqPAQiHm_fO9sMzqJtvFYL0,26646
-solana_agent-28.0.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-solana_agent-28.0.0.dist-info/RECORD,,
+solana_agent-28.1.1.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-28.1.1.dist-info/METADATA,sha256=JKmf9tN-lKJpjdBzP8UTY8S-yUTo_IrPAW2iAemnnD0,27150
+solana_agent-28.1.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+solana_agent-28.1.1.dist-info/RECORD,,