test
Browse files
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 |
-
|
| 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)]
|