Soumik555 commited on
Commit
b2c0e5d
·
1 Parent(s): cf9be77

changed model to .env gemini-flash-2.0

Browse files
Files changed (1) hide show
  1. cereberas_langchain_agent.py +1 -1
cereberas_langchain_agent.py CHANGED
@@ -21,7 +21,7 @@ cerebras_api_key = os.getenv("CEREBRAS_API_KEY")
21
  cerebras_base_url = os.getenv("CEREBRAS_BASE_URL")
22
 
23
  # Initialize ChatCerebras LLM
24
- llm = ChatCerebras(model=model_name, api_key=cerebras_api_key, base_url=cerebras_base_url, temperature=0.7)
25
 
26
  def create_agent(llm, data, tools):
27
  """Create agent with tool names"""
 
21
  cerebras_base_url = os.getenv("CEREBRAS_BASE_URL")
22
 
23
  # Initialize ChatCerebras LLM
24
+ llm = ChatCerebras(model=model_name, api_key=cerebras_api_key)
25
 
26
  def create_agent(llm, data, tools):
27
  """Create agent with tool names"""