Add Flow ID to all relevant CLIs, not completely implemented

trustgraph-cli/scripts/tg-graph-show

@@ -37,6 +37,12 @@ def main():
         help=f'API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     parser.add_argument(
         '-U', '--user',
         default=default_user,

trustgraph-cli/scripts/tg-graph-to-turtle

@@ -69,6 +69,12 @@ def main():
         help=f'API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     parser.add_argument(
         '-U', '--user',
         default=default_user,

trustgraph-cli/scripts/tg-invoke-agent

@@ -104,6 +104,12 @@ def main():
         help=f'API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     parser.add_argument(
         '-q', '--question',
         required=True,

trustgraph-cli/scripts/tg-invoke-document-rag

@@ -37,6 +37,12 @@ def main():
         help=f'API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     # parser.add_argument(
     #     '--pulsar-api-key',
     #     default=default_pulsar_api_key,

trustgraph-cli/scripts/tg-invoke-graph-rag

@@ -45,6 +45,12 @@ def main():
         help=f'API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     parser.add_argument(
         '-q', '--question',
         required=True,

trustgraph-cli/scripts/tg-invoke-llm

@@ -12,11 +12,11 @@ from trustgraph.api import Api
 
 default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
 
-def query(url, system, prompt):
+def query(url, flow, system, prompt):
 
     api = Api(url)
 
-    resp = api.text_completion(system=system, prompt=prompt)
+    resp = api.flow(flow).text_completion(system=system, prompt=prompt)
 
     print(resp)
 
@@ -45,11 +45,11 @@ def main():
         help='LLM prompt e.g. What is 2 + 2?',
     )
 
-    # parser.add_argument(
+    parser.add_argument(
-    #     '--pulsar-api-key',
+        '-f', '--flow-id',
-    #     default=default_pulsar_api_key,
+        default="0000",
-    #     help=f'Pulsar API key',
+        help=f'Flow ID (default: 0000)'
-    # )
+    )
 
     args = parser.parse_args()
 
@@ -57,6 +57,7 @@ def main():
 
         query(
             url=args.url,
+            flow = args.flow_id,
             system=args.system[0],
             prompt=args.prompt[0],
         )

trustgraph-cli/scripts/tg-invoke-prompt

@@ -40,6 +40,12 @@ def main():
         help=f'API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     parser.add_argument(
         'id',
         metavar='template-id',

trustgraph-cli/scripts/tg-load-doc-embeds

@@ -184,6 +184,12 @@ async def main(running):
         help=f'TrustGraph API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     parser.add_argument(
         '-i', '--input-file',
         # Make it mandatory, difficult to over-write an existing file

trustgraph-cli/scripts/tg-load-pdf

@@ -82,6 +82,12 @@ def main():
         help=f'API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     parser.add_argument(
         '-U', '--user',
         default=default_user,

trustgraph-cli/scripts/tg-load-text

@@ -81,11 +81,11 @@ def main():
         help=f'API URL (default: {default_url})',
     )
 
-    # parser.add_argument(
+    parser.add_argument(
-    #     '--pulsar-api-key',
+        '-f', '--flow-id',
-    #     default=default_pulsar_api_key,
+        default="0000",
-    #     help=f'Pulsar API key',
+        help=f'Flow ID (default: 0000)'
-    # )
+    )
 
     parser.add_argument(
         '-U', '--user',

trustgraph-cli/scripts/tg-load-turtle

@@ -110,6 +110,12 @@ def main():
         help=f'Pulsar host (default: {default_pulsar_host})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+    
     parser.add_argument(
         '--pulsar-api-key',
         default=default_pulsar_api_key,

trustgraph-cli/scripts/tg-save-doc-embeds

@@ -158,6 +158,12 @@ async def main(running):
         help=f'TrustGraph API URL (default: {default_url})',
     )
 
+    parser.add_argument(
+        '-f', '--flow-id',
+        default="0000",
+        help=f'Flow ID (default: 0000)'
+    )
+
     parser.add_argument(
         '-o', '--output-file',
         # Make it mandatory, difficult to over-write an existing file