mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Evan Mattson
8cf8b0f995
* Refactor ag-ui to clean up some patterns * Mypy fixes * Fix imports, typing, tests, logging. * Fix test import error * Fix imports again * Fix thread handling
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
from types import SimpleNamespace
|
|
|
|
from agent_framework_ag_ui._orchestration._tooling import merge_tools, register_additional_client_tools
|
|
|
|
|
|
class DummyTool:
|
|
def __init__(self, name: str) -> None:
|
|
self.name = name
|
|
self.declaration_only = True
|
|
|
|
|
|
def test_merge_tools_filters_duplicates() -> None:
|
|
server = [DummyTool("a"), DummyTool("b")]
|
|
client = [DummyTool("b"), DummyTool("c")]
|
|
|
|
merged = merge_tools(server, client)
|
|
|
|
assert merged is not None
|
|
names = [getattr(t, "name", None) for t in merged]
|
|
assert names == ["a", "b", "c"]
|
|
|
|
|
|
def test_register_additional_client_tools_assigns_when_configured() -> None:
|
|
class Fic:
|
|
def __init__(self) -> None:
|
|
self.additional_tools = None
|
|
|
|
holder = SimpleNamespace(function_invocation_configuration=Fic())
|
|
agent = SimpleNamespace(chat_client=holder)
|
|
|
|
tools = [DummyTool("x")]
|
|
register_additional_client_tools(agent, tools)
|
|
|
|
assert holder.function_invocation_configuration.additional_tools == tools
|