mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-05-17 03:15:14 +02:00
Updated CLI invocation and config model for tools and mcp (#438)
* Updated CLI invocation and config model for tools and mcp * CLI anomalies * Tweaked the MCP tool implementation for new model * Update agent implementation to match the new model * Fix agent tools, now all tested * Fixed integration tests * Fix MCP delete tool params
This commit is contained in:
parent
a96d02da5d
commit
81c7c1181b
11 changed files with 270 additions and 183 deletions
|
|
@ -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={}
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue