Spaces:
Paused
Paused
Ashvanth.S
commited on
Commit
·
56ca6bf
1
Parent(s):
fdeaa04
update files
Browse files- main_app.py +2 -2
- utils/agent.py +1 -1
main_app.py
CHANGED
@@ -21,8 +21,8 @@ class AnswerResponse(BaseModel):
|
|
21 |
answer: str
|
22 |
|
23 |
def init_rag_system():
|
24 |
-
pdf_path = os.
|
25 |
-
vector_store_path = os.
|
26 |
# Load embeddings
|
27 |
embeddings = get_embeddings()
|
28 |
if os.path.exists(vector_store_path) and os.listdir(vector_store_path):
|
|
|
21 |
answer: str
|
22 |
|
23 |
def init_rag_system():
|
24 |
+
pdf_path = os.environ.get("SOURCE_DATA")
|
25 |
+
vector_store_path = os.environ.get("VECTOR_STORE")
|
26 |
# Load embeddings
|
27 |
embeddings = get_embeddings()
|
28 |
if os.path.exists(vector_store_path) and os.listdir(vector_store_path):
|
utils/agent.py
CHANGED
@@ -89,7 +89,7 @@ prompt = ChatPromptTemplate.from_messages([
|
|
89 |
])
|
90 |
|
91 |
def get_model_use():
|
92 |
-
return ChatOpenAI(api_key=os.
|
93 |
|
94 |
def init_agent():
|
95 |
llm = get_model_use()
|
|
|
89 |
])
|
90 |
|
91 |
def get_model_use():
|
92 |
+
return ChatOpenAI(api_key=os.environ.get("OPEN_API_KEY"),temperature=0)
|
93 |
|
94 |
def init_agent():
|
95 |
llm = get_model_use()
|