Spaces:
Sleeping
Sleeping
Commit
·
39f6b9b
1
Parent(s):
e4b1fbd
Debug API key
Browse files- Api/app/main.py +6 -16
- Api/app/qdrant.py +2 -5
Api/app/main.py
CHANGED
|
@@ -1,29 +1,20 @@
|
|
| 1 |
import logging
|
| 2 |
from contextlib import asynccontextmanager
|
| 3 |
|
| 4 |
-
import uvicorn
|
| 5 |
-
from fastapi import APIRouter, FastAPI
|
| 6 |
-
from fastapi.middleware.cors import CORSMiddleware
|
| 7 |
-
from fastapi.responses import FileResponse
|
| 8 |
-
from fastapi.staticfiles import StaticFiles
|
| 9 |
-
from transformers import AutoModel, AutoModelForMaskedLM, AutoTokenizer, pipeline
|
| 10 |
-
|
| 11 |
from app.modules.clearVariables.routes.clearVariables_route import \
|
| 12 |
router as clear_variables_routes
|
| 13 |
from app.modules.documentHandeler.routes.document_handeler_route import \
|
| 14 |
router as upload_file_routes
|
| 15 |
from app.modules.querySearch.routes.querySearch_route import \
|
| 16 |
router as query_search_routes
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
|
| 19 |
-
from dotenv import load_dotenv
|
| 20 |
-
import os
|
| 21 |
-
from app.service.api import baseURL
|
| 22 |
-
|
| 23 |
-
load_dotenv()
|
| 24 |
-
api_key = os.getenv("QDRANT_API_KEY")
|
| 25 |
-
print("Api key: ", api_key)
|
| 26 |
-
|
| 27 |
@asynccontextmanager
|
| 28 |
async def lifespan(app: FastAPI):
|
| 29 |
dense_model_name = "sentence-transformers/all-MiniLM-L6-v2"
|
|
@@ -37,7 +28,6 @@ async def lifespan(app: FastAPI):
|
|
| 37 |
sparse_model = AutoModelForMaskedLM.from_pretrained(sparse_model_name)
|
| 38 |
|
| 39 |
qa_pipeline = pipeline("question-answering", model=qa_model_name)
|
| 40 |
-
print("Api key: ", api_key)
|
| 41 |
|
| 42 |
yield {
|
| 43 |
"dense_tokenizer": dense_tokenizer,
|
|
|
|
| 1 |
import logging
|
| 2 |
from contextlib import asynccontextmanager
|
| 3 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
from app.modules.clearVariables.routes.clearVariables_route import \
|
| 5 |
router as clear_variables_routes
|
| 6 |
from app.modules.documentHandeler.routes.document_handeler_route import \
|
| 7 |
router as upload_file_routes
|
| 8 |
from app.modules.querySearch.routes.querySearch_route import \
|
| 9 |
router as query_search_routes
|
| 10 |
+
from fastapi import APIRouter, FastAPI
|
| 11 |
+
from fastapi.middleware.cors import CORSMiddleware
|
| 12 |
+
from fastapi.responses import FileResponse
|
| 13 |
+
from fastapi.staticfiles import StaticFiles
|
| 14 |
+
from transformers import (AutoModel, AutoModelForMaskedLM, AutoTokenizer,
|
| 15 |
+
pipeline)
|
| 16 |
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
@asynccontextmanager
|
| 19 |
async def lifespan(app: FastAPI):
|
| 20 |
dense_model_name = "sentence-transformers/all-MiniLM-L6-v2"
|
|
|
|
| 28 |
sparse_model = AutoModelForMaskedLM.from_pretrained(sparse_model_name)
|
| 29 |
|
| 30 |
qa_pipeline = pipeline("question-answering", model=qa_model_name)
|
|
|
|
| 31 |
|
| 32 |
yield {
|
| 33 |
"dense_tokenizer": dense_tokenizer,
|
Api/app/qdrant.py
CHANGED
|
@@ -1,13 +1,10 @@
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
-
from dotenv import load_dotenv
|
| 4 |
-
from qdrant_client import QdrantClient, models
|
| 5 |
-
|
| 6 |
from app.service.api import baseURL
|
|
|
|
| 7 |
|
| 8 |
-
|
| 9 |
|
| 10 |
-
api_key = os.getenv("QDRANT_API_KEY")
|
| 11 |
|
| 12 |
class QdrantConnectionDb:
|
| 13 |
client = None
|
|
|
|
| 1 |
import os
|
| 2 |
|
|
|
|
|
|
|
|
|
|
| 3 |
from app.service.api import baseURL
|
| 4 |
+
from qdrant_client import QdrantClient, models
|
| 5 |
|
| 6 |
+
api_key = os.environ.get("QDRANT_API_KEY")
|
| 7 |
|
|
|
|
| 8 |
|
| 9 |
class QdrantConnectionDb:
|
| 10 |
client = None
|