solana-agent 27.1.0__py3-none-any.whl → 27.3.0__py3-none-any.whl

solana_agent/services/query.py

@@ -11,6 +11,7 @@ from solana_agent.interfaces.services.query import QueryService as QueryServiceI
 from solana_agent.interfaces.services.routing import RoutingService as RoutingServiceInterface
 from solana_agent.services.agent import AgentService
 from solana_agent.services.routing import RoutingService
+from solana_agent.services.knowledge_base import KnowledgeBaseService
 from solana_agent.interfaces.providers.memory import MemoryProvider
 
 
@@ -22,6 +23,8 @@ class QueryService(QueryServiceInterface):
         agent_service: AgentService,
         routing_service: RoutingService,
         memory_provider: Optional[MemoryProvider] = None,
+        knowledge_base: Optional[KnowledgeBaseService] = None,
+        kb_results_count: int = 3,
     ):
         """Initialize the query service.
 
@@ -33,6 +36,8 @@ class QueryService(QueryServiceInterface):
         self.agent_service = agent_service
         self.routing_service = routing_service
         self.memory_provider = memory_provider
+        self.knowledge_base = knowledge_base
+        self.kb_results_count = kb_results_count
 
     async def process(
         self,
@@ -99,12 +104,38 @@ class QueryService(QueryServiceInterface):
             if self.memory_provider:
                 memory_context = await self.memory_provider.retrieve(user_id)
 
+            # Retrieve relevant knowledge from the KB
+            kb_context = ""
+            if self.knowledge_base:
+                try:
+                    kb_results = await self.knowledge_base.query(
+                        query_text=user_text,
+                        top_k=self.kb_results_count,
+                        include_content=True,
+                        include_metadata=False
+                    )
+
+                    if kb_results:
+                        kb_context = "**KNOWLEDGE BASE (CRITICAL: MAKE THIS INFORMATION THE TOP PRIORITY):**\n"
+                        for i, result in enumerate(kb_results, 1):
+                            content = result.get("content", "").strip()
+                            kb_context += f"[{i}] {content}\n\n"
+                except Exception as e:
+                    print(f"Error retrieving knowledge: {e}")
+
             # Route query to appropriate agent
             if router:
                 agent_name = await router.route_query(user_text)
             else:
                 agent_name = await self.routing_service.route_query(user_text)
 
+           # Combine context from memory and knowledge base
+            combined_context = ""
+            if memory_context:
+                combined_context += f"CONVERSATION HISTORY:\n{memory_context}\n\n"
+            if kb_context:
+                combined_context += f"{kb_context}\n"
+
             print(f"Routed to agent: {agent_name}")
 
             # Generate response
@@ -113,7 +144,7 @@ class QueryService(QueryServiceInterface):
                     agent_name=agent_name,
                     user_id=user_id,
                     query=user_text,
-                    memory_context=memory_context,
+                    memory_context=combined_context,
                     output_format="audio",
                     audio_voice=audio_voice,
                     audio_output_format=audio_output_format,
@@ -134,7 +165,7 @@ class QueryService(QueryServiceInterface):
                     agent_name=agent_name,
                     user_id=user_id,
                     query=user_text,
-                    memory_context=memory_context,
+                    memory_context=combined_context,
                     output_format="text",
                     prompt=prompt,
                 ):

{solana_agent-27.1.0.dist-info → solana_agent-27.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 27.1.0
+Version: 27.3.0
 Summary: Agentic IQ
 License: MIT
 Keywords: ai,openai,ai agents,agi
@@ -15,9 +15,13 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: instructor (>=1.7.9,<2.0.0)
-Requires-Dist: openai (>=1.72.0,<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: openai (>=1.74.0,<2.0.0)
+Requires-Dist: pinecone (>=6.0.2,<7.0.0)
 Requires-Dist: pydantic (>=2.11.3,<3.0.0)
 Requires-Dist: pymongo (>=4.12.0,<5.0.0)
+Requires-Dist: pypdf (>=5.4.0,<6.0.0)
 Requires-Dist: zep-cloud (>=2.10.1,<3.0.0)
 Project-URL: Documentation, https://docs.solana-agent.com
 Project-URL: Repository, https://github.com/truemagic-coder/solana-agent
@@ -51,6 +55,7 @@ Build your AI business in three lines of code!
 * Business Alignment
 * Extensible Tooling
 * Simple Business Definition
+* Knowledge Base with PDF support
 * Tested & Secure
 * Built in Python
 * Powers [CometHeart](https://cometheart.com) & [WalletBubbles](https://walletbubbles.com)
@@ -70,6 +75,7 @@ Build your AI business in three lines of code!
 * Powerful tool integration using standard Python packages and/or inline tools
 * Assigned tools are utilized by agents automatically and effectively
 * Simple business definition using JSON
+* Integrated Knowledge Base with semantic search and automatic PDF chunking
 
 ## Stack
 
@@ -79,22 +85,17 @@ Build your AI business in three lines of code!
 * [OpenAI](https://openai.com), [Google](https://ai.google.dev), [xAI](https://x.ai) - LLM Providers
 * [MongoDB](https://mongodb.com) - Conversational History (optional)
 * [Zep Cloud](https://getzep.com) - Conversational Memory (optional)
+* [Pinecone](https://pinecone.io) - Knowledge Base (optional)
 
 ### LLMs
 
-* [gpt-4o-mini](https://platform.openai.com/docs/models/gpt-4o-mini)
-* [gemini-2.0-flash](https://ai.google.dev/gemini-api/docs/models#gemini-2.0-flash)
-* [grok-3-mini-fast-beta](https://docs.x.ai/docs/models#models-and-pricing)
-* [tts-1](https://platform.openai.com/docs/models/tts-1)
-* [gpt-4o-mini-transcribe](https://platform.openai.com/docs/models/gpt-4o-mini-transcribe)
-
-It is recommended to use all three LLM providers as it is the best setup.
-
-Gemini makes the routing about 2x faster.
-
-Grok is a fast reasoning model and makes the answers and conversation much better.
-
-OpenAI is required while Gemini and Grok are optional.
+* [gpt-4.1-mini](https://platform.openai.com/docs/models/gpt-4.1-mini) (agent)
+* [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) or [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)
+* [gemini-2.0-flash](https://ai.google.dev/gemini-api/docs/models#gemini-2.0-flash) (optional)
+* [grok-3-mini-fast-beta](https://docs.x.ai/docs/models#models-and-pricing) (optional)
 
 ## Installation
 
@@ -155,12 +156,6 @@ Keep this in mind while designing your agentic systems using Solana Agent.
 from solana_agent import SolanaAgent
 
 config = {
-    "grok": {
-        "api_key": "your-grok-api-key",
-    },
-    "gemini": {
-        "api_key": "your-gemini-api-key",
-    },
     "openai": {
         "api_key": "your-openai-api-key",
     },
@@ -190,12 +185,6 @@ async for response in solana_agent.process("user123", "What are the latest AI de
 from solana_agent import SolanaAgent
 
 config = {
-    "grok": {
-        "api_key": "your-grok-api-key",
-    },
-    "gemini": {
-        "api_key": "your-gemini-api-key",
-    },
     "openai": {
         "api_key": "your-openai-api-key",
     },
@@ -215,7 +204,7 @@ config = {
 
 solana_agent = SolanaAgent(config=config)
 
-audio_content = audio_file.read()
+audio_content = await audio_file.read()
 
 async for response in solana_agent.process("user123", audio_content, output_format="audio", audio_voice="nova", audio_input_format="webm", audio_output_format="aac"):
     print(response, end="")
@@ -227,12 +216,6 @@ async for response in solana_agent.process("user123", audio_content, output_form
 from solana_agent import SolanaAgent
 
 config = {
-    "grok": {
-        "api_key": "your-grok-api-key",
-    },
-    "gemini": {
-        "api_key": "your-gemini-api-key",
-    },
     "openai": {
         "api_key": "your-openai-api-key",
     },
@@ -262,12 +245,6 @@ async for response in solana_agent.process("user123", "What is the latest news o
 from solana_agent import SolanaAgent
 
 config = {
-    "grok": {
-        "api_key": "your-grok-api-key",
-    },
-    "gemini": {
-        "api_key": "your-gemini-api-key",
-    },
     "openai": {
         "api_key": "your-openai-api-key",
     },
@@ -287,7 +264,7 @@ config = {
 
 solana_agent = SolanaAgent(config=config)
 
-audio_content = audio_file.read()
+audio_content = await audio_file.read()
 
 async for response in solana_agent.process("user123", audio_content, audio_input_format="aac"):
     print(response, end="")
@@ -334,6 +311,138 @@ config = {
 }
 ```
 
+### Gemini
+
+This allows Gemini to replace OpenAI for agent and router.
+
+```python
+config = {
+    "gemini": {
+        "api_key": "your-gemini-api-key",
+    },
+}
+```
+
+### Grok
+
+This allows Grok to replace OpenAI (or Gemini) for agent.
+
+```python
+config = {
+    "grok": {
+        "api_key": "your-grok-api-key",
+    },
+}
+```
+
+### Knowledge Base
+
+The Knowledge Base (KB) is meant to store text values and/or small PDFs.
+
+```python
+config = {
+    "knowledge_base": {
+        "pinecone": {
+            "api_key": "your-pinecone-api-key",
+            "index_name": "your-pinecone-index-name",
+        }
+    },
+    "mongo": {
+        "connection_string": "your-mongo-connection-string",
+        "database": "your-database-name"
+    },
+}
+```
+
+#### Example for KB (text)
+
+```python
+from solana_agent import SolanaAgent
+
+config = {
+    "openai": {
+        "api_key": "your-openai-api-key",
+    },
+    "knowledge_base": {
+        "pinecone": {
+            "api_key": "your-pinecone-api-key",
+            "index_name": "your-pinecone-index-name",
+        }
+    },
+    "mongo": {
+        "connection_string": "your-mongo-connection-string",
+        "database": "your-database-name"
+    },
+    "agents": [
+        {
+            "name": "kb_expert",
+            "instructions": "You answer questions based on the provided knowledge base documents.",
+            "specialization": "Company Knowledge",
+        }
+    ]
+}
+
+solana_agent = SolanaAgent(config=config)
+
+doc_text = "Solana Agent is a Python framework for building multi-agent AI systems."
+doc_metadata = {
+    "source": "internal_docs",
+    "version": "1.0",
+    "tags": ["framework", "python", "ai"]
+}
+await solana_agent.kb_add_document(text=doc_text, metadata=doc_metadata)
+
+async for response in solana_agent.process("user123", "What is Solana Agent?"):
+    print(response, end="")
+```
+
+#### Example for KB (pdf)
+
+```python
+from solana_agent import SolanaAgent
+
+config = {
+    "openai": {
+        "api_key": "your-openai-api-key",
+    },
+    "knowledge_base": {
+        "pinecone": {
+            "api_key": "your-pinecone-api-key",
+            "index_name": "your-pinecone-index-name",
+        }
+    },
+    "mongo": {
+        "connection_string": "your-mongo-connection-string",
+        "database": "your-database-name"
+    },
+    "agents": [
+        {
+            "name": "kb_expert",
+            "instructions": "You answer questions based on the provided knowledge base documents.",
+            "specialization": "Company Knowledge",
+        }
+    ]
+}
+
+solana_agent = SolanaAgent(config=config)
+
+pdf_bytes = await pdf_file.read()
+
+pdf_metadata = {
+    "source": "annual_report_2024.pdf",
+    "year": 2024,
+    "tags": ["finance", "report"]
+}
+
+await solana_agent.kb_add_pdf_document(
+    pdf_data=pdf_bytes,
+    metadata=pdf_metadata,
+)
+
+async for response in solana_agent.process("user123", "Summarize the annual report for 2024."):
+    print(response, end="")
+```
+
 ## Tools
 
 Tools can be used from plugins like Solana Agent Kit (sakit) or via inline tools. Tools available via plugins integrate automatically with Solana Agent.
@@ -351,12 +460,6 @@ Tools can be used from plugins like Solana Agent Kit (sakit) or via inline tools
 from solana_agent import SolanaAgent
 
 config = {
-    "grok": {
-        "api_key": "your-grok-api-key",
-    },
-    "gemini": {
-        "api_key": "your-gemini-api-key",
-    },
     "openai": {
         "api_key": "your-openai-api-key",
     },
@@ -439,12 +542,6 @@ class TestTool(Tool):
             }
 
 config = {
-    "grok": {
-        "api_key": "your-grok-api-key",
-    },
-    "gemini": {
-        "api_key": "your-gemini-api-key",
-    },
     "openai": {
         "api_key": "your-openai-api-key",
     },
@@ -472,24 +569,14 @@ async for response in solana_agent.process("user123", "What are the latest AI de
     print(response, end="")
 ```
 
-## Agent Training
-
-Many use cases for Solana Agent require training your agents on your company data.
-
-This can be accomplished via runtime prompt injection. Integrations that work well with this method are vector stores like Pinecone and FAQs.
+## Advanced Customization
 
-This knowledge is accessible to all your AI agents.
+### Runtime Prompt Injection
 
 ```python
 from solana_agent import SolanaAgent
 
 config = {
-    "grok": {
-        "api_key": "your-grok-api-key",
-    },
-    "gemini": {
-        "api_key": "your-gemini-api-key",
-    },
     "openai": {
         "api_key": "your-openai-api-key",
     },
@@ -508,7 +595,7 @@ async for response in solana_agent.process("user123", "How do replace the latch
     print(response, end="")
 ```
 
-## Custom Routing
+### Custom Routing
 
 In advanced cases like implementing a ticketing system on-top of Solana Agent - you can use your own router.
 
@@ -517,12 +604,6 @@ from solana_agent import SolanaAgent
 from solana_agent.interfaces.services.routing import RoutingService as RoutingServiceInterface
 
 config = {
-    "grok": {
-        "api_key": "your-grok-api-key",
-    },
-    "gemini": {
-        "api_key": "your-gemini-api-key",
-    },
     "openai": {
         "api_key": "your-openai-api-key",
     },

{solana_agent-27.1.0.dist-info → solana_agent-27.3.0.dist-info}/RECORD

@@ -1,21 +1,24 @@
 solana_agent/__init__.py,sha256=ceYeUpjIitpln8YK1r0JVJU8mzG6cRPYu-HLny3d-Tw,887
 solana_agent/adapters/__init__.py,sha256=tiEEuuy0NF3ngc_tGEcRTt71zVI58v3dYY9RvMrF2Cg,204
-solana_agent/adapters/llm_adapter.py,sha256=Zx2-8LMixocDrymBFpS_UkUNm9PyL50XXLqlFCjdtFk,9050
 solana_agent/adapters/mongodb_adapter.py,sha256=qqEFbY_v1XGyFXBmwd5HSXSSHnA9wWo-Hm1vGEyIG0k,2718
+solana_agent/adapters/openai_adapter.py,sha256=-GS_ujZIF3OVq3LDWb30a4zEvUADvfECDo7GO1lQnqM,10997
+solana_agent/adapters/pinecone_adapter.py,sha256=xn_353EoFYD8KCTckOOO4e0OKzukeBWpyhGUmDawURw,22591
 solana_agent/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-solana_agent/client/solana_agent.py,sha256=cjzTUYIaCpH152XlkX7YxeY6z3Jja7KNfkAFdTxmZyc,5208
+solana_agent/client/solana_agent.py,sha256=gARyoQkWStre3bwaTrR1AoAqdCpI3sw_p0hl6z6kkmA,9982
 solana_agent/domains/__init__.py,sha256=HiC94wVPRy-QDJSSRywCRrhrFfTBeHjfi5z-QfZv46U,168
 solana_agent/domains/agent.py,sha256=WTo-pEc66V6D_35cpDE-kTsw1SJM-dtylPZ7em5em7Q,2659
 solana_agent/domains/routing.py,sha256=UDlgTjUoC9xIBVYu_dnf9-KG_bBgdEXAv_UtDOrYo0w,650
 solana_agent/factories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-solana_agent/factories/agent_factory.py,sha256=tVy29vHObuyALYcU8iNwVv200WOpj7np86_DAOXSeiQ,7897
+solana_agent/factories/agent_factory.py,sha256=7O1WdWA9etV2Wk8DRJUf7qTiFD3GOhBN2T278sfRegM,11536
 solana_agent/interfaces/__init__.py,sha256=IQs1WIM1FeKP1-kY2FEfyhol_dB-I-VAe2rD6jrVF6k,355
-solana_agent/interfaces/client/client.py,sha256=CB8YuSsn-Lvinrb12huyIVaFpJqVDh8EHsHJi9SVXM4,1690
+solana_agent/interfaces/client/client.py,sha256=rIZQXFenEpoSWUvZrCMNE-FWwYwZMCBvXSFC8vCWksg,3610
 solana_agent/interfaces/plugins/plugins.py,sha256=T8HPBsekmzVwfU_Rizp-vtzAeYkMlKMYD7U9d0Wjq9c,3338
 solana_agent/interfaces/providers/data_storage.py,sha256=NqGeFvAzhz9rr-liLPRNCGjooB2EIhe-EVsMmX__b0M,1658
-solana_agent/interfaces/providers/llm.py,sha256=DA1vPWfZPhmZuJOwD5j9VxQmXEEEmyaqKVvbphCnyds,1830
+solana_agent/interfaces/providers/llm.py,sha256=gcZeu8ErLoSmdXWKiBVZuNl31c0KIUhGwX3wPBD_xuI,2356
 solana_agent/interfaces/providers/memory.py,sha256=oNOH8WZXVW8assDigIWZAWiwkxbpDiKupxA2RB6tQvQ,1010
+solana_agent/interfaces/providers/vector_storage.py,sha256=zgdlzQuXCRREVSlB-g7QKKNjVoshSe2AFoAt4CHCAB0,1606
 solana_agent/interfaces/services/agent.py,sha256=tVs2piqWVfO1JiRd58e29oP1GWgYuCzberRSfaFfH4M,1979
+solana_agent/interfaces/services/knowledge_base.py,sha256=J4jGeWcxpKL3XpcrqdNuV1xyd5dj2d-hyAmx4AZ272w,2103
 solana_agent/interfaces/services/query.py,sha256=yo2JZPJSy2iwxtkUlMz0emm9u_27xNgnrAFJRHQoulQ,1480
 solana_agent/interfaces/services/routing.py,sha256=UzJC-z-Q9puTWPFGEo2_CAhIxuxP5IRnze7S66NSrsI,397
 solana_agent/plugins/__init__.py,sha256=coZdgJKq1ExOaj6qB810i3rEhbjdVlrkN76ozt_Ojgo,193
@@ -27,9 +30,10 @@ solana_agent/repositories/__init__.py,sha256=fP83w83CGzXLnSdq-C5wbw9EhWTYtqE2lQT
 solana_agent/repositories/memory.py,sha256=0S3oJIwrJgLokFZHc5nL6Wva_CLdnkwJKfAwmMcpa9E,7853
 solana_agent/services/__init__.py,sha256=ab_NXJmwYUCmCrCzuTlZ47bJZINW0Y0F5jfQ9OovidU,163
 solana_agent/services/agent.py,sha256=M1Aukr9xKGP9mL0jM_JqdRdWSqG4LxF0y0PDAzE_fpY,24608
-solana_agent/services/query.py,sha256=bhB6bZKWqsEf_fLXfsSC57q0CtmeDI6jUlxc9a0LGJw,11099
+solana_agent/services/knowledge_base.py,sha256=bvoLZHSrfp6BhT8i4fKhxOkaUkYGCFedLp2uTzL-7xI,32379
+solana_agent/services/query.py,sha256=TAPwgeQViCEg71q9u8oaqD0-W5sR1ipDlPAFkqGlRA8,12528
 solana_agent/services/routing.py,sha256=hC5t98KZPHty9kMX27KcuxcmZlwjm0g59uMkR8n7k_w,6818
-solana_agent-27.1.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-27.1.0.dist-info/METADATA,sha256=e03o52DNgOqoKDsZfPZEb1fo-pT7oGoUCWWlhUkHjT0,20057
-solana_agent-27.1.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-solana_agent-27.1.0.dist-info/RECORD,,
+solana_agent-27.3.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-27.3.0.dist-info/METADATA,sha256=Rij3Z-DwLTDbKX0Max2mmuOE7LcbDLyOPC6dey7tJwU,22183
+solana_agent-27.3.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+solana_agent-27.3.0.dist-info/RECORD,,