mirror of
https://github.com/rowboatlabs/rowboat.git
synced 2026-04-25 08:26:22 +02:00
34 lines
No EOL
951 B
Python
34 lines
No EOL
951 B
Python
import os
|
|
from openai import OpenAI
|
|
import dotenv
|
|
dotenv.load_dotenv()
|
|
|
|
PROVIDER_BASE_URL = os.getenv('PROVIDER_BASE_URL', '')
|
|
PROVIDER_API_KEY = os.getenv('PROVIDER_API_KEY')
|
|
PROVIDER_DEFAULT_MODEL = os.getenv('PROVIDER_DEFAULT_MODEL')
|
|
PROVIDER_COPILOT_MODEL = os.getenv('PROVIDER_COPILOT_MODEL')
|
|
|
|
if not PROVIDER_COPILOT_MODEL:
|
|
PROVIDER_COPILOT_MODEL = 'gpt-4.1'
|
|
|
|
if not PROVIDER_API_KEY:
|
|
PROVIDER_API_KEY = os.getenv('OPENAI_API_KEY')
|
|
|
|
if not PROVIDER_API_KEY:
|
|
raise(ValueError("No LLM Provider API key found"))
|
|
|
|
if not PROVIDER_DEFAULT_MODEL:
|
|
PROVIDER_DEFAULT_MODEL = 'gpt-4.1'
|
|
|
|
completions_client = None
|
|
if PROVIDER_BASE_URL:
|
|
print(f"Using provider {PROVIDER_BASE_URL}, for completions")
|
|
completions_client = OpenAI(
|
|
base_url=PROVIDER_BASE_URL,
|
|
api_key=PROVIDER_API_KEY
|
|
)
|
|
else:
|
|
print(f"Using OpenAI directly for completions")
|
|
completions_client = OpenAI(
|
|
api_key=PROVIDER_API_KEY
|
|
) |