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

Fixing tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-08 17:56:17 +01:00
parent b32aa929dc
commit 48d1443fe9
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/integration/test_agent_structured_query_integration.py
View file

@ -60,7 +60,7 @@ class TestAgentStructuredQueryIntegration:
request = AgentRequest(
question="I need to find all customers from New York. Use the structured query tool to get this information.",
state="",
group=[],
group=None,
history=[],
user="test_user"
)
@ -144,7 +144,7 @@ Args: {
request = AgentRequest(
question="Find data from a table that doesn't exist using structured query.",
state="",
group=[],
group=None,
history=[],
user="test_user"
)
@ -216,7 +216,7 @@ Args: {
request = AgentRequest(
question="First find all customers from California, then tell me how many orders they have made.",
state="",
group=[],
group=None,
history=[],
user="test_user"
)
@ -305,7 +305,7 @@ Args: {
request = AgentRequest(
question="Query the sales data for recent transactions.",
state="",
group=[],
group=None,
history=[],
user="test_user"
)
@ -403,7 +403,7 @@ Args: {
request = AgentRequest(
question="Get customer information and format it nicely.",
state="",
group=[],
group=None,
history=[],
user="test_user"
)

2
trustgraph-flow/trustgraph/agent/tool_filter.py
View file

@ -29,7 +29,7 @@ def filter_tools_by_group_and_state(
"""
# Apply defaults as specified in tech spec
if requested_groups is None or requested_groups == []:
if requested_groups is None:
requested_groups = ["default"]
if current_state is None or current_state == "":
current_state = "undefined"