Spaces:
Sleeping
Sleeping
Commit
·
0c6d84c
1
Parent(s):
9ccab25
Correções e evoluções
Browse files
llms.py
CHANGED
|
@@ -17,19 +17,19 @@ grok_llm = GrokChatModel(
|
|
| 17 |
api_key=os.getenv("X_API_KEY"),
|
| 18 |
model=os.getenv("GROK_MODEL_ID"),
|
| 19 |
base_url=os.getenv("X_API_BASE_URL"),
|
| 20 |
-
|
| 21 |
)
|
| 22 |
|
| 23 |
# Claude Sonnet
|
| 24 |
claude_llm = ChatAnthropic(
|
| 25 |
api_key=os.getenv("ANTHROPIC_API_KEY"),
|
| 26 |
model_name=os.getenv("CLAUDE_MODEL_ID"),
|
| 27 |
-
|
| 28 |
)
|
| 29 |
|
| 30 |
# Gemini
|
| 31 |
gemini_llm = ChatGoogleGenerativeAI(
|
| 32 |
google_api_key=os.getenv("GOOGLE_API_KEY"),
|
| 33 |
model=os.getenv("GEMINI_MODEL_ID"),
|
| 34 |
-
|
| 35 |
)
|
|
|
|
| 17 |
api_key=os.getenv("X_API_KEY"),
|
| 18 |
model=os.getenv("GROK_MODEL_ID"),
|
| 19 |
base_url=os.getenv("X_API_BASE_URL"),
|
| 20 |
+
timeout=900
|
| 21 |
)
|
| 22 |
|
| 23 |
# Claude Sonnet
|
| 24 |
claude_llm = ChatAnthropic(
|
| 25 |
api_key=os.getenv("ANTHROPIC_API_KEY"),
|
| 26 |
model_name=os.getenv("CLAUDE_MODEL_ID"),
|
| 27 |
+
timeout=900
|
| 28 |
)
|
| 29 |
|
| 30 |
# Gemini
|
| 31 |
gemini_llm = ChatGoogleGenerativeAI(
|
| 32 |
google_api_key=os.getenv("GOOGLE_API_KEY"),
|
| 33 |
model=os.getenv("GEMINI_MODEL_ID"),
|
| 34 |
+
timeout=900
|
| 35 |
)
|