BtB-ExpC commited on
Commit
4d9acf1
·
1 Parent(s): fe3b390

model_name parameter corrected to model for Google

Browse files
Files changed (1) hide show
  1. config/llm_config.py +1 -1
config/llm_config.py CHANGED
@@ -43,7 +43,7 @@ def create_deepseek_llm(model_name: str, temperature: float):
43
  return ChatDeepSeek(api_key=DEEPSEEK_API_KEY, model_name=model_name, temperature=temperature)
44
 
45
  def create_google_reasoning_llm(model_name: str):
46
- return ChatGoogleGenerativeAI(api_key=GOOGLE_API_KEY, model_name=model_name)
47
 
48
  # all of them in one dictionary
49
  llms = {
 
43
  return ChatDeepSeek(api_key=DEEPSEEK_API_KEY, model_name=model_name, temperature=temperature)
44
 
45
  def create_google_reasoning_llm(model_name: str):
46
+ return ChatGoogleGenerativeAI(api_key=GOOGLE_API_KEY, model=model_name)
47
 
48
  # all of them in one dictionary
49
  llms = {