|
|
"""
|
|
|
Configuration and API Client Initialization
|
|
|
Handles environment variables and API client setup
|
|
|
"""
|
|
|
|
|
|
import os
|
|
|
from dotenv import load_dotenv
|
|
|
from langchain_groq import ChatGroq
|
|
|
from tavily import TavilyClient
|
|
|
|
|
|
|
|
|
load_dotenv()
|
|
|
|
|
|
|
|
|
def initialize_clients():
|
|
|
"""Initialize Groq LLM and Tavily API clients"""
|
|
|
groq_api_key = os.getenv("GROQ_API_KEY")
|
|
|
tavily_api_key = os.getenv("TAVILY_API_KEY")
|
|
|
|
|
|
if not groq_api_key or not tavily_api_key:
|
|
|
return None, None, "Missing API keys"
|
|
|
|
|
|
try:
|
|
|
llm = ChatGroq(
|
|
|
model="openai/gpt-oss-120b",
|
|
|
groq_api_key=groq_api_key,
|
|
|
temperature=0.3
|
|
|
)
|
|
|
|
|
|
tavily_client = TavilyClient(api_key=tavily_api_key)
|
|
|
|
|
|
return llm, tavily_client, None
|
|
|
except Exception as e:
|
|
|
return None, None, str(e)
|
|
|
|
|
|
|
|
|
def get_api_keys():
|
|
|
"""Get API keys from environment"""
|
|
|
return os.getenv("GROQ_API_KEY"), os.getenv("TAVILY_API_KEY")
|
|
|
|
|
|
|
