agentforge / confg.py
Tahasaif3's picture
'code'
fd1a435
raw
history blame contribute delete
581 Bytes
import os
from dotenv import load_dotenv
from agents import AsyncOpenAI,OpenAIChatCompletionsModel,set_tracing_disabled
from tavily import TavilyClient
set_tracing_disabled(True)
load_dotenv()
gemini_api_key = os.getenv("GEMINI_API_KEY")
tavily_api_key = os.getenv("TAVILY_SECRET_KEY")
tavily_client = TavilyClient(api_key=tavily_api_key)
client_provider = AsyncOpenAI(
api_key=gemini_api_key,
base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
)
model = OpenAIChatCompletionsModel(
model="gemini-2.5-flash",
openai_client=client_provider
)