apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 04:01:02 +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( "knowledge_query": Tool(
name="knowledge_query", name="knowledge_query",
description="Query the knowledge graph for information", description="Query the knowledge graph for information",
arguments={ arguments=[
"question": Argument( Argument(
name="question", name="question",
type="string", type="string",
description="The question to ask the knowledge graph" description="The question to ask the knowledge graph"
) )
}, ],
implementation=KnowledgeQueryImpl, implementation=KnowledgeQueryImpl,
config={} config={}
), ),
"text_completion": Tool( "text_completion": Tool(
name="text_completion", name="text_completion",
description="Generate text completion using LLM", description="Generate text completion using LLM",
arguments={ arguments=[
"question": Argument( Argument(
name="question", name="question",
type="string", type="string",
description="The question to ask the LLM" description="The question to ask the LLM"
) )
}, ],
implementation=TextCompletionImpl, implementation=TextCompletionImpl,
config={} config={}
), ),
"web_search": Tool( "web_search": Tool(
name="web_search", name="web_search",
description="Search the web for information", description="Search the web for information",
arguments={ arguments=[
"query": Argument( Argument(
name="query", name="query",
type="string", type="string",
description="The search query" description="The search query"
) )
}, ],
implementation=lambda context: AsyncMock(invoke=AsyncMock(return_value="Web search results")), implementation=lambda context: AsyncMock(invoke=AsyncMock(return_value="Web search results")),
config={} config={}
) )