mbxai 2.1.1__py3-none-any.whl → 2.1.3__py3-none-any.whl

mbxai/__init__.py

@@ -7,7 +7,7 @@ from .openrouter import OpenRouterClient
 from .tools import ToolClient
 from .mcp import MCPClient
 
-__version__ = "2.1.1"
+__version__ = "2.1.3"
 
 __all__ = [
     "AgentClient",

mbxai/agent/client.py

@@ -134,6 +134,35 @@ class AgentClient:
             logger.debug(f"🔗 AI call with {len(messages)} messages (no history)")
         return self._ai_client.parse(full_messages, response_format)
 
+    def _validate_answers(self, answers: Any) -> bool:
+        """
+        Validate that answers parameter is a proper AnswerList with content.
+        
+        Args:
+            answers: The answers parameter to validate
+            
+        Returns:
+            True if answers is valid and has content, False otherwise
+        """
+        # Check if answers is the correct type
+        if not isinstance(answers, AnswerList):
+            logger.warning(f"Invalid answers type: {type(answers)}. Expected AnswerList, treating as no answers.")
+            return False
+        
+        # Check if answers has content
+        if not hasattr(answers, 'answers') or not answers.answers:
+            logger.info(f"Empty answers list provided, proceeding without answers processing.")
+            return False
+        
+        # Check if answers list contains valid Answer objects
+        for answer in answers.answers:
+            if not hasattr(answer, 'key') or not hasattr(answer, 'answer'):
+                logger.warning(f"Invalid answer object in list: {answer}. Treating as no answers.")
+                return False
+        
+        logger.debug(f"Validated {len(answers.answers)} answers")
+        return True
+
     def _extract_token_usage(self, response: Any) -> TokenUsage:
         """Extract token usage information from an AI response."""
         try:
@@ -184,39 +213,63 @@ class AgentClient:
 
     def agent(
         self,
-        prompt: str,
-        final_response_structure: Type[BaseModel],
+        prompt: str = None,
+        final_response_structure: Type[BaseModel] = None,
         ask_questions: bool = True,
         agent_id: str = None,
-        answers: AnswerList = None
+        answers: AnswerList | None = None
     ) -> AgentResponse:
         """
         Process a prompt through the agent's thinking process.
 
         Args:
-            prompt: The initial prompt from the user
-            final_response_structure: Pydantic model defining the expected final response format
+            prompt: The prompt from the user (optional if agent_id exists with history)
+            final_response_structure: Pydantic model defining the expected final response format (required for new sessions)
             ask_questions: Whether to ask clarifying questions (default: True)
             agent_id: Optional agent session ID to continue an existing conversation
             answers: Optional answers to questions (when continuing a conversation with questions)
 
         Returns:
             AgentResponse containing either questions to ask or the final response
+            
+        Raises:
+            ValueError: If neither prompt nor agent_id with history is provided, or if final_response_structure is missing for new sessions
         """
-        # Use provided agent_id or create a new one
-        if agent_id is None:
-            agent_id = str(__import__("uuid").uuid4())
+        # Validate inputs and determine session type
+        is_existing_session = agent_id is not None and agent_id in self._agent_sessions
+        existing_session = self._agent_sessions.get(agent_id, {}) if agent_id else {}
+        conversation_history = existing_session.get("conversation_history", []).copy()
+        
+        # Validation logic
+        if not is_existing_session:
+            # New session - both prompt and final_response_structure are required
+            if not prompt:
+                raise ValueError("Prompt is required when starting a new agent session")
+            if not final_response_structure:
+                raise ValueError("final_response_structure is required when starting a new agent session")
+            
+            # Create new agent_id if not provided
+            if agent_id is None:
+                agent_id = str(__import__("uuid").uuid4())
             logger.info(f"🚀 Starting new agent process (ID: {agent_id}) with prompt: {prompt[:100]}...")
         else:
-            logger.info(f"🔄 Continuing agent process (ID: {agent_id}) with prompt: {prompt[:100]}...")
+            # Existing session - use previous final_response_structure if not provided
+            if not final_response_structure:
+                final_response_structure = existing_session.get("final_response_structure")
+                if not final_response_structure:
+                    raise ValueError("final_response_structure not found in existing session and not provided")
+            
+            # Handle optional prompt for existing sessions
+            if not prompt:
+                # Use conversation history to continue without explicit prompt
+                prompt = "[Continue conversation based on history]"
+                logger.info(f"🔄 Continuing agent process (ID: {agent_id}) without explicit prompt (using history)")
+            else:
+                logger.info(f"🔄 Continuing agent process (ID: {agent_id}) with prompt: {prompt[:100]}...")
         
         # Initialize token summary
         token_summary = TokenSummary()
         
-        # Check if this is a continuing conversation
-        existing_session = self._agent_sessions.get(agent_id, {})
-        conversation_history = existing_session.get("conversation_history", []).copy()
-        
         if conversation_history:
             logger.info(f"📜 Agent {agent_id}: Loaded conversation history with {len(conversation_history)} messages")
         
@@ -228,8 +281,11 @@ class AgentClient:
         
         # Handle answers provided (skip question generation and process directly)
         if answers is not None:
-            logger.info(f"📝 Agent {agent_id}: Processing with provided answers, skipping question generation")
-            return self._process_answers_directly(agent_id, prompt, final_response_structure, answers, token_summary, history_for_ai)
+            if self._validate_answers(answers):
+                logger.info(f"📝 Agent {agent_id}: Processing with provided answers, skipping question generation")
+                return self._process_answers_directly(agent_id, prompt, final_response_structure, answers, token_summary, history_for_ai)
+            else:
+                logger.info(f"📝 Agent {agent_id}: Invalid or empty answers provided, proceeding with normal flow")
         
         # Step 1: Generate questions (if ask_questions is True)
         if ask_questions:
@@ -438,6 +494,7 @@ IMPORTANT: For each question, provide a technical key identifier that:
             self._agent_sessions[agent_id]["conversation_history"] = updated_history
             self._agent_sessions[agent_id]["step"] = "completed"
             self._agent_sessions[agent_id]["token_summary"] = token_summary
+            self._agent_sessions[agent_id]["final_response_structure"] = final_response_structure
             logger.info(f"💾 Agent {agent_id}: Updated session with conversation history ({len(updated_history)} messages)")
         else:
             # Create new session if it doesn't exist
@@ -447,7 +504,8 @@ IMPORTANT: For each question, provide a technical key identifier that:
             self._agent_sessions[agent_id] = {
                 "step": "completed",
                 "conversation_history": updated_history,
-                "token_summary": token_summary
+                "token_summary": token_summary,
+                "final_response_structure": final_response_structure
             }
             logger.info(f"💾 Agent {agent_id}: Created new session with conversation history ({len(updated_history)} messages)")
         

mbxai/examples/dialog_agent_example.py

@@ -135,6 +135,21 @@ def demonstrate_dialog_with_questions():
                 print(f"Genre: {book_rec2.genre}")
                 print(f"Reason: {book_rec2.reason}")
                 print(f"Connection to previous: {book_rec2.connection_to_previous}")
+            
+            # Continue WITHOUT explicit prompt - using only conversation history
+            print(f"\n3️⃣ Continuing conversation WITHOUT explicit prompt (history-based):")
+            try:
+                response3 = agent.agent(agent_id=agent_id, ask_questions=False)  # No prompt provided
+                
+                if response3.is_complete():
+                    book_rec3 = response3.final_response
+                    print(f"History-based recommendation:")
+                    print(f"Book: {book_rec3.title} by {book_rec3.author}")
+                    print(f"Genre: {book_rec3.genre}")
+                    print(f"Reason: {book_rec3.reason}")
+                    print(f"Connection to previous: {book_rec3.connection_to_previous}")
+            except Exception as e:
+                print(f"Error with history-based continuation: {e}")
         
         # Session cleanup
         print(f"\n🗑️ Session cleanup...")

mbxai/examples/optional_prompt_example.py

@@ -0,0 +1,164 @@
+"""
+Example demonstrating optional prompt functionality with existing agent sessions.
+"""
+
+import os
+from pydantic import BaseModel, Field
+from mbxai import AgentClient, OpenRouterClient, AnswerList, Answer
+
+
+class ConversationResponse(BaseModel):
+    """A general conversation response."""
+    response: str = Field(description="The response to the conversation")
+    context_used: str = Field(description="How previous conversation history was used")
+    continuation_type: str = Field(description="Type of continuation (new topic, follow-up, etc.)")
+
+
+class StoryResponse(BaseModel):
+    """A story response."""
+    story_continuation: str = Field(description="The next part of the story")
+    character_development: str = Field(description="How characters developed in this part")
+    plot_advancement: str = Field(description="How the plot advanced")
+
+
+def demonstrate_optional_prompt():
+    """Demonstrate using agent with optional prompts for existing sessions."""
+    print("🎭 DEMO: Optional Prompt with Existing Agent Sessions")
+    print("=" * 60)
+    
+    # Initialize the clients
+    openrouter_client = OpenRouterClient(token=os.getenv("OPENROUTER_API_KEY", "your-token-here"))
+    agent = AgentClient(openrouter_client, max_iterations=1)
+    
+    # First interaction - establish a story context
+    print("\n1️⃣ First interaction - establishing story context:")
+    response1 = agent.agent(
+        prompt="Let's create a story about a detective solving a mystery in a small town",
+        final_response_structure=StoryResponse,
+        ask_questions=False
+    )
+    
+    agent_id = response1.agent_id
+    print(f"Agent ID: {agent_id}")
+    
+    if response1.is_complete():
+        story1 = response1.final_response
+        print(f"Story start: {story1.story_continuation[:150]}...")
+        print(f"Characters: {story1.character_development}")
+    
+    # Second interaction - continue with explicit prompt
+    print(f"\n2️⃣ Second interaction - continue with explicit prompt:")
+    response2 = agent.agent(
+        prompt="The detective finds a mysterious letter. What does it say?",
+        agent_id=agent_id,
+        ask_questions=False
+    )
+    
+    if response2.is_complete():
+        story2 = response2.final_response
+        print(f"Story continuation: {story2.story_continuation[:150]}...")
+        print(f"Plot advancement: {story2.plot_advancement}")
+    
+    # Third interaction - continue WITHOUT explicit prompt (using only history)
+    print(f"\n3️⃣ Third interaction - continue WITHOUT explicit prompt (history-based):")
+    try:
+        response3 = agent.agent(
+            agent_id=agent_id,  # Only provide agent_id, no prompt
+            ask_questions=False
+        )
+        
+        if response3.is_complete():
+            story3 = response3.final_response
+            print(f"History-based continuation: {story3.story_continuation[:150]}...")
+            print(f"Character development: {story3.character_development}")
+            print(f"Plot advancement: {story3.plot_advancement}")
+        
+    except Exception as e:
+        print(f"Error with history-based continuation: {e}")
+    
+    # Fourth interaction - switch response format but use same session
+    print(f"\n4️⃣ Fourth interaction - switch to conversation format:")
+    response4 = agent.agent(
+        prompt="What do you think about this story so far?",
+        final_response_structure=ConversationResponse,
+        agent_id=agent_id,
+        ask_questions=False
+    )
+    
+    if response4.is_complete():
+        conv4 = response4.final_response
+        print(f"Analysis: {conv4.response[:150]}...")
+        print(f"Context used: {conv4.context_used}")
+        print(f"Continuation type: {conv4.continuation_type}")
+    
+    # Fifth interaction - continue conversation without prompt
+    print(f"\n5️⃣ Fifth interaction - continue conversation analysis without prompt:")
+    try:
+        response5 = agent.agent(
+            agent_id=agent_id,
+            ask_questions=False
+        )
+        
+        if response5.is_complete():
+            conv5 = response5.final_response
+            print(f"Continued analysis: {conv5.response[:150]}...")
+            print(f"Context used: {conv5.context_used}")
+            print(f"Continuation type: {conv5.continuation_type}")
+        
+    except Exception as e:
+        print(f"Error with continued conversation: {e}")
+    
+    # Show final session state
+    session_info = agent.get_session_info(agent_id)
+    print(f"\n📊 Final session state:")
+    print(f"   - Total messages: {session_info['conversation_length']}")
+    print(f"   - Session step: {session_info.get('step', 'unknown')}")
+    print(f"   - Has final_response_structure: {'final_response_structure' in session_info}")
+    
+    # Cleanup
+    agent.delete_session(agent_id)
+
+
+def demonstrate_error_cases():
+    """Demonstrate error cases with optional prompt."""
+    print("\n🚨 DEMO: Error Cases with Optional Prompt")
+    print("=" * 50)
+    
+    openrouter_client = OpenRouterClient(token=os.getenv("OPENROUTER_API_KEY", "your-token-here"))
+    agent = AgentClient(openrouter_client, max_iterations=1)
+    
+    # Error case 1: No prompt and no existing session
+    print("\n❌ Error case 1: No prompt, no existing session")
+    try:
+        response = agent.agent(ask_questions=False)
+        print("This should have failed!")
+    except ValueError as e:
+        print(f"Expected error: {e}")
+    
+    # Error case 2: No prompt and no final_response_structure for new session
+    print("\n❌ Error case 2: No final_response_structure for new session")
+    try:
+        response = agent.agent(prompt="Test", ask_questions=False)
+        print("This should have failed!")
+    except ValueError as e:
+        print(f"Expected error: {e}")
+    
+    # Error case 3: Unknown agent_id without prompt
+    print("\n❌ Error case 3: Unknown agent_id without prompt")
+    try:
+        response = agent.agent(agent_id="unknown-id", ask_questions=False)
+        print("This should have failed!")
+    except ValueError as e:
+        print(f"Expected error: {e}")
+    
+    print("\n✅ All error cases handled correctly!")
+
+
+if __name__ == "__main__":
+    try:
+        demonstrate_optional_prompt()
+        demonstrate_error_cases()
+    except Exception as e:
+        print(f"Unexpected error: {e}")
+        import traceback
+        traceback.print_exc()

mbxai/mcp/server.py

@@ -31,7 +31,7 @@ class MCPServer:
         self.app = FastAPI(
             title=self.name,
             description=self.description,
-            version="2.1.1",
+            version="2.1.3",
         )
         
         # Initialize MCP server

{mbxai-2.1.1.dist-info → mbxai-2.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mbxai
-Version: 2.1.1
+Version: 2.1.3
 Summary: MBX AI SDK
 Project-URL: Homepage, https://www.mibexx.de
 Project-URL: Documentation, https://www.mibexx.de

{mbxai-2.1.1.dist-info → mbxai-2.1.3.dist-info}/RECORD

@@ -1,7 +1,7 @@
-mbxai/__init__.py,sha256=e4xVVEhVt0ryj9sh9mHtCKINk2GfFdnSuQndPtTOdHw,407
+mbxai/__init__.py,sha256=Jc0jQm0epVrufRDNvp7YQYDWO6TgoF_XyC64qVu4rf8,407
 mbxai/core.py,sha256=WMvmU9TTa7M_m-qWsUew4xH8Ul6xseCZ2iBCXJTW-Bs,196
 mbxai/agent/__init__.py,sha256=5j3mW2NZtAU1s2w8n833axWBQsxW8U0qKwoQ9JtQZ4k,289
-mbxai/agent/client.py,sha256=zVJVa-7xxQqkWcmAwK69RLP0K7mR4_L1El0tQz_-P_k,34729
+mbxai/agent/client.py,sha256=M4n1RKeSi3e59nvJUjDdFR0dgWP-X1pe5j1YzjSV1js,37870
 mbxai/agent/models.py,sha256=sjBtaAENDABHl8IqTON1gxFFSZIaQYUCBFHB5804_Fw,5780
 mbxai/examples/agent_example.py,sha256=7gQHcMVWBu2xdxnVNzz4UfW0lkUnw9a5DN2-YoIRxXE,7420
 mbxai/examples/agent_iterations_example.py,sha256=xMqZhBWS67EkRkArjOAY2fCgLkQ32Qn9E4CSfEKW4MU,7905
@@ -10,8 +10,9 @@ mbxai/examples/agent_tool_registration_example.py,sha256=oWm0-d4mdba-VQ3HobiCIR0
 mbxai/examples/agent_validation_example.py,sha256=xlEf5Mwq5_Iu8bNU4cuHGZVYvAyZNhO2GMFmOom-CLo,4185
 mbxai/examples/auto_schema_example.py,sha256=ymuJJqqDxYznZT2VN6zVFEM7m_lDuccZ1AKSx-xzLTM,8174
 mbxai/examples/conversation_history_test.py,sha256=TpOh5ruQlXDPTPEu_0qTACAaQPSklKp8RYiOm1UzqPI,7773
-mbxai/examples/dialog_agent_example.py,sha256=k502Y_pq6uddWEcH-5i0MxqyakxHKSy-KvHv1s4G1dw,7015
+mbxai/examples/dialog_agent_example.py,sha256=Za4m_JPusn3f60xYE0DTfqGwyz0rXoiCHLP-AFkiQYQ,7884
 mbxai/examples/openrouter_example.py,sha256=-grXHKMmFLoh-yUIEMc31n8Gg1S7uSazBWCIOWxgbyQ,1317
+mbxai/examples/optional_prompt_example.py,sha256=dG9aRKZL_xIZX2OgbnYAJX-4_QGi3op31nn2fRthYKo,6386
 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
@@ -25,7 +26,7 @@ mbxai/examples/mcp/mcp_server_example.py,sha256=nFfg22Jnc6HMW_ezLO3So1xwDdx2_rIt
 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=DN4a7qwoz_YSLpeGhZlez1XIqddYFkn7RC-2VKno3Uc,3332
+mbxai/mcp/server.py,sha256=BPVCF-E-fXngysenjG5k9e4HvLEAxB_49r_LJ_D0etE,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
@@ -35,7 +36,7 @@ mbxai/tools/__init__.py,sha256=ogxrHvgJ7OR62Lmd5x9Eh5d2C0jqWyQis7Zy3yKpZ78,218
 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-2.1.1.dist-info/METADATA,sha256=PlNhE1zB42gvsB3HbLJTSVgA0JKmjhGtdX_ThlCD1Hk,10018
-mbxai-2.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mbxai-2.1.1.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
-mbxai-2.1.1.dist-info/RECORD,,
+mbxai-2.1.3.dist-info/METADATA,sha256=21Rj50qEl1nFc_kTjY2d5PeE1WQsaLg2KtDRqMEMnCU,10018
+mbxai-2.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mbxai-2.1.3.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
+mbxai-2.1.3.dist-info/RECORD,,