elfarash commited on
Commit
07aaf2b
·
1 Parent(s): 60ce5be
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,8 +4,7 @@ from langchain_openai import ChatOpenAI
4
  from langchain_core.messages import HumanMessage, SystemMessage, AIMessage
5
 
6
  # 1. Setup API Key (As requested)
7
- os.environ["OPENAI_API_KEY"] = os.environ.get("OPENAI_API_KEY")
8
-
9
  # 2. Define System Prompt (Full Detailed Content + Constraint)
10
  SYSTEM_PROMPT_TEXT = """
11
  Role: You are the Saudi Azm Employee Stocks Program (ESP) Specialist.
@@ -135,7 +134,7 @@ def generate_response(message, history, model_choice):
135
  api_model_name = model_choice
136
 
137
  # Initialize LLM
138
- llm = ChatOpenAI(model=api_model_name, temperature=0)
139
 
140
  # Construct Message History
141
  messages = [SystemMessage(content=SYSTEM_PROMPT_TEXT)]
 
4
  from langchain_core.messages import HumanMessage, SystemMessage, AIMessage
5
 
6
  # 1. Setup API Key (As requested)
7
+ api_key = os.getenv("OPENAI_API_KEY")
 
8
  # 2. Define System Prompt (Full Detailed Content + Constraint)
9
  SYSTEM_PROMPT_TEXT = """
10
  Role: You are the Saudi Azm Employee Stocks Program (ESP) Specialist.
 
134
  api_model_name = model_choice
135
 
136
  # Initialize LLM
137
+ llm = ChatOpenAI(model=api_model_name, temperature=0, openai_api_key=api_key)
138
 
139
  # Construct Message History
140
  messages = [SystemMessage(content=SYSTEM_PROMPT_TEXT)]