mralamdari commited on
Commit
80a148f
·
verified ·
1 Parent(s): 5839fb4

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -170,8 +170,8 @@ from transformers import Blip2Processor, Blip2ForConditionalGeneration
170
 
171
  from tavily import TavilyClient
172
 
173
- # GPT_4o_API_KEY = os.environ["DeepSeek_API_KEY"]
174
- GPT_4o_API_KEY = os.environ["GPT_4o_API_KEY"]
175
  TAVILY_API_KEY = os.environ["TAVILY_API_KEY"]
176
  Pixtral12B_API = os.environ["Pixtral12B_API"]
177
 
@@ -180,7 +180,7 @@ API_URL = "https://openrouter.ai/api/v1/chat/completions"
180
 
181
  model = ChatOpenAI(openai_api_key=GPT_4o_API_KEY,
182
  base_url="https://openrouter.ai/api/v1",
183
- model_name="gpt-4o",
184
  temperature=0)
185
  tavily = TavilyClient(api_key=TAVILY_API_KEY)
186
 
 
170
 
171
  from tavily import TavilyClient
172
 
173
+ GPT_4o_API_KEY = os.environ["DeepSeek_API_KEY"]
174
+ # GPT_4o_API_KEY = os.environ["GPT_4o_API_KEY"]
175
  TAVILY_API_KEY = os.environ["TAVILY_API_KEY"]
176
  Pixtral12B_API = os.environ["Pixtral12B_API"]
177
 
 
180
 
181
  model = ChatOpenAI(openai_api_key=GPT_4o_API_KEY,
182
  base_url="https://openrouter.ai/api/v1",
183
+ # model_name="gpt-4o",
184
  temperature=0)
185
  tavily = TavilyClient(api_key=TAVILY_API_KEY)
186