Progress Form Example

from typing import List, Optional
from pydantic import BaseModel, Field
from pydantic2.agents.progress_form import BaseProgressForm


class StartupForm(BaseModel):
    """Structure for storing startup form data"""
    idea_desc: str = Field(default="", description="Description of startup idea")
    target_mkt: str = Field(default="", description="Target market info")
    biz_model: str = Field(default="", description="Business model info")
    team_info: str = Field(default="", description="Team background")


class StartupFormResponse(BaseModel):
    """Response format for startup form analysis"""
    feedback: str = Field(description="Detailed feedback on the startup idea")
    score: float = Field(ge=0, le=10, description="Overall score of the startup idea")
    strengths: List[str] = Field(description="Key strengths of the startup idea")
    weaknesses: List[str] = Field(description="Areas for improvement")
    next_steps: List[str] = Field(description="Recommended next steps")
    market_potential: Optional[float] = Field(
        ge=0, le=10,
        description="Market potential score"
    )


class StartupFormProcessor(BaseProgressForm):
    """Processor for startup form data"""

    def __init__(self, user_id: str):
        super().__init__(
            user_id=user_id,
            client_id="startup_form",
            form_class=StartupForm,
            verbose=False,
            verbose_clients=False,
            form_prompt="""
                Ask in Russian.
            """,
        )

        # Register tools
        self.tools = [self.analyze_startup]

        # Configure test agent
        self.configure_test_agent(
            prompt="""
            You are a naive startup founder who is asking for help to make a startup.
            Talk like a stupid.
            """,
            client=self._get_test_agent_client(temperature=0.7)
        )

    def analyze_startup(
        self,
        message: str,
    ) -> StartupFormResponse:
        """Analyze complete startup info when form is complete"""
        client = self._get_tool_client(temperature=0.7)

        client.message_handler.add_message_system(
            """
            You are a startup analyst. Generate a comprehensive analysis:
            1. Evaluate overall viability (0-10)
            2. List 3-5 key strengths
            3. List 2-3 areas for improvement
            4. Suggest 2-3 specific next steps
            5. Score market potential (0-10)
            """
        )

        client.message_handler.add_message_block(
            "STARTUP_INFO",
            self.current_state.form.model_dump()
        )

        try:
            result: StartupFormResponse = client.generate(
                result_type=StartupFormResponse
            )

            print("\n")
            print("="*50)
            print("🎉 STARTUP ANALYSIS SUCCESS 🎉")
            print("="*50)
            print(result.model_dump_json(indent=2))
            print("="*50)

            return result
        except Exception as e:
            raise Exception(f"Error analyzing startup: {str(e)}")


def main():
    """Example usage of StartupFormProcessor"""
    processor = StartupFormProcessor(user_id="test_user")
    processor.run_test_dialog()


if __name__ == "__main__":
    main()

from pydantic2 import PydanticAIClient, ModelSettings

# Create custom client
custom_client = PydanticAIClient(
    model_name="openai/gpt-4o-mini-2024-07-18",
    client_id="custom_startup_analyzer",
    user_id="test_user",
    verbose=False,
    retries=3,
    online=True,
    max_budget=1,
    model_settings=ModelSettings(
        max_tokens=1000,
        temperature=0.7,
        top_p=1,
        frequency_penalty=0,
    )
)

# Use custom client in your processor
class CustomStartupFormProcessor(BaseProgressForm):
    def __init__(self, user_id: str):
        super().__init__(
            user_id=user_id,
            client_id="startup_form",
            form_class=StartupForm,
            verbose=True,
            client_agent=custom_client  # Pass custom client
        )

        # Configure test agent with custom client
        self.configure_test_agent(
            prompt="Your prompt here...",
            client=custom_client
        )

    def analyze_startup(self, message: str) -> StartupFormResponse:
        # Use custom client for analysis
        result = custom_client.generate(
            result_type=StartupFormResponse,
            messages=[
                {
                    "role": "system",
                    "content": "You are a startup analyst..."
                },
                {
                    "role": "user",
                    "content": "Analyze this startup data..."
                }
            ]
        )
        return result

client = PydanticAIClient(
    model_name="openai/gpt-4o-mini-2024-07-18",  # Choose model
    model_settings=ModelSettings(
        max_tokens=2000,  # Longer responses
        temperature=0.9,  # More creative
    )
)