fix some tests

Author: truemagic-coder

README.md

@@ -81,7 +81,7 @@ Build your AI agents in three lines of code!
 **OpenAI**
 * [gpt-4.1](https://platform.openai.com/docs/models/gpt-4.1) (agent - can be overridden)
 * [gpt-4.1-nano](https://platform.openai.com/docs/models/gpt-4.1-nano) (router)
-* [text-embedding-3-large](https://platform.openai.com/docs/models/text-embedding-3-large) (embedding)
+* [text-embedding-3-small](https://platform.openai.com/docs/models/text-embedding-3-small) (embedding)
 * [tts-1](https://platform.openai.com/docs/models/tts-1) (audio TTS)
 * [gpt-4o-mini-transcribe](https://platform.openai.com/docs/models/gpt-4o-mini-transcribe) (audio transcription)
 * [gpt-image-1](https://platform.openai.com/docs/models/gpt-image-1) (image generation - can be overridden)

pyproject.toml

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

solana_agent/adapters/openai_adapter.py

@@ -36,8 +36,8 @@
 DEFAULT_CHAT_MODEL = "gpt-4.1"
 DEFAULT_VISION_MODEL = "gpt-4.1"
 DEFAULT_PARSE_MODEL = "gpt-4.1-nano"
-DEFAULT_EMBEDDING_MODEL = "text-embedding-3-large"
-DEFAULT_EMBEDDING_DIMENSIONS = 3072
+DEFAULT_EMBEDDING_MODEL = "text-embedding-3-small"
+DEFAULT_EMBEDDING_DIMENSIONS = 1536
 DEFAULT_TRANSCRIPTION_MODEL = "gpt-4o-mini-transcribe"
 DEFAULT_TTS_MODEL = "tts-1"
 
@@ -526,7 +526,7 @@ async def embed_text(
 
         Args:
             text: The text to embed.
-            model: The embedding model to use (defaults to text-embedding-3-large).
+            model: The embedding model to use (defaults to text-embedding-3-small).
             dimensions: Desired output dimensions for the embedding.
 
         Returns:

solana_agent/adapters/pinecone_adapter.py

@@ -33,7 +33,7 @@ def __init__(
         self,
         api_key: Optional[str] = None,
         index_name: Optional[str] = None,
-        # Default for OpenAI text-embedding-3-large, MUST match external embedder
+        # Default for OpenAI text-embedding-3-small, MUST match external embedder
         embedding_dimensions: int = 3072,
         cloud_provider: str = "aws",
         region: str = "us-east-1",

solana_agent/factories/agent_factory.py

@@ -243,10 +243,10 @@ def create_from_config(config: Dict[str, Any]) -> QueryService:
 
                 # Determine OpenAI model and dimensions for KBService
                 openai_model_name = openai_embed_config.get(
-                    "model_name", "text-embedding-3-large"
+                    "model_name", "text-embedding-3-small"
                 )
-                if openai_model_name == "text-embedding-3-large":
-                    openai_dimensions = 3072
+                if openai_model_name == "text-embedding-3-large":  # pragma: no cover
+                    openai_dimensions = 3072  # pragma: no cover
                 elif openai_model_name == "text-embedding-3-small":  # pragma: no cover
                     openai_dimensions = 1536  # pragma: no cover
                 else:  # pragma: no cover
@@ -324,7 +324,7 @@ def create_from_config(config: Dict[str, Any]) -> QueryService:
                         pinecone_adapter=pinecone_adapter,
                         openai_adapter=llm_adapter,
                         embedding_model=graph_memory_config.get(
-                            "embedding_model", "text-embedding-3-large"
+                            "embedding_model", "text-embedding-3-small"
                         ),
                     )
                     logger.info("Graph Memory Service initialized successfully.")

solana_agent/services/graph_memory.py

@@ -15,7 +15,7 @@ def __init__(
         graph_adapter: MongoDBGraphAdapter,
         pinecone_adapter: PineconeAdapter,
         openai_adapter: OpenAIAdapter,
-        embedding_model: str = "text-embedding-3-large",
+        embedding_model: str = "text-embedding-3-small",
     ):
         self.graph = graph_adapter
         self.pinecone = pinecone_adapter

solana_agent/services/knowledge_base.py

@@ -33,7 +33,7 @@ def __init__(
         pinecone_adapter: PineconeAdapter,
         mongodb_adapter: MongoDBAdapter,
         openai_api_key: str,
-        openai_model_name: str = "text-embedding-3-large",
+        openai_model_name: str = "text-embedding-3-small",
         collection_name: str = "knowledge_documents",
         rerank_results: bool = False,
         rerank_top_k: int = 3,
@@ -70,7 +70,7 @@ def __init__(
             )
 
         # Determine expected embedding dimensions based on model name
-        if openai_model_name == "text-embedding-3-large":
+        if openai_model_name == "text-embedding-3-small":
             openai_dimensions = 3072
         elif openai_model_name == "text-embedding-3-small":
             openai_dimensions = 1536

tests/unit/factories/test_agent_factory.py

@@ -703,7 +703,7 @@ def test_create_from_config_with_knowledge_base(
             pinecone_adapter=mock_pinecone_instance,
             mongodb_adapter=mock_mongo_instance,
             openai_api_key="test-openai-key",  # From config
-            openai_model_name="text-embedding-3-large",
+            openai_model_name="text-embedding-3-small",
             collection_name="knowledge_documents",  # From config
             rerank_results=True,  # From config (pinecone.use_reranking)
             rerank_top_k=3,  # From config (knowledge_base.results_count)