Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -170,8 +170,8 @@ from transformers import Blip2Processor, Blip2ForConditionalGeneration
|
|
| 170 |
|
| 171 |
from tavily import TavilyClient
|
| 172 |
|
| 173 |
-
|
| 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 |
|