mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Eduard van Valkenburg
a7d924a7d2
* changed AIFunction to FunctionTool and @ai_function to @tool * test and mypy fixes * mypy fix * switch function tool to always_require * fix noop * fix github copilot imports * test fixes * fix ollama test * fixes for tests * fix tests * reverted change to always_require and extended timeout * fix test
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import asyncio
|
|
import logging
|
|
|
|
from agent_framework import ChatAgent, tool
|
|
|
|
from agent_framework_bedrock import BedrockChatClient
|
|
|
|
|
|
@tool(approval_mode="never_require")
|
|
def get_weather(city: str) -> dict[str, str]:
|
|
"""Return a mock forecast for the requested city."""
|
|
normalized = city.strip() or "New York"
|
|
return {"city": normalized, "forecast": "72F and sunny"}
|
|
|
|
|
|
async def main() -> None:
|
|
"""Run the Bedrock sample agent, invoke the weather tool, and log the response."""
|
|
agent = ChatAgent(
|
|
chat_client=BedrockChatClient(),
|
|
instructions="You are a concise travel assistant.",
|
|
name="BedrockWeatherAgent",
|
|
tool_choice="auto",
|
|
tools=[get_weather],
|
|
)
|
|
|
|
response = await agent.run("Use the weather tool to check the forecast for new york.")
|
|
logging.info("\nAssistant reply:", response.text or "<no text returned>")
|
|
logging.info("\nConversation transcript:")
|
|
for message in response.messages:
|
|
for idx, content in enumerate(message.contents, start=1):
|
|
match content.type:
|
|
case "text":
|
|
logging.info(f" {idx}. text -> {content.text}")
|
|
case "function_call":
|
|
logging.info(f" {idx}. function_call ({content.name}) -> {content.arguments}")
|
|
case "function_result":
|
|
logging.info(f" {idx}. function_result ({content.call_id}) -> {content.result}")
|
|
case _:
|
|
logging.info(f" {idx}. {content.type}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|