victorafarias commited on
Commit
0c6d84c
·
1 Parent(s): 9ccab25

Correções e evoluções

Browse files
Files changed (1) hide show
  1. llms.py +3 -3
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
- client_kwargs={"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
- client_kwargs={"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
- client_kwargs={"timeout": 900}
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
  )