mbxai 1.6.0__py3-none-any.whl → 2.0.1__py3-none-any.whl

mbxai-2.0.1.dist-info/METADATA

@@ -0,0 +1,346 @@
+Metadata-Version: 2.4
+Name: mbxai
+Version: 2.0.1
+Summary: MBX AI SDK
+Project-URL: Homepage, https://www.mibexx.de
+Project-URL: Documentation, https://www.mibexx.de
+Project-URL: Repository, https://github.com/yourusername/mbxai.git
+Author: MBX AI
+License: MIT
+License-File: LICENSE
+Classifier: Development Status :: 4 - Beta
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.12
+Requires-Dist: fastapi>=0.115.12
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: mcp>=1.7.1
+Requires-Dist: openai>=1.77.0
+Requires-Dist: pydantic-settings>=2.9.1
+Requires-Dist: pydantic>=2.9.1
+Requires-Dist: python-multipart>=0.0.20
+Requires-Dist: sse-starlette>=2.3.4
+Requires-Dist: starlette>=0.46.2
+Requires-Dist: typing-inspection<=0.4.0
+Requires-Dist: uvicorn>=0.34.2
+Provides-Extra: dev
+Requires-Dist: black>=24.3.0; extra == 'dev'
+Requires-Dist: isort>=5.13.2; extra == 'dev'
+Requires-Dist: mypy>=1.8.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.26.0; extra == 'dev'
+Requires-Dist: pytest-cov>=6.1.1; extra == 'dev'
+Requires-Dist: pytest>=8.3.5; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# MBX AI
+
+A Python library for building AI applications with LLMs.
+
+## Features
+
+- **OpenRouter Integration**: Connect to various LLM providers through OpenRouter
+- **Intelligent Agent System**: AgentClient with dialog-based thinking, question generation, and quality iteration
+- **Tool Integration**: Easily integrate tools with LLMs using the Model Context Protocol (MCP)
+- **Structured Output**: Get structured, typed responses from LLMs
+- **Chat Interface**: Simple chat interface for interacting with LLMs
+- **FastAPI Server**: Built-in FastAPI server for tool integration
+
+## Installation
+
+```bash
+pip install mbxai
+```
+
+## Quick Start
+
+### Basic Usage
+
+```python
+from mbxai import OpenRouterClient
+
+# Initialize the client
+client = OpenRouterClient(api_key="your-api-key")
+
+# Chat with an LLM
+response = await client.chat([
+    {"role": "user", "content": "Hello, how are you?"}
+])
+print(response.choices[0].message.content)
+```
+
+### Quick Agent Example
+
+```python
+from mbxai import AgentClient, OpenRouterClient
+from pydantic import BaseModel, Field
+
+class TravelPlan(BaseModel):
+    destination: str = Field(description="Travel destination")
+    activities: list[str] = Field(description="Recommended activities")
+    budget: str = Field(description="Estimated budget")
+
+# Initialize agent
+client = OpenRouterClient(token="your-api-key")
+agent = AgentClient(client)
+
+# Get intelligent response with automatic quality improvement
+response = agent.agent(
+    prompt="Plan a weekend trip to a mountain destination",
+    final_response_structure=TravelPlan,
+    ask_questions=False
+)
+
+plan = response.final_response
+print(f"Destination: {plan.destination}")
+print(f"Activities: {', '.join(plan.activities)}")
+```
+
+### Using Tools
+
+```python
+from mbxai import OpenRouterClient, ToolClient
+from pydantic import BaseModel
+
+# Define your tool's input and output models
+class CalculatorInput(BaseModel):
+    a: float
+    b: float
+
+class CalculatorOutput(BaseModel):
+    result: float
+
+# Create a calculator tool
+async def calculator(input: CalculatorInput) -> CalculatorOutput:
+    return CalculatorOutput(result=input.a + input.b)
+
+# Initialize the client with tools
+client = ToolClient(OpenRouterClient(api_key="your-api-key"))
+client.add_tool(calculator)
+
+# Use the tool in a chat
+response = await client.chat([
+    {"role": "user", "content": "What is 2 + 3?"}
+])
+print(response.choices[0].message.content)
+```
+
+### Using MCP (Model Context Protocol)
+
+```python
+from mbxai import OpenRouterClient, MCPClient
+from mbxai.mcp import MCPServer
+from mcp.server.fastmcp import FastMCP
+from pydantic import BaseModel
+
+# Define your tool's input and output models
+class CalculatorInput(BaseModel):
+    a: float
+    b: float
+
+class CalculatorOutput(BaseModel):
+    result: float
+
+# Create a FastMCP instance
+mcp = FastMCP("calculator-service")
+
+# Create a calculator tool
+@mcp.tool()
+async def calculator(argument: CalculatorInput) -> CalculatorOutput:
+    return CalculatorOutput(result=argument.a + argument.b)
+
+# Start the MCP server
+server = MCPServer("calculator-service")
+await server.add_tool(calculator)
+await server.start()
+
+# Initialize the MCP client
+client = MCPClient(OpenRouterClient(api_key="your-api-key"))
+await client.register_mcp_server("calculator-service", "http://localhost:8000")
+
+# Use the tool in a chat
+response = await client.chat([
+    {"role": "user", "content": "What is 2 + 3?"}
+])
+print(response.choices[0].message.content)
+```
+
+### Using AgentClient (Intelligent Dialog System)
+
+The `AgentClient` provides an intelligent dialog-based thinking process that can ask clarifying questions, iterate on responses, and provide structured outputs.
+
+#### Basic Agent Usage
+
+```python
+from mbxai import AgentClient, OpenRouterClient
+from pydantic import BaseModel, Field
+
+# Define your response structure
+class BookRecommendation(BaseModel):
+    title: str = Field(description="The title of the recommended book")
+    author: str = Field(description="The author of the book")
+    genre: str = Field(description="The genre of the book")
+    reason: str = Field(description="Why this book is recommended")
+
+# Initialize the agent
+client = OpenRouterClient(token="your-api-key")
+agent = AgentClient(client)
+
+# Get a recommendation with questions
+response = agent.agent(
+    prompt="I want a book recommendation",
+    final_response_structure=BookRecommendation,
+    ask_questions=True  # Agent will ask clarifying questions
+)
+
+if response.has_questions():
+    # Display questions to user
+    for question in response.questions:
+        print(f"Q: {question.question}")
+    
+    # Collect answers and continue
+    from mbxai import AnswerList, Answer
+    answers = AnswerList(answers=[
+        Answer(key="genre", answer="I love science fiction"),
+        Answer(key="complexity", answer="I prefer complex narratives")
+    ])
+    
+    # Continue the conversation
+    final_response = agent.answer_to_agent(response.agent_id, answers)
+    book_rec = final_response.final_response
+    print(f"Recommended: {book_rec.title} by {book_rec.author}")
+else:
+    # Direct response without questions
+    book_rec = response.final_response
+    print(f"Recommended: {book_rec.title} by {book_rec.author}")
+```
+
+#### Agent with Tool Integration
+
+```python
+from mbxai import AgentClient, ToolClient, OpenRouterClient
+
+# Initialize with tool support
+openrouter_client = OpenRouterClient(token="your-api-key")
+tool_client = ToolClient(openrouter_client)
+agent = AgentClient(tool_client)
+
+# Register tools via the agent (schema auto-generated!)
+def get_weather(location: str, unit: str = "fahrenheit") -> dict:
+    """Get weather information for a location.
+    
+    Args:
+        location: The city or location name
+        unit: Temperature unit (fahrenheit or celsius)
+    """
+    return {"location": location, "temperature": "72°F", "conditions": "Sunny"}
+
+agent.register_tool(
+    name="get_weather",
+    description="Get current weather for a location",
+    function=get_weather
+    # Schema automatically generated from function signature!
+)
+
+# Use agent with tools
+class WeatherResponse(BaseModel):
+    location: str = Field(description="The location")
+    weather: str = Field(description="Weather description")
+    recommendations: list[str] = Field(description="Clothing recommendations")
+
+response = agent.agent(
+    prompt="What's the weather in San Francisco and what should I wear?",
+    final_response_structure=WeatherResponse,
+    ask_questions=False
+)
+
+weather_info = response.final_response
+print(f"Weather: {weather_info.weather}")
+```
+
+#### Agent Configuration
+
+```python
+# Configure quality iterations (default: 2)
+agent = AgentClient(
+    ai_client=openrouter_client,
+    max_iterations=3  # More iterations = higher quality, slower response
+)
+
+# Different configurations for different use cases:
+# max_iterations=0: Fastest, basic quality (chatbots)
+# max_iterations=1: Fast, good quality (content generation)
+# max_iterations=2: Balanced (default, recommended)
+# max_iterations=3+: Highest quality (analysis, reports)
+```
+
+#### Agent with MCP Client
+
+```python
+from mbxai import AgentClient, MCPClient
+
+# Initialize with MCP support
+mcp_client = MCPClient(OpenRouterClient(token="your-api-key"))
+agent = AgentClient(mcp_client)
+
+# Register MCP servers
+agent.register_mcp_server("data-analysis", "http://localhost:8000")
+
+# Register individual tools
+agent.register_tool("analyze_data", "Analyze dataset", analyze_function, schema)
+
+# Use agent with full MCP capabilities
+response = agent.agent(
+    prompt="Analyze the sales data and provide insights",
+    final_response_structure=AnalysisReport,
+    ask_questions=True
+)
+```
+
+#### Agent Features
+
+- **Intelligent Questions**: Automatically generates clarifying questions when needed
+- **Quality Iteration**: Improves responses through multiple AI review cycles  
+- **Tool Integration**: Seamlessly works with ToolClient and MCPClient
+- **Structured Output**: Always returns properly typed Pydantic models
+- **Session Management**: Handles multi-turn conversations with question/answer flow
+- **Configurable**: Adjust quality vs speed with max_iterations parameter
+
+#### Supported AI Clients
+
+| Client | Structured Responses | Tool Registration | MCP Server Registration |
+|--------|---------------------|-------------------|------------------------|
+| OpenRouterClient | ✅ | ❌ | ❌ |
+| ToolClient | ✅ | ✅ | ❌ |
+| MCPClient | ✅ | ✅ | ✅ |
+
+## Development
+
+### Setup
+
+1. Clone the repository:
+```bash
+git clone https://github.com/yourusername/mbxai.git
+cd mbxai
+```
+
+2. Create a virtual environment:
+```bash
+python -m venv .venv
+source .venv/bin/activate  # On Windows: .venv\Scripts\activate
+```
+
+3. Install dependencies:
+```bash
+pip install -e ".[dev]"
+```
+
+### Running Tests
+
+```bash
+pytest tests/
+```
+
+## License
+
+MIT License

{mbxai-1.6.0.dist-info → mbxai-2.0.1.dist-info}/RECORD

@@ -1,28 +1,37 @@
-mbxai/__init__.py,sha256=k9FHAi4YDTkuiAHKU5ktijzvf4s2I3EFN1s7jqJPA0o,47
+mbxai/__init__.py,sha256=_p4VGOS8nRm6AeSJthiybglAPS5xZN8kO7Lozd9yqtI,407
 mbxai/core.py,sha256=WMvmU9TTa7M_m-qWsUew4xH8Ul6xseCZ2iBCXJTW-Bs,196
+mbxai/agent/__init__.py,sha256=5j3mW2NZtAU1s2w8n833axWBQsxW8U0qKwoQ9JtQZ4k,289
+mbxai/agent/client.py,sha256=wMgFu_sSkMDx9THemsNmvGZZ3dViJkvOS0p0XrcvKn8,18047
+mbxai/agent/models.py,sha256=oaFBBE_-ufZKUYgjIwbQW4y33BmzXPOnoILzK_vbF24,2147
+mbxai/examples/agent_example.py,sha256=uECWy8QX1IhJMVcdw6EJy6sLcvO8vKgEF_YHJOhZO6Y,5947
+mbxai/examples/agent_iterations_example.py,sha256=xMqZhBWS67EkRkArjOAY2fCgLkQ32Qn9E4CSfEKW4MU,7905
+mbxai/examples/agent_tool_registration_example.py,sha256=oWm0-d4mdba-VQ3HobiCIR0IHtEDCtJenb8Lnm9QqCw,9108
+mbxai/examples/agent_validation_example.py,sha256=xlEf5Mwq5_Iu8bNU4cuHGZVYvAyZNhO2GMFmOom-CLo,4185
+mbxai/examples/auto_schema_example.py,sha256=ymuJJqqDxYznZT2VN6zVFEM7m_lDuccZ1AKSx-xzLTM,8174
 mbxai/examples/openrouter_example.py,sha256=-grXHKMmFLoh-yUIEMc31n8Gg1S7uSazBWCIOWxgbyQ,1317
 mbxai/examples/parse_example.py,sha256=eCKMJoOl6qwo8sDP6Trc6ncgjPlgTqi5tPE2kB5_P0k,3821
 mbxai/examples/parse_tool_example.py,sha256=duHN8scI9ZK6XZ5hdiz1Adzyc-_7tH9Ls9qP4S0bf5s,5477
 mbxai/examples/request.json,sha256=fjVMses305wVUXgcmjESCvPgP81Js8Kk6zHjZ8EDyEg,5434
 mbxai/examples/response.json,sha256=4SGJJyQjWWeN__Mrxm6ZtHIo1NUtLEheldd5KaA2mHw,856
 mbxai/examples/send_request.py,sha256=O5gCHUHy7RvkEFo9IQATgnSOfOdu8OqKHfjAlLDwWPg,6023
+mbxai/examples/simple_agent_test.py,sha256=joCVszUpRkrxHv2DM9QTAh1r6S8iv16pZ-zSPZSBQiU,6391
 mbxai/examples/tool_client_example.py,sha256=9DNaejXLA85dPbExMiv5y76qlFhzOJF9E5EnMOsy_Dc,3993
 mbxai/examples/mcp/mcp_client_example.py,sha256=d5-TRHNDdp3nT_NGt0tKpT3VUAJVvqAHSyqkzk9Dd2s,2972
 mbxai/examples/mcp/mcp_server_example.py,sha256=nFfg22Jnc6HMW_ezLO3So1xwDdx2_rItj5CR-y_Nevs,3966
 mbxai/mcp/__init__.py,sha256=_ek9iYdYqW5saKetj4qDci11jxesQDiHPJRpHMKkxgU,175
 mbxai/mcp/client.py,sha256=QRzId6o4_WRWVv3rtm8cfZZGaoY_UlaOO-oqNjY-tmw,5219
 mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
-mbxai/mcp/server.py,sha256=tSc05RxhXpRdXSpu-noJmK0V48uru0v1JpcUr0P5fmA,3332
+mbxai/mcp/server.py,sha256=H9ZBcS2HE-tnRdm_lNadT5kFSNB2MV4UXw8h3EvXks0,3332
 mbxai/openrouter/__init__.py,sha256=Ito9Qp_B6q-RLGAQcYyTJVWwR2YAZvNqE-HIYXxhtD8,298
 mbxai/openrouter/client.py,sha256=3LD6WDJ8wjo_nefH5d1NJCsrWPvBc_KBf2NsItUoSt8,18302
 mbxai/openrouter/config.py,sha256=Ia93s-auim9Sq71eunVDbn9ET5xX2zusXpV4JBdHAzs,3251
 mbxai/openrouter/models.py,sha256=b3IjjtZAjeGOf2rLsdnCD1HacjTnS8jmv_ZXorc-KJQ,2604
 mbxai/openrouter/schema.py,sha256=H_77ZrA9zmbX155bWpCJj1jehUyJPS0QybEW1IVAoe0,540
 mbxai/tools/__init__.py,sha256=ogxrHvgJ7OR62Lmd5x9Eh5d2C0jqWyQis7Zy3yKpZ78,218
-mbxai/tools/client.py,sha256=j6yB2hYxvWbaQ5SqN1Fs_YFdPtwettdcMoXcdeV-520,14930
+mbxai/tools/client.py,sha256=2wFPD-UN3Y2DSyrnqxt2vvFgTYHzUl14_y0r6fhAWmM,17198
 mbxai/tools/example.py,sha256=1HgKK39zzUuwFbnp3f0ThyWVfA_8P28PZcTwaUw5K78,2232
 mbxai/tools/types.py,sha256=OFfM7scDGTm4FOcJA2ecj-fxL1MEBkqPsT3hqCL1Jto,9505
-mbxai-1.6.0.dist-info/METADATA,sha256=8xzN7T2h5j8XyFMaq2CpXi3zMoe7tyM1RvjzYCpJz6U,4147
-mbxai-1.6.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mbxai-1.6.0.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
-mbxai-1.6.0.dist-info/RECORD,,
+mbxai-2.0.1.dist-info/METADATA,sha256=ISbULkYR1NGO4OjxwBb4LasbnZ0l7EdNpzz3-MZmK44,10018
+mbxai-2.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mbxai-2.0.1.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
+mbxai-2.0.1.dist-info/RECORD,,

mbxai-1.6.0.dist-info/METADATA

@@ -1,169 +0,0 @@
-Metadata-Version: 2.4
-Name: mbxai
-Version: 1.6.0
-Summary: MBX AI SDK
-Project-URL: Homepage, https://www.mibexx.de
-Project-URL: Documentation, https://www.mibexx.de
-Project-URL: Repository, https://github.com/yourusername/mbxai.git
-Author: MBX AI
-License: MIT
-License-File: LICENSE
-Classifier: Development Status :: 4 - Beta
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3.12
-Requires-Python: >=3.12
-Requires-Dist: fastapi>=0.115.12
-Requires-Dist: httpx>=0.27.0
-Requires-Dist: mcp>=1.7.1
-Requires-Dist: openai>=1.77.0
-Requires-Dist: pydantic-settings>=2.9.1
-Requires-Dist: pydantic>=2.9.1
-Requires-Dist: python-multipart>=0.0.20
-Requires-Dist: sse-starlette>=2.3.4
-Requires-Dist: starlette>=0.46.2
-Requires-Dist: typing-inspection<=0.4.0
-Requires-Dist: uvicorn>=0.34.2
-Provides-Extra: dev
-Requires-Dist: black>=24.3.0; extra == 'dev'
-Requires-Dist: isort>=5.13.2; extra == 'dev'
-Requires-Dist: mypy>=1.8.0; extra == 'dev'
-Requires-Dist: pytest-asyncio>=0.26.0; extra == 'dev'
-Requires-Dist: pytest-cov>=6.1.1; extra == 'dev'
-Requires-Dist: pytest>=8.3.5; extra == 'dev'
-Description-Content-Type: text/markdown
-
-# MBX AI
-
-A Python library for building AI applications with LLMs.
-
-## Features
-
-- **OpenRouter Integration**: Connect to various LLM providers through OpenRouter
-- **Tool Integration**: Easily integrate tools with LLMs using the Model Context Protocol (MCP)
-- **Structured Output**: Get structured, typed responses from LLMs
-- **Chat Interface**: Simple chat interface for interacting with LLMs
-- **FastAPI Server**: Built-in FastAPI server for tool integration
-
-## Installation
-
-```bash
-pip install mbxai
-```
-
-## Quick Start
-
-### Basic Usage
-
-```python
-from mbxai import OpenRouterClient
-
-# Initialize the client
-client = OpenRouterClient(api_key="your-api-key")
-
-# Chat with an LLM
-response = await client.chat([
-    {"role": "user", "content": "Hello, how are you?"}
-])
-print(response.choices[0].message.content)
-```
-
-### Using Tools
-
-```python
-from mbxai import OpenRouterClient, ToolClient
-from pydantic import BaseModel
-
-# Define your tool's input and output models
-class CalculatorInput(BaseModel):
-    a: float
-    b: float
-
-class CalculatorOutput(BaseModel):
-    result: float
-
-# Create a calculator tool
-async def calculator(input: CalculatorInput) -> CalculatorOutput:
-    return CalculatorOutput(result=input.a + input.b)
-
-# Initialize the client with tools
-client = ToolClient(OpenRouterClient(api_key="your-api-key"))
-client.add_tool(calculator)
-
-# Use the tool in a chat
-response = await client.chat([
-    {"role": "user", "content": "What is 2 + 3?"}
-])
-print(response.choices[0].message.content)
-```
-
-### Using MCP (Model Context Protocol)
-
-```python
-from mbxai import OpenRouterClient, MCPClient
-from mbxai.mcp import MCPServer
-from mcp.server.fastmcp import FastMCP
-from pydantic import BaseModel
-
-# Define your tool's input and output models
-class CalculatorInput(BaseModel):
-    a: float
-    b: float
-
-class CalculatorOutput(BaseModel):
-    result: float
-
-# Create a FastMCP instance
-mcp = FastMCP("calculator-service")
-
-# Create a calculator tool
-@mcp.tool()
-async def calculator(argument: CalculatorInput) -> CalculatorOutput:
-    return CalculatorOutput(result=argument.a + argument.b)
-
-# Start the MCP server
-server = MCPServer("calculator-service")
-await server.add_tool(calculator)
-await server.start()
-
-# Initialize the MCP client
-client = MCPClient(OpenRouterClient(api_key="your-api-key"))
-await client.register_mcp_server("calculator-service", "http://localhost:8000")
-
-# Use the tool in a chat
-response = await client.chat([
-    {"role": "user", "content": "What is 2 + 3?"}
-])
-print(response.choices[0].message.content)
-```
-
-## Development
-
-### Setup
-
-1. Clone the repository:
-```bash
-git clone https://github.com/yourusername/mbxai.git
-cd mbxai
-```
-
-2. Create a virtual environment:
-```bash
-python -m venv .venv
-source .venv/bin/activate  # On Windows: .venv\Scripts\activate
-```
-
-3. Install dependencies:
-```bash
-pip install -e ".[dev]"
-```
-
-### Running Tests
-
-```bash
-pytest tests/
-```
-
-## License
-
-MIT License