Skip to content

Commit

Permalink
chore: Update .env.example with Google Application Credentials path
Browse files Browse the repository at this point in the history
Signed-off-by: Palaniappan R <[email protected]>
  • Loading branch information
palaniappan-r committed Jul 15, 2024
1 parent 232c9a7 commit f89c373
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
1 change: 1 addition & 0 deletions backend/.env.example
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
GOOGLE_API_KEY={{GOOGLE_API_KEY}}
GOOGLE_APPLICATION_CREDENTIALS={{PATH_TO_GOOGLE_APPLICATION_CREDENTIALS}}

USE_CUDA=false
GOOGLE_GEMINI="1_pro" #1_pro or 1.5_flash or 1.5_pro
Expand Down
4 changes: 2 additions & 2 deletions backend/src/agents/retriever_graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,9 +65,9 @@ def agent(self, state: AgentState) -> dict[str, list[str]]:
response = model.invoke(messages)

if response is None or response.tool_calls is None: # type: ignore
return {'tools': []} # type: ignore
return {'tools': []}

return {'tools': response.tool_calls}
return {'tools': response.tool_calls} # type: ignore

def generate(self, state: AgentState) -> dict[str, list[AnyMessage]]:
query = state['messages'][-1].content
Expand Down

0 comments on commit f89c373

Please sign in to comment.