Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
62 changes: 62 additions & 0 deletions examples/hosted_mcp/connectors.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
import argparse
import asyncio
import json
import os
from datetime import datetime

from agents import Agent, HostedMCPTool, Runner

# import logging
# logging.basicConfig(level=logging.DEBUG)


async def main(verbose: bool, stream: bool):
# 1. Visit https://developers.google.com/oauthplayground/
# 2. Input https://www.googleapis.com/auth/calendar.events as the required scope
# 3. Grab the acccess token starting with "ya29."
authorization = os.environ["GOOGLE_CALENDAR_AUTHORIZATION"]
agent = Agent(
name="Assistant",
instructions="You are a helpful assistant that can help a user with their calendar.",
tools=[
HostedMCPTool(
tool_config={
"type": "mcp",
"server_label": "google_calendar",
# see https://platform.openai.com/docs/guides/tools-connectors-mcp#connectors
"connector_id": "connector_googlecalendar",
"authorization": authorization,
"require_approval": "never",
}
)
],
)

today = datetime.now().strftime("%Y-%m-%d")
if stream:
result = Runner.run_streamed(agent, f"What is my schedule for {today}?")
async for event in result.stream_events():
if event.type == "raw_response_event":
if event.data.type.startswith("response.output_item"):
print(json.dumps(event.data.to_dict(), indent=2))
if event.data.type.startswith("response.mcp"):
print(json.dumps(event.data.to_dict(), indent=2))
if event.data.type == "response.output_text.delta":
print(event.data.delta, end="", flush=True)
print()
else:
res = await Runner.run(agent, f"What is my schedule for {today}?")
print(res.final_output)

if verbose:
for item in res.new_items:
Comment on lines +51 to +52

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

[P1] Use streamed result instead of undefined res when verbose

When stream=True the code never assigns to res, yet the verbose block always references res.new_items. Calling the example with --stream --verbose (or main(True, True)) will raise UnboundLocalError: cannot access local variable 'res' after the streaming loop completes, so verbose streaming currently cannot work.

Useful? React with 👍 / 👎.

print(item)


if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--verbose", action="store_true", default=False)
parser.add_argument("--stream", action="store_true", default=False)
args = parser.parse_args()

asyncio.run(main(args.verbose, args.stream))