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

agentic_blocks/llm.py

@@ -22,8 +22,9 @@ 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:
+) -> Any:
     """
     Call an LLM completion API with the provided messages.
 
@@ -32,10 +33,11 @@ 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:
-        The assistant's response content as a string
+        The complete message object from the OpenAI API response
 
     Raises:
         LLMError: If API call fails or configuration is invalid
@@ -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):
@@ -79,8 +86,8 @@ def call_llm(
         # Make completion request
         response = client.chat.completions.create(**completion_params)
 
-        # Extract and return response content
-        return response.choices[0].message.content or ""
+        # Return the complete message object
+        return response.choices[0].message
 
     except Exception as e:
         raise LLMError(f"Failed to call LLM API: {e}")
@@ -125,12 +132,17 @@ def example_usage():
         # Call with Messages object
         print("Using Messages object:")
         response1 = call_llm(messages_obj, temperature=0.7)
-        print(f"Response: {response1}")
+        print(f"Response: {response1.content}")
 
         # Call with raw message list
         print("\nUsing raw message list:")
         response2 = call_llm(messages_list, tools=tools, temperature=0.5)
-        print(f"Response: {response2}")
+        if hasattr(response2, 'tool_calls') and response2.tool_calls:
+            print(f"Tool calls requested: {len(response2.tool_calls)}")
+            for i, tool_call in enumerate(response2.tool_calls):
+                print(f"  {i+1}. {tool_call.function.name}({tool_call.function.arguments})")
+        else:
+            print(f"Response: {response2.content}")
 
     except LLMError as e:
         print(f"Error: {e}")

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.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.3
+Version: 0.1.5
 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.5.dist-info/RECORD

@@ -0,0 +1,9 @@
+agentic_blocks/__init__.py,sha256=LJy2tzTwX9ZjPw8dqkXOWiude7ZDDIaBIvaLC8U4d_Y,435
+agentic_blocks/llm.py,sha256=EetJFAUMAvRYFExjZrVCyeewhqhtnE7o3PvlW9NgKHw,4584
+agentic_blocks/mcp_client.py,sha256=15mIN_Qw0OVNJAvfgO3jVZS4-AU4TtvEQSFDlL9ruqA,9773
+agentic_blocks/messages.py,sha256=rwcb_goGwfPiDIzl6Up46pAl-5Kw5aFb0uqd6pQtJPY,8162
+agentic_blocks-0.1.5.dist-info/licenses/LICENSE,sha256=r4IcBaAjTv3-yfjXgDPuRD953Qci0Y0nQn5JfHwLyBY,1073
+agentic_blocks-0.1.5.dist-info/METADATA,sha256=zHSVjhDqcM2pnKhX_r_0x6iRNqbBHt-Xc-G2hI4PTx8,9445
+agentic_blocks-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agentic_blocks-0.1.5.dist-info/top_level.txt,sha256=-1a4RAemqicXLU1rRzw4QHV3KlNeQDNxVs3m2gAT238,15
+agentic_blocks-0.1.5.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,,