model_name parameter corrected to model for Google
Browse files- 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,
|
| 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 = {
|