Update main.py
Browse files
main.py
CHANGED
@@ -35,7 +35,7 @@ class QueryRequest(BaseModel):
|
|
35 |
|
36 |
def save_embeddings(index_id, document_list):
|
37 |
try:
|
38 |
-
folder_path = f"indexes/{index_id}"
|
39 |
os.makedirs(folder_path, exist_ok=True)
|
40 |
|
41 |
# Save embeddings
|
@@ -51,7 +51,7 @@ def save_embeddings(index_id, document_list):
|
|
51 |
|
52 |
def load_embeddings(index_id):
|
53 |
try:
|
54 |
-
folder_path = f"indexes/{index_id}"
|
55 |
|
56 |
if not os.path.exists(folder_path):
|
57 |
logger.error(f"Index not found for index_id: {index_id}")
|
@@ -96,4 +96,4 @@ async def query_index(request: QueryRequest):
|
|
96 |
|
97 |
if __name__ == "__main__":
|
98 |
import uvicorn
|
99 |
-
uvicorn.run(app, host="0.0.0.0", port=
|
|
|
35 |
|
36 |
def save_embeddings(index_id, document_list):
|
37 |
try:
|
38 |
+
folder_path = f"/app/indexes/{index_id}"
|
39 |
os.makedirs(folder_path, exist_ok=True)
|
40 |
|
41 |
# Save embeddings
|
|
|
51 |
|
52 |
def load_embeddings(index_id):
|
53 |
try:
|
54 |
+
folder_path = f"/app/indexes/{index_id}"
|
55 |
|
56 |
if not os.path.exists(folder_path):
|
57 |
logger.error(f"Index not found for index_id: {index_id}")
|
|
|
96 |
|
97 |
if __name__ == "__main__":
|
98 |
import uvicorn
|
99 |
+
uvicorn.run(app, host="0.0.0.0", port=8000)
|