agentic-blocks 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

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/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.dist-info → agentic_blocks-0.1.4.dist-info}/METADATA

@@ -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.4.dist-info/RECORD

@@ -0,0 +1,9 @@
+agentic_blocks/__init__.py,sha256=LJy2tzTwX9ZjPw8dqkXOWiude7ZDDIaBIvaLC8U4d_Y,435
+agentic_blocks/llm.py,sha256=uXNowV-uGq7G9igqgR-AgERhsYv_myTXFzOUeMjERc8,4253
+agentic_blocks/mcp_client.py,sha256=15mIN_Qw0OVNJAvfgO3jVZS4-AU4TtvEQSFDlL9ruqA,9773
+agentic_blocks/messages.py,sha256=rwcb_goGwfPiDIzl6Up46pAl-5Kw5aFb0uqd6pQtJPY,8162
+agentic_blocks-0.1.4.dist-info/licenses/LICENSE,sha256=r4IcBaAjTv3-yfjXgDPuRD953Qci0Y0nQn5JfHwLyBY,1073
+agentic_blocks-0.1.4.dist-info/METADATA,sha256=ec0HsLSkTeR91sqeNL8rwokCFe8VksPEcrRZ-kO-8Do,9445
+agentic_blocks-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agentic_blocks-0.1.4.dist-info/top_level.txt,sha256=-1a4RAemqicXLU1rRzw4QHV3KlNeQDNxVs3m2gAT238,15
+agentic_blocks-0.1.4.dist-info/RECORD,,

agentic_blocks-0.1.3.dist-info/RECORD

@@ -1,9 +0,0 @@
-agentic_blocks/__init__.py,sha256=LJy2tzTwX9ZjPw8dqkXOWiude7ZDDIaBIvaLC8U4d_Y,435
-agentic_blocks/llm.py,sha256=Rbh0rYDeVyo3S7hRG6tiIfALjchd64zTNz3Q465LUZs,3964
-agentic_blocks/mcp_client.py,sha256=WsEdgzXw8bM76iJWD5BzoJF7MK9wXcUUd5T0TUAWusw,9175
-agentic_blocks/messages.py,sha256=rwcb_goGwfPiDIzl6Up46pAl-5Kw5aFb0uqd6pQtJPY,8162
-agentic_blocks-0.1.3.dist-info/licenses/LICENSE,sha256=r4IcBaAjTv3-yfjXgDPuRD953Qci0Y0nQn5JfHwLyBY,1073
-agentic_blocks-0.1.3.dist-info/METADATA,sha256=Vyggnjbzq4el1dJP82dUjTKeTj1A8Ap-HXZvUlDNVs4,9445
-agentic_blocks-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agentic_blocks-0.1.3.dist-info/top_level.txt,sha256=-1a4RAemqicXLU1rRzw4QHV3KlNeQDNxVs3m2gAT238,15
-agentic_blocks-0.1.3.dist-info/RECORD,,