Update genai_handler.py
Browse files- genai_handler.py +5 -3
genai_handler.py
CHANGED
@@ -7,12 +7,14 @@ from openai import OpenAI, APIError, RateLimitError
|
|
7 |
|
8 |
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
9 |
|
10 |
-
API_KEY_ENV_VAR = "
|
11 |
api_key_value = os.getenv(API_KEY_ENV_VAR)
|
12 |
|
13 |
# --- Configuration ---
|
14 |
-
BASE_URL = "https://beta.sree.shop/v1"
|
15 |
-
MODEL_NAME = "Provider-5/gpt-4o" # Use the specified model
|
|
|
|
|
16 |
|
17 |
try:
|
18 |
client = OpenAI(api_key=api_key_value, base_url=BASE_URL)
|
|
|
7 |
|
8 |
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
9 |
|
10 |
+
API_KEY_ENV_VAR = "QA"
|
11 |
api_key_value = os.getenv(API_KEY_ENV_VAR)
|
12 |
|
13 |
# --- Configuration ---
|
14 |
+
# BASE_URL = "https://beta.sree.shop/v1"
|
15 |
+
# MODEL_NAME = "Provider-5/gpt-4o" # Use the specified model
|
16 |
+
BASE_URL = "https://openrouter.ai/api/v1"
|
17 |
+
MODEL_NAME = "meta-llama/llama-4-scout:free"
|
18 |
|
19 |
try:
|
20 |
client = OpenAI(api_key=api_key_value, base_url=BASE_URL)
|