apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-25 00:16:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Make sure text-completion-openai works with empty base URL string (#387)

Browse source
This commit is contained in:
cybermaggedon 2025-05-22 19:00:29 +01:00 • committed by GitHub
parent dc0ce1041b
commit 2641e418cd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
trustgraph-flow/trustgraph/model/text_completion/openai/llm.py
View file

@ -24,7 +24,10 @@ default_model = 'gpt-3.5-turbo'
default_temperature = 0.0 default_temperature = 0.0
default_max_output = 4096 default_max_output = 4096
default_api_key = os.getenv("OPENAI_TOKEN") 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): class Processor(ConsumerProducer):
@ -35,7 +38,7 @@ class Processor(ConsumerProducer):
subscriber = params.get("subscriber", default_subscriber) subscriber = params.get("subscriber", default_subscriber)
model = params.get("model", default_model) model = params.get("model", default_model)
api_key = params.get("api_key", default_api_key) 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) temperature = params.get("temperature", default_temperature)
max_output = params.get("max_output", default_max_output) max_output = params.get("max_output", default_max_output)
@ -72,7 +75,11 @@ class Processor(ConsumerProducer):
self.model = model self.model = model
self.temperature = temperature self.temperature = temperature
self.max_output = max_output 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) print("Initialised", flush=True)
@ -144,7 +151,7 @@ class Processor(ConsumerProducer):
# Apart from rate limits, treat all exceptions as unrecoverable # 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) print("Send error response...", flush=True)