openfree commited on
Commit
790fa45
·
verified ·
1 Parent(s): b0e8b7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -12,9 +12,8 @@ from aimakerspace.vectordatabase import VectorDatabase
12
  from aimakerspace.openai_utils.chatmodel import ChatOpenAI
13
  import chainlit as cl
14
 
15
- # Get OpenAI API key from environment variables
16
- openai_api_key = os.getenv("OPENAI_API_KEY")
17
- if not openai_api_key:
18
  raise ValueError("OPENAI_API_KEY environment variable is not set")
19
 
20
  system_template = """\
@@ -116,8 +115,8 @@ async def on_chat_start():
116
  vector_db = VectorDatabase()
117
  vector_db = await vector_db.abuild_from_list(texts)
118
 
119
- # Initialize ChatOpenAI with API key from environment variable
120
- chat_openai = ChatOpenAI(api_key=openai_api_key)
121
 
122
  # Create a chain
123
  retrieval_augmented_qa_pipeline = RetrievalAugmentedQAPipeline(
 
12
  from aimakerspace.openai_utils.chatmodel import ChatOpenAI
13
  import chainlit as cl
14
 
15
+ # Ensure OpenAI API key is set in environment variables
16
+ if not os.getenv("OPENAI_API_KEY"):
 
17
  raise ValueError("OPENAI_API_KEY environment variable is not set")
18
 
19
  system_template = """\
 
115
  vector_db = VectorDatabase()
116
  vector_db = await vector_db.abuild_from_list(texts)
117
 
118
+ # Initialize ChatOpenAI (API key will be automatically picked up from environment variable)
119
+ chat_openai = ChatOpenAI()
120
 
121
  # Create a chain
122
  retrieval_augmented_qa_pipeline = RetrievalAugmentedQAPipeline(