-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconfig.py
More file actions
30 lines (25 loc) · 883 Bytes
/
config.py
File metadata and controls
30 lines (25 loc) · 883 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import os
from dotenv import load_dotenv
load_dotenv() # Optional, for local dev with .env file
class Settings:
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
PROVIDER = os.getenv("PROVIDER", "enclovai")
settings = Settings()
MODEL_CONFIG = {
"default_model": os.getenv("AI_MODEL", "openai/gpt-4.1"),
"models": {
"openai": {
"base_url": "https://api.openai.com/v1",
"model_name": "openai/gpt-4.1"
},
"github": {
"base_url": "https://models.github.ai/inference",
"model_name": "openai/gpt-4.1"
},
# Add more providers as needed
}
}
def get_model_config(provider_key="openai"):
provider = MODEL_CONFIG["models"].get(provider_key, MODEL_CONFIG["models"]["openai"])
return provider["base_url"], provider["model_name"]