fast-agent-mcp 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

{fast_agent_mcp-0.1.3.dist-info → fast_agent_mcp-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fast-agent-mcp
-Version: 0.1.3
+Version: 0.1.4
 Summary: Define, Prompt and Test MCP enabled Agents and Workflows
 Author-email: Shaun Smith <fastagent@llmindset.co.uk>, Sarmad Qadri <sarmad@lastmileai.dev>
 License:                                  Apache License

{fast_agent_mcp-0.1.3.dist-info → fast_agent_mcp-0.1.4.dist-info}/RECORD

@@ -24,7 +24,7 @@ mcp_agent/core/enhanced_prompt.py,sha256=Zh1I_PZtQmH_v70dJ-gK5lnLOc42h6_Ai8FGi3a
 mcp_agent/core/error_handling.py,sha256=D3HMW5odrbJvaKqcpCGj6eDXrbFcuqYaCZz7fyYiTu4,623
 mcp_agent/core/exceptions.py,sha256=a2-JGRwFFRoQEPuAq0JC5PhAJ5TO3xVJfdS4-VN29cw,2225
 mcp_agent/core/fastagent.py,sha256=G0GNOD1JMD37rAaZdYRNsa7gXYjvYiV6Cg39laBKRW8,59714
-mcp_agent/core/proxies.py,sha256=5eIqi63wcJMG9aLu_8QF78mS9w4hYas-bcdJqcWZ-fU,6638
+mcp_agent/core/proxies.py,sha256=35k-j-umlQ4sTBK1Z8qNsBS3ciG5ZqAUVazL-Tlmk6s,6238
 mcp_agent/core/server_validation.py,sha256=_59cn16nNT4HGPwg19HgxMtHK4MsdWYDUw_CuL-5xek,1696
 mcp_agent/core/simulator_registry.py,sha256=rcd1cyFGx8MAnN5O0UgwElmVKU_uoIBh9s24pxP33Jc,573
 mcp_agent/core/types.py,sha256=Zhi9iW7uiOfdpSt9NC0FCtGRFtJPg4mpZPK2aYi7a7M,817
@@ -118,8 +118,8 @@ mcp_agent/workflows/swarm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 mcp_agent/workflows/swarm/swarm.py,sha256=-lAIeSWDqbGHGRPTvjiP9nIKWvxxy9DAojl9yQzO1Pw,11050
 mcp_agent/workflows/swarm/swarm_anthropic.py,sha256=pW8zFx5baUWGd5Vw3nIDF2oVOOGNorij4qvGJKdYPcs,1624
 mcp_agent/workflows/swarm/swarm_openai.py,sha256=wfteywvAGkT5bLmIxX_StHJq8144whYmCRnJASAjOes,1596
-fast_agent_mcp-0.1.3.dist-info/METADATA,sha256=Hxlhct68x9F9beQ16vrdjC_AkzP9ex6-VuAuCkNpl-0,27924
-fast_agent_mcp-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fast_agent_mcp-0.1.3.dist-info/entry_points.txt,sha256=2IXtSmDK9XjWN__RWuRIJTgWyW17wJnJ_h-pb0pZAxo,174
-fast_agent_mcp-0.1.3.dist-info/licenses/LICENSE,sha256=cN3FxDURL9XuzE5mhK9L2paZo82LTfjwCYVT7e3j0e4,10939
-fast_agent_mcp-0.1.3.dist-info/RECORD,,
+fast_agent_mcp-0.1.4.dist-info/METADATA,sha256=I8Te4FG-3ClIawLJBEYIkUkaJacG4BiKfF-cuMuUBrs,27924
+fast_agent_mcp-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fast_agent_mcp-0.1.4.dist-info/entry_points.txt,sha256=2IXtSmDK9XjWN__RWuRIJTgWyW17wJnJ_h-pb0pZAxo,174
+fast_agent_mcp-0.1.4.dist-info/licenses/LICENSE,sha256=cN3FxDURL9XuzE5mhK9L2paZo82LTfjwCYVT7e3j0e4,10939
+fast_agent_mcp-0.1.4.dist-info/RECORD,,

mcp_agent/core/proxies.py

@@ -119,52 +119,48 @@ class ChainProxy(BaseAgentProxy):
         self._continue_with_final = True  # Default behavior
         self._cumulative = False  # Default to sequential chaining
 
-    async def generate_str(self, message: str) -> str:
-        """Chain message through a sequence of agents with optional cumulative behavior"""
-        if not self._cumulative:
-            # Original sequential behavior
-            current_message = message
-            for agent_name in self._sequence:
-                proxy = self._agent_proxies[agent_name]
-                current_message = await proxy.generate_str(current_message)
-            return current_message
-        else:
-            # Cumulative behavior
-            original_message = message
-            agent_responses = {}
-
-            for agent_name in self._sequence:
-                proxy = self._agent_proxies[agent_name]
-
-                if not agent_responses:  # First agent
-                    response = await proxy.generate_str(original_message)
-                else:
-                    # Construct context with previous responses
-                    context_message = "The following request was sent to the agents:\n"
-                    context_message += f"<fastagent:request>\n{original_message}\n</fastagent:request>\n\n"
+    async def generate_str(self, message: str, **kwargs) -> str:
+        """Chain message through a sequence of agents.
 
-                    context_message += "Previous agent responses:\n"
+        For the first agent in the chain, pass all kwargs to maintain transparency.
 
-                    for prev_name in self._sequence:
-                        if prev_name in agent_responses:
-                            prev_response = agent_responses[prev_name]
-                            context_message += f'<fastagent:response agent="{prev_name}">\n{prev_response}\n</fastagent:response>\n\n'
+        Two modes of operation:
+        1. Sequential (default): Each agent receives only the output of the previous agent
+        2. Cumulative: Each agent receives all previous agent responses concatenated
+        """
+        if not self._sequence:
+            return message
 
-                    context_message += f"Your task is to build upon this work to address: {original_message}"
+        # Process the first agent (same for both modes)
+        first_agent = self._sequence[0]
+        first_proxy = self._agent_proxies[first_agent]
+        first_response = await first_proxy.generate_str(message, **kwargs)
 
-                    response = await proxy.generate_str(context_message)
+        if len(self._sequence) == 1:
+            return first_response
 
-                agent_responses[agent_name] = response
+        if self._cumulative:
+            # Cumulative mode: each agent gets all previous responses
+            cumulative_response = f"<{first_agent}>\n{first_response}\n</{first_agent}>"
 
-            # Format final output with ALL responses in XML format
-            final_output = "The following request was sent to the agents:\n"
-            final_output += (
-                f"<fastagent:request>\n{original_message}\n</fastagent:request>\n\n"
-            )
+            # Process subsequent agents with cumulative results
+            for agent_name in self._sequence[1:]:
+                proxy = self._agent_proxies[agent_name]
+                # Pass all previous responses to next agent
+                agent_response = await proxy.generate_str(cumulative_response)
+                # Add this agent's response to the cumulative result
+                cumulative_response += (
+                    f"\n\n<{agent_name}>\n{agent_response}\n</{agent_name}>"
+                )
+
+            return cumulative_response
+        else:
+            # Sequential chaining (original behavior)
+            current_message = first_response
 
-            for agent_name in self._sequence:
-                response = agent_responses[agent_name]
-                final_output += f'<fastagent:response agent="{agent_name}">\n{response}\n</fastagent:response>\n\n'
+            # For subsequent agents, just pass the message from previous agent
+            for agent_name in self._sequence[1:]:
+                proxy = self._agent_proxies[agent_name]
+                current_message = await proxy.generate_str(current_message)
 
-            # Return the XML-structured combination of all responses
-            return final_output.strip()
+            return current_message