solana-agent 28.0.0__tar.gz → 28.1.0__tar.gz

{solana_agent-28.0.0 → solana_agent-28.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 28.0.0
+Version: 28.1.0
 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://logfire.pydantic.dev/)
 * 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://logfire.pydantic.dev) - 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 → solana_agent-28.1.0}/README.md

@@ -32,6 +32,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)
@@ -43,6 +44,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://logfire.pydantic.dev/)
 * 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
@@ -61,10 +63,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://logfire.pydantic.dev) - Observability and Tracing (optional)
 
 ### AI Models Used
 
@@ -288,6 +293,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 → solana_agent-28.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "28.0.0"
+version = "28.1.0"
 description = "AI Agents for Solana"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -34,6 +34,7 @@ llama-index-core = "^0.12.30"
 llama-index-embeddings-openai = "^0.3.1"
 pypdf = "^5.4.0"
 scrubadub = "^2.0.1"
+logfire = "^3.14.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.5"

{solana_agent-28.0.0 → solana_agent-28.1.0}/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-28.0.0 → solana_agent-28.1.0}/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-28.0.0 → solana_agent-28.1.0}/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}