Build a Quickstart Generator with Connectors and Web Search

%pip install mistralai --quiet

PROMPT = (
    "Write a Polars quickstart for a developer who knows pandas. "
    "Cover: installation, reading a CSV, filtering rows, groupby aggregation, "
    "and lazy evaluation. End with 3 gotchas when migrating from pandas. "
    "Include runnable code examples."
)

print("--- Step 1: Baseline (no tools) ---\n")
response = await client.beta.conversations.start_async(
    model="mistral-medium-latest",
    inputs=[{"role": "user", "content": PROMPT}],
)
for output in response.outputs:
    if getattr(output, "type", None) == "message.output":
        content = output.content
        if isinstance(content, str):
            print(content)
        else:
            print("".join(getattr(c, "text", str(c)) for c in content))

print("--- Step 3: Context7 connector (official docs) ---\n")

conversation_id = None
async for event in await client.beta.conversations.start_stream_async(
    agent_id=connector_agent.id,
    inputs=[{"role": "user", "content": PROMPT}],
    timeout_ms=300_000,
):
    data = event.data
    event_type = getattr(data, "type", None)
    if event_type == "conversation.response.started":
        conversation_id = data.conversation_id
    elif event_type == "message.output.delta":
        print(".", end="", flush=True)
    else:
        name = getattr(data, "name", "")
        print(f"\n[{event_type}]{' ' + name if name else ''}")

messages = await client.beta.conversations.get_messages_async(
    conversation_id=conversation_id
)
last_output = next(
    (m for m in reversed(messages.messages)
     if getattr(m, "type", None) == "message.output"),
    None,
)
if last_output:
    print("\n")
    content = last_output.content
    if isinstance(content, str):
        print(content)
    else:
        print("".join(getattr(c, "text", str(c)) for c in content))

print("--- Step 4: Web search + Context7 connector ---\n")

combined_agent = await client.beta.agents.update_async(
    agent_id=connector_agent.id,
    tools=[
        {"type": "web_search"},
        {"type": "connector", "connector_id": connector.id},
    ],
)

conversation_id = None
async for event in await client.beta.conversations.start_stream_async(
    agent_id=combined_agent.id,
    inputs=[{"role": "user", "content": PROMPT}],
    timeout_ms=300_000,
):
    data = event.data
    event_type = getattr(data, "type", None)
    if event_type == "conversation.response.started":
        conversation_id = data.conversation_id
    elif event_type == "message.output.delta":
        print(".", end="", flush=True)
    else:
        name = getattr(data, "name", "")
        print(f"\n[{event_type}]{' ' + name if name else ''}")

messages = await client.beta.conversations.get_messages_async(
    conversation_id=conversation_id
)
last_output = next(
    (m for m in reversed(messages.messages)
     if getattr(m, "type", None) == "message.output"),
    None,
)
if last_output:
    print("\n")
    content = last_output.content
    if isinstance(content, str):
        print(content)
    else:
        print("".join(getattr(c, "text", str(c)) for c in content))

FILTERED_PROMPT = (
    "Using the Polars documentation, explain lazy evaluation in Polars. "
    "Cover: what LazyFrame is, how to build a lazy query with .lazy(), "
    "how .collect() triggers execution, and when to prefer lazy over eager. "
    "Include a runnable before/after code example."
)

if DOC_TOOL_NAME:
    print(f"--- Step 5: Filtered connector (only {DOC_TOOL_NAME}) ---\n")

    filtered_agent = await client.beta.agents.update_async(
        agent_id=combined_agent.id,
        tools=[
            {
                "type": "connector",
                "connector_id": connector.id,
                "tool_configuration": {
                    "include": [DOC_TOOL_NAME],
                },
            },
        ],
    )

    conversation_id = None
    async for event in await client.beta.conversations.start_stream_async(
        agent_id=filtered_agent.id,
        inputs=[{"role": "user", "content": FILTERED_PROMPT}],
        timeout_ms=300_000,
    ):
        data = event.data
        event_type = getattr(data, "type", None)
        if event_type == "conversation.response.started":
            conversation_id = data.conversation_id
        elif event_type == "message.output.delta":
            print(".", end="", flush=True)
        else:
            name = getattr(data, "name", "")
            print(f"\n[{event_type}]{' ' + name if name else ''}")

    messages = await client.beta.conversations.get_messages_async(
        conversation_id=conversation_id
    )
    last_output = next(
        (m for m in reversed(messages.messages)
         if getattr(m, "type", None) == "message.output"),
        None,
    )
    if last_output:
        print("\n")
        content = last_output.content
        if isinstance(content, str):
            print(content)
        else:
            print("".join(getattr(c, "text", str(c)) for c in content))
else:
    print("Skipped — DOC_TOOL_NAME was not set. Set it manually from the tool list in Step 3.")

await client.beta.agents.delete_async(agent_id=connector_agent.id)
print(f"Agent deleted: {connector_agent.name}")

result = await client.beta.connectors.delete_async(connector_id=connector.id)
print(f"Connector deleted: {connector.name}  —  {result.message}")