0504ankitsharma commited on
Commit
abf5123
·
verified ·
1 Parent(s): ba988d8

Upload server.py

Browse files
Files changed (1) hide show
  1. server.py +21 -6
server.py CHANGED
@@ -134,12 +134,12 @@ class QueryRequest(BaseModel):
134
  def train():
135
 
136
  # Check if the database should be cleared (using the --clear flag).
137
- parser = argparse.ArgumentParser()
138
- parser.add_argument("--reset", action="store_true", help="Reset the database.")
139
- args = parser.parse_args()
140
- if args.reset:
141
- print("✨ Clearing Database")
142
- clear_database()
143
 
144
  # Create (or update) the data store.
145
  documents = load_documents()
@@ -149,6 +149,21 @@ def train():
149
 
150
 
151
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
152
  @app.get("/query")
153
  def query(request: QueryRequest):
154
  query_text = request.query_text
 
134
  def train():
135
 
136
  # Check if the database should be cleared (using the --clear flag).
137
+ # parser = argparse.ArgumentParser()
138
+ # parser.add_argument("--reset", action="store_true", help="Reset the database.")
139
+ # args = parser.parse_args()
140
+ # if args.reset:
141
+ # print("✨ Clearing Database")
142
+ # clear_database()
143
 
144
  # Create (or update) the data store.
145
  documents = load_documents()
 
149
 
150
 
151
 
152
+ @app.get("/cleardb")
153
+ def cleardb():
154
+
155
+ # Check if the database should be cleared (using the --clear flag).
156
+ # parser = argparse.ArgumentParser()
157
+ # parser.add_argument("--reset", action="store_true", help="Reset the database.")
158
+ # args = parser.parse_args()
159
+ # if args.reset:
160
+ print("✨ Clearing Database")
161
+ clear_database()
162
+
163
+
164
+
165
+
166
+
167
  @app.get("/query")
168
  def query(request: QueryRequest):
169
  query_text = request.query_text