trustgraph-cli 2.8.13__tar.gz → 2.8.14__tar.gz

{trustgraph_cli-2.8.13 → trustgraph_cli-2.8.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.8.13
+Version: 2.8.14
 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-2.8.13 → trustgraph_cli-2.8.14}/trustgraph/cli/set_prompt.py

@@ -4,7 +4,7 @@ Sets a prompt template.
 
 import argparse
 import os
-from trustgraph.api import Api, ConfigKey, ConfigValue
+from trustgraph.api import Api, ConfigValue
 import json
 import tabulate
 import textwrap
@@ -27,12 +27,6 @@ def set_prompt(url, id, prompt, response, schema, token=None, workspace="default
 
     api = Api(url, token=token, workspace=workspace).config()
 
-    values = api.get([
-        ConfigKey(type="prompt", key="template-index")
-    ])
-
-    ix = json.loads(values[0].value)
-
     object = {
         "id": id,
         "prompt": prompt,
@@ -46,13 +40,7 @@ def set_prompt(url, id, prompt, response, schema, token=None, workspace="default
     if schema:
         object["schema"] = schema
 
-    if id not in ix:
-        ix.append(id)
-
-    values = api.put([
-        ConfigValue(
-            type="prompt", key="template-index", value=json.dumps(ix)
-        ),
+    api.put([
         ConfigValue(
             type="prompt", key=f"template.{id}", value=json.dumps(object)
         )

{trustgraph_cli-2.8.13 → trustgraph_cli-2.8.14}/trustgraph/cli/set_token_costs.py

@@ -4,7 +4,7 @@ Sets a model's token costs.
 
 import argparse
 import os
-from trustgraph.api import Api, ConfigKey, ConfigValue
+from trustgraph.api import Api, ConfigValue
 import json
 import tabulate
 import textwrap
@@ -31,12 +31,6 @@ def set_prompt(url, id, prompt, response, schema, workspace="default"):
 
     api = Api(url, workspace=workspace)
 
-    values = api.config_get([
-        ConfigKey(type="prompt", key="template-index")
-    ])
-
-    ix = json.loads(values[0].value)
-
     object = {
         "id": id,
         "prompt": prompt,
@@ -50,13 +44,7 @@ def set_prompt(url, id, prompt, response, schema, workspace="default"):
     if schema:
         object["schema"] = schema
 
-    if id not in ix:
-        ix.append(id)
-
-    values = api.config_put([
-        ConfigValue(
-            type="prompt", key="template-index", value=json.dumps(ix)
-        ),
+    api.config_put([
         ConfigValue(
             type="prompt", key=f"template.{id}", value=json.dumps(object)
         )

{trustgraph_cli-2.8.13 → trustgraph_cli-2.8.14}/trustgraph/cli/show_prompts.py

@@ -17,17 +17,21 @@ def show_config(url, token=None, workspace="default"):
 
     api = Api(url, token=token, workspace=workspace).config()
 
-    values = api.get([
+    system_values = api.get([
         ConfigKey(type="prompt", key="system"),
-        ConfigKey(type="prompt", key="template-index")
     ])
 
-    system = json.loads(values[0].value)
-    ix = json.loads(values[1].value)
+    system = json.loads(system_values[0].value)
+
+    all_keys = api.list(type="prompt")
+    ix = sorted([
+        k for k in all_keys
+        if k.startswith("template.")
+    ])
 
     values = api.get([
-        ConfigKey(type="prompt", key=f"template.{v}")
-        for v in ix
+        ConfigKey(type="prompt", key=k)
+        for k in ix
     ])
 
     print()
@@ -43,6 +47,7 @@ def show_config(url, token=None, workspace="default"):
 
     for n, key in enumerate(ix):
 
+        key = key.removeprefix("template.")
         data = json.loads(values[n].value)
 
         table = []

{trustgraph_cli-2.8.13 → trustgraph_cli-2.8.14}/trustgraph/cli/verify_system_status.py

@@ -231,18 +231,10 @@ def check_prompts(url: str, timeout: int, tr, token: Optional[str] = None, works
         api = Api(url, token=token, timeout=timeout, workspace=workspace)
         config = api.config()
 
-        # Import ConfigKey here to avoid top-level import issues
-        from trustgraph.api.types import ConfigKey
-        import json
+        all_keys = config.list(type="prompt")
+        ix = [k for k in all_keys if k.startswith("template.")]
 
-        # Get the template-index which lists all prompts
-        values = config.get([
-            ConfigKey(type="prompt", key="template-index")
-        ])
-
-        ix = json.loads(values[0].value)
-
-        if ix and len(ix) > 0:
+        if ix:
             return True, tr.t("cli.verify_system_status.prompts.found", count=len(ix))
         else:
             return False, tr.t("cli.verify_system_status.prompts.none")

trustgraph_cli-2.8.14/trustgraph/cli_version.py

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

{trustgraph_cli-2.8.13 → trustgraph_cli-2.8.14}/trustgraph_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-cli
-Version: 2.8.13
+Version: 2.8.14
 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-2.8.13/trustgraph/cli_version.py

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