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

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +9 -9
pipeline.py CHANGED
@@ -37,15 +37,15 @@ gemini_llm = ChatGoogleGenerativeAI(
37
  )
38
 
39
  # Initialize ManagedAgent for web search using Gemini
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
- )
49
 
50
  class QueryInput(BaseModel):
51
  query: str = Field(..., min_length=1, description="The input query string")
 
37
  )
38
 
39
  # Initialize ManagedAgent for web search using Gemini
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
+ # )
49
 
50
  class QueryInput(BaseModel):
51
  query: str = Field(..., min_length=1, description="The input query string")