trustgraph-flow 0.15.4__tar.gz → 0.15.6__tar.gz

{trustgraph-flow-0.15.4 → trustgraph-flow-0.15.6}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.1
 Name: trustgraph-flow
-Version: 0.15.4
+Version: 0.15.6
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Home-page: https://github.com/trustgraph-ai/trustgraph
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 License: UNKNOWN
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.15.4.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.15.6.tar.gz
 Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)

{trustgraph-flow-0.15.4 → trustgraph-flow-0.15.6}/trustgraph/agent/react/service.py

@@ -32,6 +32,7 @@ module = ".".join(__name__.split(".")[1:-1])
 default_input_queue = agent_request_queue
 default_output_queue = agent_response_queue
 default_subscriber = module
+default_max_iterations = 15
 
 class Processor(ConsumerProducer):
 
@@ -39,6 +40,8 @@ class Processor(ConsumerProducer):
 
         additional = params.get("context", None)
 
+        self.max_iterations = int(params.get("max_iterations", default_max_iterations))
+
         tools = {}
 
         # Parsing the prompt information to the prompt configuration
@@ -67,8 +70,9 @@ class Processor(ConsumerProducer):
                     )
 
                 if len(ttoks) == 1:
+
                     tools[toks[0]] = Tool(
-                        name = ttoks[0],
+                        name = toks[0],
                         description = "",
                         implementation = impl,
                         config = { "input": "query" },
@@ -76,7 +80,7 @@ class Processor(ConsumerProducer):
                     )
                 else:
                     tools[toks[0]] = Tool(
-                        name = ttoks[0],
+                        name = toks[0],
                         description = "",
                         implementation = impl,
                         config = { "input": ttoks[1] },
@@ -226,7 +230,7 @@ class Processor(ConsumerProducer):
 
             print(f"Question: {v.question}", flush=True)
 
-            if len(history) > 10:
+            if len(history) >= self.max_iterations:
                 raise RuntimeError("Too many agent iterations")
 
             print(f"History: {history}", flush=True)
@@ -394,6 +398,12 @@ description.'''
             help=f'Optional, specifies additional context text for the LLM.'
         )
 
+        parser.add_argument(
+            '--max-iterations',
+            default=default_max_iterations,
+            help=f'Maximum number of react iterations (default: {default_max_iterations})',
+        )
+
 def run():
 
     Processor.start(module, __doc__)

trustgraph-flow-0.15.6/trustgraph/flow_version.py

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

{trustgraph-flow-0.15.4 → trustgraph-flow-0.15.6}/trustgraph/model/text_completion/openai/llm.py

@@ -23,7 +23,7 @@ default_subscriber = module
 default_model = 'gpt-3.5-turbo'
 default_temperature = 0.0
 default_max_output = 4096
-default_api_key = os.getenv("OPENAI_KEY")
+default_api_key = os.getenv("OPENAI_TOKEN")
 
 class Processor(ConsumerProducer):
 

{trustgraph-flow-0.15.4 → trustgraph-flow-0.15.6}/trustgraph_flow.egg-info/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.1
 Name: trustgraph-flow
-Version: 0.15.4
+Version: 0.15.6
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Home-page: https://github.com/trustgraph-ai/trustgraph
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 License: UNKNOWN
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.15.4.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.15.6.tar.gz
 Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)

trustgraph-flow-0.15.4/trustgraph/flow_version.py

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