trustgraph-flow 1.2.12__tar.gz → 1.2.14__tar.gz

{trustgraph_flow-1.2.12 → trustgraph_flow-1.2.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-flow
-Version: 1.2.12
+Version: 1.2.14
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

{trustgraph_flow-1.2.12 → trustgraph_flow-1.2.14}/trustgraph/agent/react/agent_manager.py

@@ -104,6 +104,13 @@ class AgentManager:
             # Parse Action
             if line.startswith("Action:"):
                 action = line[7:].strip()
+
+                # Get rid of quotation prefix/suffix if present
+                while action and action[0] == '"':
+                    action = action[1:]
+
+                while action and action[-1] == '"':
+                    action = action[:-1]
             
             # Parse Args
             if line.startswith("Args:"):
@@ -250,9 +257,14 @@ class AgentManager:
 
             await think(act.thought)
 
+            logger.debug(f"ACTION: {act.name}")
+
+            logger.debug(f"Tools: {self.tools.keys()}")
+
             if act.name in self.tools:
                 action = self.tools[act.name]
             else:
+                logger.debug(f"Tools: {self.tools}")
                 raise RuntimeError(f"No action for {act.name}!")
 
             logger.debug(f"TOOL>>> {act}")

{trustgraph_flow-1.2.12 → trustgraph_flow-1.2.14}/trustgraph/agent/react/service.py

@@ -45,7 +45,7 @@ class Processor(AgentService):
         )
 
         self.agent = AgentManager(
-            tools=[],
+            tools={},
             additional_context="",
         )
 
@@ -106,11 +106,21 @@ class Processor(AgentService):
                         impl = TextCompletionImpl
                         arguments = TextCompletionImpl.get_arguments()
                     elif impl_id == "mcp-tool":
+                        # For MCP tools, arguments come from config (similar to prompt tools)
+                        config_args = data.get("arguments", [])
+                        arguments = [
+                            Argument(
+                                name=arg.get("name"),
+                                type=arg.get("type"),
+                                description=arg.get("description")
+                            )
+                            for arg in config_args
+                        ]
                         impl = functools.partial(
                             McpToolImpl, 
-                            mcp_tool_id=data.get("mcp-tool")
+                            mcp_tool_id=data.get("mcp-tool"),
+                            arguments=arguments
                         )
-                        arguments = McpToolImpl.get_arguments()
                     elif impl_id == "prompt":
                         # For prompt tools, arguments come from config
                         config_args = data.get("arguments", [])

{trustgraph_flow-1.2.12 → trustgraph_flow-1.2.14}/trustgraph/agent/react/tools.py

@@ -57,15 +57,14 @@ class TextCompletionImpl:
 # the mcp-tool service.
 class McpToolImpl:
 
-    def __init__(self, context, mcp_tool_id):
+    def __init__(self, context, mcp_tool_id, arguments=None):
         self.context = context
         self.mcp_tool_id = mcp_tool_id
+        self.arguments = arguments or []
     
-    @staticmethod
-    def get_arguments():
-        # MCP tools define their own arguments dynamically
-        # For now, we return empty list and let the MCP service handle validation
-        return []
+    def get_arguments(self):
+        # Return configured arguments if available, otherwise empty list for backward compatibility
+        return self.arguments
 
     async def invoke(self, **arguments):
 

trustgraph_flow-1.2.14/trustgraph/flow_version.py

@@ -0,0 +1 @@
+__version__ = "1.2.14"

{trustgraph_flow-1.2.12 → trustgraph_flow-1.2.14}/trustgraph_flow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-flow
-Version: 1.2.12
+Version: 1.2.14
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_flow-1.2.12/trustgraph/flow_version.py

@@ -1 +0,0 @@
-__version__ = "1.2.12"