v31.6.0 (#137)

Author: truemagic-coder

README.md

@@ -164,6 +164,7 @@ from solana_agent import SolanaAgent
 config = {
     "openai": {
         "api_key": "your-openai-api-key",
+        "model": "gpt-4.1",  # Optional, defaults to gpt-4.1
     },
     "agents": [
         {

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "31.5.0"
+version = "31.6.0"
 description = "AI Agents for Solana"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"

solana_agent/factories/agent_factory.py

@@ -125,8 +125,11 @@ def create_from_config(config: Dict[str, Any]) -> QueryService:  # pragma: no co
         elif "openai" in config and "api_key" in config["openai"]:
             llm_api_key = config["openai"]["api_key"]
             llm_base_url = None  # Use default OpenAI endpoint
-            llm_model = None  # Will use OpenAI adapter defaults
-            logger.info("Using OpenAI as LLM provider")
+            llm_model = config["openai"].get("model")  # Optional model override
+            if llm_model:
+                logger.info(f"Using OpenAI as LLM provider with model: {llm_model}")
+            else:
+                logger.info("Using OpenAI as LLM provider")
         else:
             raise ValueError(
                 "Either OpenAI, Grok, or Groq API key is required in config."

tests/unit/factories/test_agent_factory.py

@@ -250,6 +250,24 @@ def groq_with_logfire_config(groq_config):
     return config
 
 
+@pytest.fixture
+def openai_with_model_config():
+    """Config with OpenAI and a custom model specified."""
+    return {
+        "openai": {
+            "api_key": "test-openai-key",
+            "model": "gpt-4.1-mini",
+        },
+        "agents": [
+            {
+                "name": "test_agent",
+                "instructions": "You are a test agent.",
+                "specialization": "Testing",
+            }
+        ],
+    }
+
+
 @pytest.fixture
 def invalid_logfire_config_missing_key(base_config):
     """Config with logfire section missing api_key."""
@@ -1855,3 +1873,44 @@ def test_create_groq_without_logfire(
         mock_routing_service.assert_called_once()
         mock_query_service.assert_called_once()
         assert result == mock_query_instance
+
+    @patch("solana_agent.factories.agent_factory.MongoDBAdapter")
+    @patch("solana_agent.factories.agent_factory.OpenAIAdapter")
+    @patch("solana_agent.factories.agent_factory.AgentService")
+    @patch("solana_agent.factories.agent_factory.RoutingService")
+    @patch("solana_agent.factories.agent_factory.QueryService")
+    def test_create_openai_with_model(
+        self,
+        mock_query_service,
+        mock_routing_service,
+        mock_agent_service,
+        mock_openai_adapter,
+        mock_mongo_adapter,
+        openai_with_model_config,
+    ):
+        """Test creating services with OpenAI and a custom model specified."""
+        # Setup mocks
+        mock_openai_instance = MagicMock()
+        mock_openai_adapter.return_value = mock_openai_instance
+        mock_agent_instance = MagicMock()
+        mock_agent_service.return_value = mock_agent_instance
+        mock_agent_instance.tool_registry.list_all_tools.return_value = []
+        mock_routing_instance = MagicMock()
+        mock_routing_service.return_value = mock_routing_instance
+        mock_query_instance = MagicMock()
+        mock_query_service.return_value = mock_query_instance
+
+        # Call the factory
+        result = SolanaAgentFactory.create_from_config(openai_with_model_config)
+
+        # Verify OpenAIAdapter was called with OpenAI config and custom model
+        mock_openai_adapter.assert_called_once_with(
+            api_key="test-openai-key",
+            base_url=None,
+            model="gpt-4.1-mini",
+        )
+        # Verify other services were called
+        mock_agent_service.assert_called_once()
+        mock_routing_service.assert_called_once()
+        mock_query_service.assert_called_once()
+        assert result == mock_query_instance