Ashgen12 commited on
Commit
7be4201
·
verified ·
1 Parent(s): d66aba2

Update genai_handler.py

Browse files
Files changed (1) hide show
  1. 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 = "API_TOKEN"
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)