trustgraph-cli 1.3.15__tar.gz → 1.3.16__tar.gz

{trustgraph_cli-1.3.15 → trustgraph_cli-1.3.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 1.3.15
+Version: 1.3.16
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

{trustgraph_cli-1.3.15 → trustgraph_cli-1.3.16}/trustgraph/cli/invoke_agent.py

@@ -29,7 +29,7 @@ def output(text, prefix="> ", width=78):
 
 async def question(
         url, question, flow_id, user, collection,
-        plan=None, state=None, verbose=False
+        plan=None, state=None, group=None, verbose=False
 ):
 
     if not url.endswith("/"):
@@ -55,15 +55,24 @@ async def question(
 
     async with connect(url) as ws:
 
-        req = json.dumps({
+        req = {
             "id": mid,
             "service": "agent",
             "flow": flow_id,
             "request": {
                 "question": question,
+                "user": user,
+                "history": []
             }
+        }
+        
+        # Only add optional fields if they have values
+        if state is not None:
+            req["request"]["state"] = state
+        if group is not None:
+            req["request"]["group"] = group
             
-        })
+        req = json.dumps(req)
 
         await ws.send(req)
 
@@ -140,6 +149,12 @@ def main():
         help=f'Agent initial state (default: unspecified)'
     )
 
+    parser.add_argument(
+        '-g', '--group',
+        nargs='+',
+        help='Agent tool groups (can specify multiple)'
+    )
+
     parser.add_argument(
         '-v', '--verbose',
         action="store_true",
@@ -159,6 +174,7 @@ def main():
                 collection = args.collection,
                 plan = args.plan,
                 state = args.state,
+                group = args.group,
                 verbose = args.verbose,
             )
         )

{trustgraph_cli-1.3.15 → trustgraph_cli-1.3.16}/trustgraph/cli/invoke_structured_query.py

@@ -79,11 +79,11 @@ def format_table_data(rows, table_name, output_format):
     else:
         return json.dumps({table_name: rows}, indent=2)
 
-def structured_query(url, flow_id, question, output_format='table'):
+def structured_query(url, flow_id, question, user='trustgraph', collection='default', output_format='table'):
 
     api = Api(url).flow().id(flow_id)
 
-    resp = api.structured_query(question=question)
+    resp = api.structured_query(question=question, user=user, collection=collection)
 
     # Check for errors
     if "error" in resp and resp["error"]:
@@ -132,6 +132,18 @@ def main():
         help='Natural language question to execute',
     )
 
+    parser.add_argument(
+        '--user',
+        default='trustgraph',
+        help='Cassandra keyspace identifier (default: trustgraph)'
+    )
+
+    parser.add_argument(
+        '--collection',
+        default='default',
+        help='Data collection identifier (default: default)'
+    )
+
     parser.add_argument(
         '--format', 
         choices=['table', 'json', 'csv'],
@@ -147,6 +159,8 @@ def main():
             url=args.url,
             flow_id=args.flow_id,
             question=args.question,
+            user=args.user,
+            collection=args.collection,
             output_format=args.format,
         )
 

{trustgraph_cli-1.3.15 → trustgraph_cli-1.3.16}/trustgraph/cli/set_tool.py

@@ -97,11 +97,11 @@ def set_tool(
             for a in arguments
         ]
 
-    if group: object["group"] = group
+    if group is not None: object["group"] = group
 
     if state: object["state"] = state
 
-    if applicable_states: object["applicable-states"] = applicable_states
+    if applicable_states is not None: object["applicable-states"] = applicable_states
 
     values = api.put([
         ConfigValue(
@@ -254,9 +254,9 @@ def main():
             collection=args.collection,
             template=args.template,
             arguments=arguments,
-            group=args.group or [],
+            group=args.group,
             state=args.state,
-            applicable_states=getattr(args, 'applicable_states', None) or [],
+            applicable_states=args.applicable_states,
         )
 
     except Exception as e:

{trustgraph_cli-1.3.15 → trustgraph_cli-1.3.16}/trustgraph/cli/show_tools.py

@@ -52,6 +52,29 @@ def show_config(url):
                     f"arg {n}",
                     f"{arg['name']}: {arg['type']}\n{arg['description']}"
                 ))
+        
+        # Display group information
+        if "group" in data:
+            groups = data["group"]
+            if groups:
+                table.append(("groups", ", ".join(groups)))
+            else:
+                table.append(("groups", "(empty - no groups)"))
+        
+        # Display state transition information
+        if "state" in data:
+            table.append(("next state", data["state"]))
+        
+        # Display applicable states
+        if "applicable-states" in data:
+            states = data["applicable-states"]
+            if states:
+                if "*" in states:
+                    table.append(("available in", "all states"))
+                else:
+                    table.append(("available in", ", ".join(states)))
+            else:
+                table.append(("available in", "(empty - never available)"))
 
         print()
 

trustgraph_cli-1.3.16/trustgraph/cli_version.py

@@ -0,0 +1 @@
+__version__ = "1.3.16"

{trustgraph_cli-1.3.15 → trustgraph_cli-1.3.16}/trustgraph_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 1.3.15
+Version: 1.3.16
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_cli-1.3.15/trustgraph/cli_version.py

@@ -1 +0,0 @@
-__version__ = "1.3.15"