77import uuid
88import warnings
99from pydantic import BaseModel
10-
1110from langchain_community .chat_models import ErnieBotChat
1211from langchain_nvidia_ai_endpoints import ChatNVIDIA
1312from langchain .chat_models import init_chat_model
14-
1513from ..helpers import models_tokens
1614from ..models import (
1715 OneApi ,
1816 DeepSeek
1917)
2018from ..utils .logging import set_verbosity_warning , set_verbosity_info
2119
22-
23-
2420class AbstractGraph (ABC ):
2521 """
2622 Scaffolding class for creating a graph representation and executing it.
@@ -53,6 +49,9 @@ class AbstractGraph(ABC):
5349 def __init__ (self , prompt : str , config : dict ,
5450 source : Optional [str ] = None , schema : Optional [BaseModel ] = None ):
5551
52+ if config .get ("llm" ).get ("temperature" ) is None :
53+ config ["llm" ]["temperature" ] = 0
54+
5655 self .prompt = prompt
5756 self .source = source
5857 self .config = config
@@ -212,7 +211,7 @@ def handle_model(model_name, provider, token_key, default_token=8192):
212211 print ("model not found, using default token size (8192)" )
213212 self .model_token = 8192
214213 return ErnieBotChat (llm_params )
215-
214+
216215 if "oneapi" in llm_params ["model" ]:
217216 # take the model after the last dash
218217 llm_params ["model" ] = llm_params ["model" ].split ("/" )[- 1 ]
@@ -221,7 +220,7 @@ def handle_model(model_name, provider, token_key, default_token=8192):
221220 except KeyError as exc :
222221 raise KeyError ("Model not supported" ) from exc
223222 return OneApi (llm_params )
224-
223+
225224 if "nvidia" in llm_params ["model" ]:
226225 try :
227226 self .model_token = models_tokens ["nvidia" ][llm_params ["model" ].split ("/" )[- 1 ]]
0 commit comments