from pydantic import BaseModel, Field

from camel.agents import ChatAgent
from camel.models import ModelFactory
from camel.toolkits import MathToolkit
from camel.types import ModelPlatformType, ModelType


class Result(BaseModel):
    sum_result: str = Field(description="Result of the addition")
    product_result: str = Field(description="Result of the multiplication")
    division_result: str = Field(description="Result of the division")
    capital_result: str = Field(description="Result of the capital search")

streaming_model = ModelFactory.create(
    model_platform=ModelPlatformType.DEFAULT,
    model_type=ModelType.DEFAULT,
    model_config_dict={
        "stream": True,
        "stream_options": {"include_usage": True},
    },
)

agent = ChatAgent(
    system_message="You are a helpful assistant. ",
    model=streaming_model,
    tools=MathToolkit().get_tools(),
    stream_accumulate=False,  # Delta mode
)

streaming_response = agent.step(
    "Calculate: 1) 123.45 + 678.90  2) 100 * 3.14159  3) 1000 / 7, "
    "also search what is the capital of Germany",
    response_format=Result,
)

# Collect content from streaming chunks
content_parts = []
for chunk in streaming_response:
    if chunk.msgs[0].content:
        content_parts.append(chunk.msgs[0].content)

# Verify we got some structured response
full_content = "".join(content_parts)

print(full_content)