Skip to content

Commit

Permalink
refactor: Update llm parameter value in get_response_from_llm function
Browse files Browse the repository at this point in the history
  • Loading branch information
preetvadaliya committed Jul 16, 2024
1 parent c1df437 commit 0a825fe
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion functions/handlers.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ def get_response_from_llm(query, llm):
if llm == 'gpt-4' or llm == 'gpt-3.5-turbo-instruct':
llm_model = ChatOpenAI(api_key=environ.get('OPEN_AI_API_KEY'), temperature=0, model=llm)
response = get_health_ai_response(query, llm_model)
elif llm == 'gemini':
elif llm == 'google':
llm_model = ChatGoogleGenerativeAI(
model='gemini-1.5-pro-latest', google_api_key=environ.get('GOOGLE_API_KEY')
)
Expand Down

0 comments on commit 0a825fe

Please sign in to comment.