solana-agent 17.0.3__py3-none-any.whl → 17.0.4__py3-none-any.whl

solana_agent/services/query.py

@@ -92,38 +92,70 @@ class QueryService(QueryServiceInterface):
 
             # Route query to appropriate agent
             agent_name = await self.routing_service.route_query(user_text)
-            print(f"DEBUG: Routed to agent: {agent_name}")
+            print(f"Routed to agent: {agent_name}")
 
-            # For text output, we can directly collect and yield chunks
-            if output_format == "text":
-                full_response = ""
+            # For audio mode, we need to carefully handle the response to make sure
+            # tool calls are properly executed and formatted for audio
+            if output_format == "audio":
+                # Use the agent service to generate the response
+                text_response = ""
+
+                # First, get complete text response
+                # Note: This is a separate call from the audio generation
+                temp_response = ""
                 async for chunk in self.agent_service.generate_response(
                     agent_name=agent_name,
                     user_id=user_id,
                     query=user_text,
                     memory_context=memory_context,
-                    output_format="text",
+                    output_format="text"
                 ):
-                    yield chunk
-                    full_response += chunk
+                    temp_response += chunk
 
-            # For audio output, we'll yield audio chunks while collecting text separately
-            else:
-                async for chunk in self.agent_service.generate_response(
+                # Store the complete text for memory
+                text_response = temp_response
+
+                # Now generate audio from same request
+                async for audio_chunk in self.agent_service.generate_response(
                     agent_name=agent_name,
                     user_id=user_id,
                     query=user_text,
                     memory_context=memory_context,
                     output_format="audio",
+                    audio_voice=audio_voice,
                     audio_input_format=audio_input_format,
                     audio_output_format=audio_output_format,
-                    audio_voice=audio_voice,
+                    audio_instructions=audio_instructions
+                ):
+                    yield audio_chunk
+
+                # Store conversation in memory
+                if self.memory_provider and text_response:
+                    await self._store_conversation(
+                        user_id=user_id,
+                        user_message=user_text,
+                        assistant_message=text_response
+                    )
+            else:
+                # For text mode, we can collect the response and store it directly
+                full_text_response = ""
+                async for chunk in self.agent_service.generate_response(
+                    agent_name=agent_name,
+                    user_id=user_id,
+                    query=user_text,
+                    memory_context=memory_context,
+                    output_format="text"
                 ):
                     yield chunk
+                    full_text_response += chunk
 
-            # Store conversation with the full text response
-            if self.memory_provider:
-                await self._store_conversation(user_id, user_text, self.agent_service.last_text_response)
+                # Store conversation in memory
+                if self.memory_provider and full_text_response:
+                    await self._store_conversation(
+                        user_id=user_id,
+                        user_message=user_text,
+                        assistant_message=full_text_response
+                    )
 
         except Exception as e:
             error_msg = f"I apologize for the technical difficulty. {str(e)}"
@@ -248,25 +280,26 @@ class QueryService(QueryServiceInterface):
             }
 
     async def _store_conversation(
-        self, user_id: str, user_text: str, response_text: str
+        self, user_id: str, user_message: str, assistant_message: str
     ) -> None:
         """Store conversation history in memory provider.
 
         Args:
             user_id: User ID
-            user_text: User message
-            response_text: Assistant response
+            user_message: User message
+            assistant_message: Assistant message
         """
         if self.memory_provider:
             try:
                 # Truncate excessively long responses
-                truncated_response = self._truncate(response_text)
+                truncated_assistant_message = self._truncate(assistant_message)
+                truncated_user_message = self._truncate(user_message)
 
                 await self.memory_provider.store(
                     user_id,
                     [
-                        {"role": "user", "content": user_text},
-                        {"role": "assistant", "content": truncated_response},
+                        {"role": "user", "content": truncated_user_message},
+                        {"role": "assistant", "content": truncated_assistant_message},
                     ],
                 )
             except Exception as e:

{solana_agent-17.0.3.dist-info → solana_agent-17.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 17.0.3
+Version: 17.0.4
 Summary: The Future of Work
 License: MIT
 Keywords: ai,openai,ai agents,agi
@@ -13,7 +13,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Dist: openai (>=1.68.0,<2.0.0)
+Requires-Dist: openai (>=1.68.1,<2.0.0)
 Requires-Dist: pydantic (>=2.10.6,<3.0.0)
 Requires-Dist: pymongo (>=4.11.3,<5.0.0)
 Requires-Dist: zep-cloud (>=2.7.0,<3.0.0)

{solana_agent-17.0.3.dist-info → solana_agent-17.0.4.dist-info}/RECORD

@@ -29,9 +29,9 @@ solana_agent/repositories/agent.py,sha256=e1rnsQiigkKwJNLKro86a3b6TBiky3GMfmCRc5
 solana_agent/repositories/memory.py,sha256=GABGwaz00thjviHewLvb18NeKE8dkBROxy_stsiiWrE,4722
 solana_agent/services/__init__.py,sha256=ab_NXJmwYUCmCrCzuTlZ47bJZINW0Y0F5jfQ9OovidU,163
 solana_agent/services/agent.py,sha256=j0aI_BGaY5Nhkb9ga0r4BqI3qMKu5TOc4QPQsU3NHyQ,17000
-solana_agent/services/query.py,sha256=1BAb4UjpydwtJjut8oq724BJohe9ceMigMX-rkI5j7s,10820
+solana_agent/services/query.py,sha256=_RHcT7LDvaVhgie96--l7N4ktEtL6UDtgk1_2t25PtQ,12305
 solana_agent/services/routing.py,sha256=TPJ2Pas4acE93QzMEV6ZP670OtTNrVEPa76fz6urEV4,4996
-solana_agent-17.0.3.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-17.0.3.dist-info/METADATA,sha256=kioM7hvaI97S8skANaI4oAcS2gnKgGMvI2jKTTHf5vg,4888
-solana_agent-17.0.3.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-solana_agent-17.0.3.dist-info/RECORD,,
+solana_agent-17.0.4.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-17.0.4.dist-info/METADATA,sha256=_QLt1zJUJQ-wBy6qYBycjZ6w8bxvcpdwYrxbr2vZWZk,4888
+solana_agent-17.0.4.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+solana_agent-17.0.4.dist-info/RECORD,,