agentic-blocks 0.1.3__tar.gz → 0.1.4__tar.gz

{agentic_blocks-0.1.3/src/agentic_blocks.egg-info → agentic_blocks-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.3
+Version: 0.1.4
 Summary: Simple building blocks for agentic AI systems with MCP client and conversation management
 Author-email: Magnus Bjelkenhed <bjelkenhed@gmail.com>
 License: MIT

{agentic_blocks-0.1.3 → agentic_blocks-0.1.4}/pyproject.toml

@@ -14,7 +14,7 @@ agentic_blocks = []
 
 [project]
 name = "agentic-blocks"
-version = "0.1.3"
+version = "0.1.4"
 description = "Simple building blocks for agentic AI systems with MCP client and conversation management"
 readme = "README.md"
 requires-python = ">=3.11"

{agentic_blocks-0.1.3 → agentic_blocks-0.1.4}/src/agentic_blocks/llm.py

@@ -22,6 +22,7 @@ def call_llm(
     tools: Optional[List[Dict[str, Any]]] = None,
     api_key: Optional[str] = None,
     model: str = "gpt-4o-mini",
+    base_url: Optional[str] = None,
     **kwargs,
 ) -> str:
     """
@@ -32,6 +33,7 @@ def call_llm(
         tools: Optional list of tools in OpenAI function calling format
         api_key: OpenAI API key (if not provided, loads from .env OPENAI_API_KEY)
         model: Model name to use for completion
+        base_url: Base URL for the API (useful for VLLM or other OpenAI-compatible servers)
         **kwargs: Additional parameters to pass to OpenAI API
 
     Returns:
@@ -47,13 +49,18 @@ def call_llm(
     if not api_key:
         api_key = os.getenv("OPENAI_API_KEY")
 
-    if not api_key:
+    if not api_key and not base_url:
         raise LLMError(
             "OpenAI API key not found. Set OPENAI_API_KEY environment variable or pass api_key parameter."
         )
 
     # Initialize OpenAI client
-    client = OpenAI(api_key=api_key)
+    client_kwargs = {}
+    if api_key:
+        client_kwargs["api_key"] = api_key
+    if base_url:
+        client_kwargs["base_url"] = base_url
+    client = OpenAI(**client_kwargs)
 
     # Handle different message input types
     if isinstance(messages, Messages):

{agentic_blocks-0.1.3 → agentic_blocks-0.1.4}/src/agentic_blocks/mcp_client.py

@@ -16,6 +16,23 @@ from mcp.client.streamable_http import streamablehttp_client
 logger = logging.getLogger(__name__)
 
 
+def handle_jupyter_env():
+    """Apply nest_asyncio if running in a Jupyter notebook environment."""
+    try:
+        # Check if we're in a running event loop (like Jupyter)
+        asyncio.get_running_loop()
+        try:
+            import nest_asyncio
+            nest_asyncio.apply()
+        except ImportError:
+            logger.warning(
+                "nest_asyncio not available. Install with: pip install nest-asyncio"
+            )
+    except RuntimeError:
+        # No event loop running, no need for nest_asyncio
+        pass
+
+
 class MCPEndpointError(Exception):
     """Exception raised when there's an error connecting to or using an MCP endpoint."""
 
@@ -77,6 +94,7 @@ class MCPClient:
         Raises:
             MCPEndpointError: If connection or listing fails
         """
+        handle_jupyter_env()
         return asyncio.run(self.list_tools_async())
 
     async def list_tools_async(self) -> List[Dict[str, Any]]:
@@ -158,6 +176,7 @@ class MCPClient:
         Raises:
             MCPEndpointError: If connection or tool call fails
         """
+        handle_jupyter_env()
         return asyncio.run(self.call_tool_async(tool_name, arguments))
 
     async def call_tool_async(

{agentic_blocks-0.1.3 → agentic_blocks-0.1.4/src/agentic_blocks.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.3
+Version: 0.1.4
 Summary: Simple building blocks for agentic AI systems with MCP client and conversation management
 Author-email: Magnus Bjelkenhed <bjelkenhed@gmail.com>
 License: MIT