universal-mcp 0.1.24rc9__py3-none-any.whl → 0.1.24rc10__py3-none-any.whl

universal_mcp/agents/autoagent/__main__.py

@@ -8,13 +8,11 @@ async def main():
     agent = AutoAgent(
         name="autoagent",
         instructions="You are a helpful assistant that can use tools to help the user.",
-        model="anthropic/claude-4-sonnet-20250514",
+        model="azure/gpt-4.1",
         tool_registry=AgentrRegistry(),
     )
-    result = await agent.invoke(
-        user_input="Send an email to Manoj from my google mail account, manoj@agentr.dev, with the subject 'Hello from auto agent' and the body 'testing'"
-    )
-    print(result)
+    await agent.run_interactive()
+    # print(result)
 
 
 if __name__ == "__main__":

universal_mcp/agents/autoagent/graph.py

@@ -33,12 +33,11 @@ async def build_graph(tool_registry: ToolRegistry, instructions: str = ""):
         """Ask the user a question. Use this tool to ask the user for any missing information for performing a task, or when you have multiple apps to choose from for performing a task."""
         full_question = question
         return f"ASKING_USER: {full_question}"
-    
+
     @tool()
     async def load_tools(tools: list[str]) -> list[str]:
         """Choose the tools you want to use by passing their tool ids.  Loads the tools for the chosen tools and returns the tool ids."""
         return tools
-    
 
     async def call_model(
         state: State,
@@ -48,28 +47,26 @@ async def build_graph(tool_registry: ToolRegistry, instructions: str = ""):
         app_ids = await tool_registry.list_all_apps()
         connections = tool_registry.client.list_my_connections()
         connection_ids = set([connection["app_id"] for connection in connections])
-        connected_apps = [app['id'] for app in app_ids if app["id"] in connection_ids]
-        unconnected_apps = [app['id'] for app in app_ids if app["id"] not in connection_ids]
-        app_id_descriptions = "These are the apps connected to the user's account:\n" + "\n".join([f"{app}" for app in connected_apps])
+        connected_apps = [app["id"] for app in app_ids if app["id"] in connection_ids]
+        unconnected_apps = [app["id"] for app in app_ids if app["id"] not in connection_ids]
+        app_id_descriptions = "These are the apps connected to the user's account:\n" + "\n".join(
+            [f"{app}" for app in connected_apps]
+        )
         if unconnected_apps:
-            app_id_descriptions += "\n\nOther (not connected) apps: " + "\n".join([f"{app}" for app in unconnected_apps])
+            app_id_descriptions += "\n\nOther (not connected) apps: " + "\n".join(
+                [f"{app}" for app in unconnected_apps]
+            )
         print(app_id_descriptions)
         system_prompt = system_prompt.format(system_time=datetime.now(tz=UTC).isoformat(), app_ids=app_id_descriptions)
-        
+
         messages = [{"role": "system", "content": system_prompt + "\n" + instructions}, *state["messages"]]
         model = load_chat_model(runtime.context.model)
         # Load tools from tool registry
         loaded_tools = await tool_registry.export_tools(tools=state["selected_tool_ids"], format=ToolFormat.LANGCHAIN)
         model_with_tools = model.bind_tools([search_tools, ask_user, load_tools, *loaded_tools], tool_choice="auto")
         response_raw = model_with_tools.invoke(messages)
-        token_usage = state.get("token_usage", {})
-        for key in ["input_tokens", "output_tokens", "total_tokens"]:
-            if key in token_usage:
-                token_usage[key] += response_raw.usage_metadata[key]
-            else:
-                token_usage[key] = response_raw.usage_metadata[key]
-        response = cast(AIMessage, response_raw) 
-        return {"messages": [response], "token_usage": token_usage}
+        response = cast(AIMessage, response_raw)
+        return {"messages": [response]}
 
     # Define the conditional edge that determines whether to continue or not
     def should_continue(state: State):
@@ -108,12 +105,12 @@ async def build_graph(tool_registry: ToolRegistry, instructions: str = ""):
                 tools = await search_tools.ainvoke(tool_call["args"])
                 outputs.append(
                     ToolMessage(
-                        content=json.dumps(tools)+"\n\nUse the load_tools tool to load the tools you want to use.",
+                        content=json.dumps(tools) + "\n\nUse the load_tools tool to load the tools you want to use.",
                         name=tool_call["name"],
                         tool_call_id=tool_call["id"],
                     )
                 )
-                
+
             elif tool_call["name"] == load_tools.name:
                 tool_ids = await load_tools.ainvoke(tool_call["args"])
                 print(tool_ids)
@@ -132,23 +129,19 @@ async def build_graph(tool_registry: ToolRegistry, instructions: str = ""):
                         ToolMessage(
                             content=json.dumps(tool_result),
                             name=tool_call["name"],
-                                tool_call_id=tool_call["id"],
-                            )
+                            tool_call_id=tool_call["id"],
                         )
+                    )
                 except Exception as e:
                     outputs.append(
                         ToolMessage(
-                            content=json.dumps("Error: "+str(e)),
+                            content=json.dumps("Error: " + str(e)),
                             name=tool_call["name"],
                             tool_call_id=tool_call["id"],
                         )
                     )
         return {"messages": outputs, "selected_tool_ids": tool_ids}
 
-                
-
-
-
     builder = StateGraph(State, context_schema=Context)
 
     builder.add_node("agent", call_model)

universal_mcp/agents/autoagent/state.py

@@ -25,4 +25,3 @@ def _enqueue(left: list, right: list) -> list:
 
 class State(AgentState):
     selected_tool_ids: Annotated[list[str], _enqueue]
-    token_usage: dict[str, int]

universal_mcp/agents/base.py

@@ -33,6 +33,7 @@ class BaseAgent:
         async for event, _ in self._graph.astream(
             {"messages": [{"role": "user", "content": user_input}]},
             config={"configurable": {"thread_id": thread_id}},
+            context={"system_prompt": self.instructions, "model": self.model},
             stream_mode="messages",
         ):
             event = cast(AIMessageChunk, event)
@@ -101,4 +102,7 @@ class BaseAgent:
                 self.cli.display_info("\nGoodbye! 👋")
                 break
             except Exception as e:
+                import traceback
+
+                traceback.print_exc()
                 self.cli.display_error(f"An error occurred: {str(e)}")

{universal_mcp-0.1.24rc9.dist-info → universal_mcp-0.1.24rc10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.24rc9
+Version: 0.1.24rc10
 Summary: Universal MCP acts as a middle ware for your API applications. It can store your credentials, authorize, enable disable apps on the fly and much more.
 Author-email: Manoj Bajaj <manojbajaj95@gmail.com>
 License: MIT

{universal_mcp-0.1.24rc9.dist-info → universal_mcp-0.1.24rc10.dist-info}/RECORD

@@ -14,7 +14,7 @@ universal_mcp/agentr/registry.py,sha256=O60qOsuby1GTkgc9GXVf9BjHmjMTyDbxqbDfIJ13
 universal_mcp/agentr/server.py,sha256=bIPmHMiKKwnUYnxmfZVRh1thcn7Rytm_-bNiXTfANzc,2098
 universal_mcp/agents/__init__.py,sha256=AMBDQs3p5PePjzdoHYNoPYEsUK_PLHGNVPGxK7yrKVo,270
 universal_mcp/agents/auto.py,sha256=BknCoeexTbFvwmVzYdGiiH72S_r6_5s9tmjH9M0I4d4,25410
-universal_mcp/agents/base.py,sha256=o41gxvwOqWY0IZnupAHfKeH7OG0dgHUQDDpqntsFg6M,4128
+universal_mcp/agents/base.py,sha256=8CZD1ADTIbun6uQE9O7QO-5oz4UUJdwdFWIMBLq8E10,4279
 universal_mcp/agents/cli.py,sha256=7GdRBpu9rhZPiC2vaNQXWI7K-0yCnvdlmE0IFpvy2Gk,539
 universal_mcp/agents/hil.py,sha256=6xi0hhK5g-rhCrAMcGbjcKMReLWPC8rnFZMBOF3N_cY,3687
 universal_mcp/agents/llm.py,sha256=jollWOAUv15IouzbLpuqKzbjj2x2ioZUukSQNyNjb4Y,1382
@@ -23,11 +23,11 @@ universal_mcp/agents/simple.py,sha256=JL8TFyXlA1F4zcArgKhlqVIbLWXetwM05z4MPDJgFe
 universal_mcp/agents/tools.py,sha256=7Vdw0VZYxXVAzAYSpRKWHzVl9Ll6NOnVRlc4cTXguUQ,1335
 universal_mcp/agents/utils.py,sha256=7kwFpD0Rv6JqHG-LlNCVwSu_xRX-N119mUmiBroHJL4,4109
 universal_mcp/agents/autoagent/__init__.py,sha256=E_vMnFz8Z120qdqaKXPNP_--4Tes4jImen7m_iZvtVo,913
-universal_mcp/agents/autoagent/__main__.py,sha256=ps0cT7b9DN-jQK8pOKFVRZf3Oz_dVSBSYWOa8sc7VNc,647
+universal_mcp/agents/autoagent/__main__.py,sha256=tvA-fGN-8x9wA0MyPXydgJ2PxNTY1t3YtweYuZGrIX0,468
 universal_mcp/agents/autoagent/context.py,sha256=RgjW1uCslucxYJpdmi4govd-0V1_9e6Y_kjWl3FpLrE,847
-universal_mcp/agents/autoagent/graph.py,sha256=0vjJD_FIwaAijAnqyvZ5BtfTvLKB3xwPqej16e-D-kE,7250
+universal_mcp/agents/autoagent/graph.py,sha256=8q44BXZLjB00OBXcqEdV-DrZOFkTkdeov62cI7kRwX8,6901
 universal_mcp/agents/autoagent/prompts.py,sha256=ptnXyOarigq96nVW-P1ceT2WRilKvh7NPfE_Cy0NTz4,719
-universal_mcp/agents/autoagent/state.py,sha256=pTKbgTK8TTx7qIioVt98u9KZpeWacWaRmKJWRv3Jj40,791
+universal_mcp/agents/autoagent/state.py,sha256=TQeGZD99okclkoCh5oz-VYIlEsC9yLQyDpnBnm7QCN8,759
 universal_mcp/agents/autoagent/studio.py,sha256=nfVRzPXwBjDORHA0wln2k3Nz-zQXNKgZMvgeqBvkdtM,644
 universal_mcp/agents/autoagent/utils.py,sha256=AFq-8scw_WlSZxDnTzxSNrOSiGYsIlqkqtQLDWf_rMU,431
 universal_mcp/agents/codeact/__init__.py,sha256=5D_I3lI_3tWjZERRoFav_bPe9UDaJ53pDzZYtyixg3E,10097
@@ -72,8 +72,8 @@ universal_mcp/utils/openapi/readme.py,sha256=R2Jp7DUXYNsXPDV6eFTkLiy7MXbSULUj1vH
 universal_mcp/utils/openapi/test_generator.py,sha256=h44gQXEXmrw4pD3-XNHKB7T9c2lDomqrJxVO6oszCqM,12186
 universal_mcp/utils/templates/README.md.j2,sha256=Mrm181YX-o_-WEfKs01Bi2RJy43rBiq2j6fTtbWgbTA,401
 universal_mcp/utils/templates/api_client.py.j2,sha256=972Im7LNUAq3yZTfwDcgivnb-b8u6_JLKWXwoIwXXXQ,908
-universal_mcp-0.1.24rc9.dist-info/METADATA,sha256=8fuQP2tvQjMrU4RrHqhSEdQmN1YyesKjz_9aJ4r-l1w,3143
-universal_mcp-0.1.24rc9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-universal_mcp-0.1.24rc9.dist-info/entry_points.txt,sha256=QlBrVKmA2jIM0q-C-3TQMNJTTWOsOFQvgedBq2rZTS8,56
-universal_mcp-0.1.24rc9.dist-info/licenses/LICENSE,sha256=NweDZVPslBAZFzlgByF158b85GR0f5_tLQgq1NS48To,1063
-universal_mcp-0.1.24rc9.dist-info/RECORD,,
+universal_mcp-0.1.24rc10.dist-info/METADATA,sha256=3VAPupygqW8ScZVKMma6ymtFMaE6rv_CCvedJVMQAtQ,3144
+universal_mcp-0.1.24rc10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+universal_mcp-0.1.24rc10.dist-info/entry_points.txt,sha256=QlBrVKmA2jIM0q-C-3TQMNJTTWOsOFQvgedBq2rZTS8,56
+universal_mcp-0.1.24rc10.dist-info/licenses/LICENSE,sha256=NweDZVPslBAZFzlgByF158b85GR0f5_tLQgq1NS48To,1063
+universal_mcp-0.1.24rc10.dist-info/RECORD,,