mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Dmytro Struk
f087b864fb
* Fixed ollama_chat_client sample * Fixed ollama_chat_multimodal sample * Fixed function_tool_with_approval_and_sessions sample * Updated function_tool_with_session_injection sample * Small clean-up * Update 01_round_robin_group_chat.py * Update 02_selector_group_chat.py * Update 03_swarm.py * Update 03_assistant_agent_thread_and_stream.py * Update 04_agent_as_tool.py * Resolved comments
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import asyncio
|
|
from datetime import datetime
|
|
|
|
from agent_framework import Message, tool
|
|
from agent_framework.ollama import OllamaChatClient
|
|
|
|
"""
|
|
Ollama Chat Client Example
|
|
|
|
This sample demonstrates using the native Ollama Chat Client directly.
|
|
|
|
Ensure to install Ollama and have a model running locally before running the sample.
|
|
Not all Models support function calling, to test function calling try llama3.2
|
|
Set the model to use via the OLLAMA_MODEL_ID environment variable or modify the code below.
|
|
https://ollama.com/
|
|
|
|
"""
|
|
|
|
|
|
# NOTE: approval_mode="never_require" is for sample brevity. Use "always_require" in production; see samples/02-agents/tools/function_tool_with_approval.py and samples/02-agents/tools/function_tool_with_approval_and_sessions.py.
|
|
@tool(approval_mode="never_require")
|
|
def get_time():
|
|
"""Get the current time."""
|
|
return f"The current time is {datetime.now().strftime('%I:%M %p')}."
|
|
|
|
|
|
async def main() -> None:
|
|
client = OllamaChatClient()
|
|
message = "What time is it? Use a tool call"
|
|
messages = [Message(role="user", text=message)]
|
|
stream = False
|
|
print(f"User: {message}")
|
|
if stream:
|
|
print("Assistant: ", end="")
|
|
async for chunk in client.get_response(messages, tools=get_time, stream=True):
|
|
if str(chunk):
|
|
print(str(chunk), end="")
|
|
print("")
|
|
else:
|
|
response = await client.get_response(messages, tools=get_time)
|
|
print(f"Assistant: {response}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|