universal-mcp 0.1.24rc11__tar.gz → 0.1.24rc12__tar.gz

{universal_mcp-0.1.24rc11 → universal_mcp-0.1.24rc12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp
-Version: 0.1.24rc11
+Version: 0.1.24rc12
 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
@@ -14,6 +14,7 @@ Requires-Dist: jinja2>=3.1.3
 Requires-Dist: jsonref>=1.1.0
 Requires-Dist: keyring>=25.6.0
 Requires-Dist: langchain-anthropic>=0.3.19
+Requires-Dist: langchain-google-genai>=2.1.10
 Requires-Dist: langchain-google-vertexai>=2.0.28
 Requires-Dist: langchain-mcp-adapters>=0.1.9
 Requires-Dist: langchain-openai>=0.3.27

{universal_mcp-0.1.24rc11 → universal_mcp-0.1.24rc12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "universal-mcp"
-version = "0.1.24-rc11"
+version = "0.1.24-rc12"
 description = "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."
 readme = "README.md"
 authors = [
@@ -21,6 +21,7 @@ dependencies = [
     "jsonref>=1.1.0",
     "keyring>=25.6.0",
     "langchain-anthropic>=0.3.19",
+    "langchain-google-genai>=2.1.10",
     "langchain-google-vertexai>=2.0.28",
     "langchain-mcp-adapters>=0.1.9",
     "langchain-openai>=0.3.27",

{universal_mcp-0.1.24rc11 → universal_mcp-0.1.24rc12}/src/universal_mcp/agents/base.py

@@ -6,10 +6,10 @@ from langchain_core.messages import AIMessageChunk
 from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.types import Command
-
 from .utils import RichCLI
 
 
+
 class BaseAgent:
     def __init__(self, name: str, instructions: str, model: str, memory: BaseCheckpointSaver | None = None, **kwargs):
         self.name = name
@@ -30,6 +30,7 @@ class BaseAgent:
 
     async def stream(self, thread_id: str, user_input: str):
         await self.ainit()
+        aggregate = None
         async for event, metadata in self._graph.astream(
             {"messages": [{"role": "user", "content": user_input}]},
             config={"configurable": {"thread_id": thread_id}},
@@ -38,25 +39,24 @@ class BaseAgent:
             stream_usage=True,
         ):
             # Only forward assistant token chunks that are not tool-related.
-            event = cast(AIMessageChunk, event)
-            if "finish_reason" in event.response_metadata:
-                # Got LLM finish reason ignore it
-                # logger.debug(f"Finish event: {event}")
+            type_ = type(event)
+            if type_ != AIMessageChunk:
+                # TODO: Handle other types of events
                 continue
-
-            # Skip chunks that correspond to tool calls or tool execution phases
-            # - tool_call_chunks present => model is emitting tool call(s)
-            # - metadata tags or node names may indicate tool/quiet phases
+            event = cast(AIMessageChunk, event)
+            aggregate = event if aggregate is None else aggregate + event
             tags = metadata.get("tags", []) if isinstance(metadata, dict) else []
-
             is_quiet = isinstance(tags, list) and ("quiet" in tags)
 
             if is_quiet:
                 continue
-
-            # Emit only the token chunks for the final assistant message.
-            # logger.debug(f"Event: {event}, Metadata: {metadata}")
-            yield event
+            if "finish_reason" in event.response_metadata:
+                # Got LLM finish reason ignore it
+                # logger.debug(f"Finish event: {event}, Metadata: {metadata}")
+                pass
+            else:
+                # logger.debug(f"Event: {event}, Metadata: {metadata}")
+                yield event
         # Send a final finished message
         # The last event would be finish
         event = cast(AIMessageChunk, event)

{universal_mcp-0.1.24rc11 → universal_mcp-0.1.24rc12}/src/universal_mcp/agents/llm.py

@@ -1,5 +1,7 @@
+import os
 from langchain_anthropic import ChatAnthropic
 from langchain_core.language_models import BaseChatModel
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_google_vertexai import ChatVertexAI
 from langchain_google_vertexai.model_garden import ChatAnthropicVertex
 from langchain_openai import AzureChatOpenAI
@@ -12,15 +14,19 @@ def load_chat_model(fully_specified_name: str, tags: list[str] | None = None) ->
     """
     provider, model = fully_specified_name.split("/", maxsplit=1)
     if provider == "google_anthropic_vertex":
-        return ChatAnthropicVertex(model=model, temperature=0.2, location="asia-east1", tags=tags)
+        return ChatAnthropicVertex(model=model, temperature=0.2, location="asia-east1", tags=tags, stream_usage=True,)
     elif provider == "anthropic":
         return ChatAnthropic(
-            model=model, temperature=1, thinking={"type": "enabled", "budget_tokens": 2048}, max_tokens=4096, tags=tags
+            model=model, temperature=1, thinking={"type": "enabled", "budget_tokens": 2048}, max_tokens=4096, tags=tags, stream_usage=True,
         )  # pyright: ignore[reportCallIssue]
     elif provider == "azure":
-        return AzureChatOpenAI(model=model, api_version="2024-12-01-preview", azure_deployment=model, tags=tags)
-    elif provider == "vertex":
-        return ChatVertexAI(model=model, temperature=0.1, tags=tags)
+        return AzureChatOpenAI(model=model, api_version="2024-12-01-preview", azure_deployment=model, tags=tags, stream_usage=True,)
+    elif provider == "gemini":
+        return ChatGoogleGenerativeAI(
+            model= model,
+            temperature=1.0,
+            max_retries=2
+        )
     else:
         raise ValueError(f"Unsupported provider: {provider}")