LaunchLab-chat / config /chabot_config.py
MuhammadSaad16's picture
Add application file
1ef3cc9
import os
from dotenv import load_dotenv
from agents import AsyncOpenAI, OpenAIChatCompletionsModel, set_tracing_disabled
set_tracing_disabled(True)
load_dotenv()
openai_api_key = os.getenv("OPENAI_API_KEY")
gemini_api_key = os.getenv("GEMINI_API_KEY") # Optional, ignore if not set
# No strict check—use OpenAI directly (Gemini fallback if you want later)
if not openai_api_key:
raise ValueError(
"OPENAI_API_KEY is not set. Please add it to your .env file: OPENAI_API_KEY=your_key_here"
)
client_provider = AsyncOpenAI(
api_key=openai_api_key,
base_url="https://api.openai.com/v1/",
)
# If you want Gemini fallback (uncomment below, but CEO ke against hai abhi)
# if openai_api_key:
# ... (OpenAI part)
# else:
# if not gemini_api_key:
# raise ValueError("No API key found!")
# client_provider = AsyncOpenAI(
# api_key=gemini_api_key,
# base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
# )
model = OpenAIChatCompletionsModel(
model="gpt-4o", # FIXED: Using valid OpenAI model (fastest GPT-4 variant)
openai_client=client_provider
)
print("Setup complete! Model ready with OpenAI GPT-4o") # Debug line