trustgraph-flow 1.0.1__tar.gz → 1.0.2__tar.gz

{trustgraph-flow-1.0.1 → trustgraph-flow-1.0.2}/PKG-INFO

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

trustgraph-flow-1.0.2/trustgraph/flow_version.py

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

{trustgraph-flow-1.0.1 → trustgraph-flow-1.0.2}/trustgraph/model/text_completion/openai/llm.py

@@ -16,7 +16,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(LlmService):
 
@@ -24,7 +27,7 @@ class Processor(LlmService):
     
         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)
 
@@ -43,7 +46,11 @@ class Processor(LlmService):
         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)
 
@@ -102,7 +109,7 @@ class Processor(LlmService):
 
             # Apart from rate limits, treat all exceptions as unrecoverable
 
-            print(f"Exception: {e}")
+            print(f"Exception: {type(e)} {e}")
             raise e
 
     @staticmethod

{trustgraph-flow-1.0.1 → trustgraph-flow-1.0.2}/trustgraph_flow.egg-info/PKG-INFO

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

trustgraph-flow-1.0.1/trustgraph/flow_version.py

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