Skip to content

Update graph.py #112

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
+3 −3
Open

Update graph.py #112

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions backend/src/agent/graph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@
def generate_query(state: OverallState, config: RunnableConfig) -> QueryGenerationState:
"""LangGraph node that generates search queries based on the User's question.

Uses Gemini 2.0 Flash to create an optimized search queries for web research based on
Uses Gemini 2.5 Flash to create an optimized search queries for web research based on
the User's question.

Args:
Expand All @@ -60,7 +60,7 @@ def generate_query(state: OverallState, config: RunnableConfig) -> QueryGenerati
if state.get("initial_search_query_count") is None:
state["initial_search_query_count"] = configurable.number_of_initial_queries

# init Gemini 2.0 Flash
# init Gemini 2.5 Flash
llm = ChatGoogleGenerativeAI(
model=configurable.query_generator_model,
temperature=1.0,
Expand Down Expand Up @@ -95,7 +95,7 @@ def continue_to_web_research(state: QueryGenerationState):
def web_research(state: WebSearchState, config: RunnableConfig) -> OverallState:
"""LangGraph node that performs web research using the native Google Search API tool.

Executes a web search using the native Google Search API tool in combination with Gemini 2.0 Flash.
Executes a web search using the native Google Search API tool in combination with Gemini 2.5 Flash.

Args:
state: Current graph state containing the search query and research loop count
Expand Down