apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-21 19:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Fixed integration tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-07-16 23:01:02 +01:00
parent d5bfef33ca
commit a32875647b
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/integration/test_agent_manager_integration.py
View file

@ -69,39 +69,39 @@ class TestAgentManagerIntegration:
"knowledge_query": Tool(
name="knowledge_query",
description="Query the knowledge graph for information",
arguments={
"question": Argument(
arguments=[
Argument(
name="question",
type="string",
description="The question to ask the knowledge graph"
)
},
],
implementation=KnowledgeQueryImpl,
config={}
),
"text_completion": Tool(
name="text_completion",
description="Generate text completion using LLM",
arguments={
"question": Argument(
arguments=[
Argument(
name="question",
type="string",
description="The question to ask the LLM"
)
},
],
implementation=TextCompletionImpl,
config={}
),
"web_search": Tool(
name="web_search",
description="Search the web for information",
arguments={
"query": Argument(
arguments=[
Argument(
name="query",
type="string",
description="The search query"
)
},
],
implementation=lambda context: AsyncMock(invoke=AsyncMock(return_value="Web search results")),
config={}
)