[Upstream PR 1600] Fix PR 1585 supporting connectors

examples/hosted_mcp/connectors.py

@@ -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:
+            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))