Ashvanth.S commited on
Commit
56ca6bf
·
1 Parent(s): fdeaa04

update files

Browse files
Files changed (2) hide show
  1. main_app.py +2 -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.getenv("SOURCE_DATA")
25
- vector_store_path = os.getenv("VECTOR_STORE")
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.getenv("OPEN_API_KEY"),temperature=0)
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()