ppsingh commited on
Commit
36c612e
1 Parent(s): 29b43fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -24,7 +24,6 @@ installed_packages = pkg_resources.working_set
24
  package_list_ = ""
25
  for package in installed_packages:
26
  package_list_ = package_list_ + f"{package.key}=={package.version}\n"
27
-
28
  print(package_list_)
29
 
30
 
@@ -33,6 +32,8 @@ print(package_list_)
33
  # vectorestore colection are stored on persistent storage so this needs to be run only once
34
  # hence, comment out line below when creating for first time
35
  vectorstores = process_pdf()
 
 
36
 
37
  # -------------------------------------------------------------
38
  # Functions
@@ -102,9 +103,9 @@ async def chat(query,history,sources,reports,subtype,year):
102
 
103
  ##------------------------decide which collection to fetch------------------------------
104
  if len(reports) == 0:
105
- vectorstore = get_local_qdrant(sources)
106
  else:
107
- vectorstore = get_local_qdrant("allreports")
108
 
109
  ##------------------------------get context----------------------------------------------------
110
  context_retrieved_lst = []
 
24
  package_list_ = ""
25
  for package in installed_packages:
26
  package_list_ = package_list_ + f"{package.key}=={package.version}\n"
 
27
  print(package_list_)
28
 
29
 
 
32
  # vectorestore colection are stored on persistent storage so this needs to be run only once
33
  # hence, comment out line below when creating for first time
34
  vectorstores = process_pdf()
35
+ # once the vectore embeddings are created we will qdrant client to access these
36
+ client = QdrantClient(path=f"./data/local_qdrant")
37
 
38
  # -------------------------------------------------------------
39
  # Functions
 
103
 
104
  ##------------------------decide which collection to fetch------------------------------
105
  if len(reports) == 0:
106
+ vectorstore = get_local_qdrant(client,sources)
107
  else:
108
+ vectorstore = get_local_qdrant(client,"allreports")
109
 
110
  ##------------------------------get context----------------------------------------------------
111
  context_retrieved_lst = []