trustgraph-flow 0.22.10__tar.gz → 0.22.11__tar.gz

{trustgraph-flow-0.22.10 → trustgraph-flow-0.22.11}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph-flow
-Version: 0.22.10
+Version: 0.22.11
 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
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.22.10.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.22.11.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3

trustgraph-flow-0.22.11/trustgraph/flow_version.py

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

{trustgraph-flow-0.22.10 → trustgraph-flow-0.22.11}/trustgraph/model/text_completion/openai/llm.py

@@ -24,7 +24,10 @@ default_model = 'gpt-3.5-turbo'
 default_temperature = 0.0
 default_max_output = 4096
 default_api_key = os.getenv("OPENAI_TOKEN")
-default_base_url = os.getenv("OPENAI_BASE_URL", None)
+default_base_url = os.getenv("OPENAI_BASE_URL")
+
+if default_base_url is None or default_base_url == "":
+    default_base_url = "https://api.openai.com/v1"
 
 class Processor(ConsumerProducer):
 
@@ -35,7 +38,7 @@ class Processor(ConsumerProducer):
         subscriber = params.get("subscriber", default_subscriber)
         model = params.get("model", default_model)
         api_key = params.get("api_key", default_api_key)
-        base_url = params.get("base_url", default_base_url)
+        base_url = params.get("url", default_base_url)
         temperature = params.get("temperature", default_temperature)
         max_output = params.get("max_output", default_max_output)
 
@@ -72,7 +75,11 @@ class Processor(ConsumerProducer):
         self.model = model
         self.temperature = temperature
         self.max_output = max_output
-        self.openai = OpenAI(base_url=base_url, api_key=api_key)
+
+        if base_url:
+            self.openai = OpenAI(base_url=base_url, api_key=api_key)
+        else:
+            self.openai = OpenAI(api_key=api_key)
 
         print("Initialised", flush=True)
 
@@ -144,7 +151,7 @@ class Processor(ConsumerProducer):
 
             # Apart from rate limits, treat all exceptions as unrecoverable
 
-            print(f"Exception: {e}")
+            print(f"Exception: {type(e)} {e}")
 
             print("Send error response...", flush=True)
 

{trustgraph-flow-0.22.10 → trustgraph-flow-0.22.11}/trustgraph_flow.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph-flow
-Version: 0.22.10
+Version: 0.22.11
 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
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.22.10.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.22.11.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3

trustgraph-flow-0.22.10/trustgraph/flow_version.py

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