Phoenix21 commited on
Commit
0ef5d8a
·
verified ·
1 Parent(s): 1fc8ba9

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -2
pipeline.py CHANGED
@@ -30,8 +30,9 @@ client = Mistral(api_key=mistral_api_key)
30
  # Ensure GOOGLE_API_KEY is set in your environment variables.
31
  gemini_llm = ChatGoogleGenerativeAI(
32
  model="gemini-1.5-pro",
33
- temperature=0,
34
  max_retries=2,
 
35
  # Additional parameters or safety_settings can be added here if needed
36
  )
37
 
@@ -39,8 +40,9 @@ gemini_llm = ChatGoogleGenerativeAI(
39
  pydantic_agent = ManagedAgent(
40
  llm=ChatGoogleGenerativeAI(
41
  model="gemini-1.5-pro",
42
- temperature=0,
43
  max_retries=2,
 
44
  ),
45
  tools=[DuckDuckGoSearchTool()]
46
  )
 
30
  # Ensure GOOGLE_API_KEY is set in your environment variables.
31
  gemini_llm = ChatGoogleGenerativeAI(
32
  model="gemini-1.5-pro",
33
+ temperature=0.5,
34
  max_retries=2,
35
+ google_api_key=os.environ.get("GEMINI_API_KEY"),
36
  # Additional parameters or safety_settings can be added here if needed
37
  )
38
 
 
40
  pydantic_agent = ManagedAgent(
41
  llm=ChatGoogleGenerativeAI(
42
  model="gemini-1.5-pro",
43
+ temperature=0.5,
44
  max_retries=2,
45
+ google_api_key=os.environ.get("GEMINI_API_KEY"),
46
  ),
47
  tools=[DuckDuckGoSearchTool()]
48
  )