cryogenic22 commited on
Commit
2f2f744
·
verified ·
1 Parent(s): 3ad6907

Update utils/database.py

Browse files
Files changed (1) hide show
  1. utils/database.py +2 -2
utils/database.py CHANGED
@@ -337,8 +337,8 @@ def initialize_qa_system(vector_store):
337
  try:
338
  llm = ChatOpenAI(
339
  temperature=0.5,
340
- model_name="gpt-4",
341
- api_key=os.environ.get("OPENAI_API_KEY")
342
  )
343
 
344
  # Create retriever function
 
337
  try:
338
  llm = ChatOpenAI(
339
  temperature=0.5,
340
+ model_name="gemini-1.5-pro",
341
+ api_key=os.environ.get("GEMINI_API_KEY")
342
  )
343
 
344
  # Create retriever function