diff --git a/context_chat_backend/controller.py b/context_chat_backend/controller.py index 80437f2..d8e43d2 100644 --- a/context_chat_backend/controller.py +++ b/context_chat_backend/controller.py @@ -1,6 +1,4 @@ -import time -import time from os import getenv from typing import Annotated @@ -9,7 +7,6 @@ from fastapi import BackgroundTasks, Body, FastAPI, Request, UploadFile from langchain.llms.base import LLM from pydantic import BaseModel, FieldValidationInfo, field_validator -from datetime import datetime from .chain import ScopeType, embed_sources, process_query, process_scoped_query from .download import download_all_models @@ -191,7 +188,7 @@ def _(userId: Annotated[str, Body()], providerKey: Annotated[str, Body()]): return JSONResponse('Error: VectorDB delete failed, check vectordb logs for more info.', 400) return JSONResponse('All valid sources deleted') - + @app.put('/loadSources') @enabled_guard(app) diff --git a/requirements_cuda.txt b/requirements_cuda.txt index 9f52566..bf8dd2e 100644 --- a/requirements_cuda.txt +++ b/requirements_cuda.txt @@ -153,4 +153,4 @@ zipp==3.17.0 torch==2.2.0 torchaudio==2.2.0 torchvision==0.17.0 -ctransformers[cuda]==0.2.27 \ No newline at end of file +ctransformers[cuda]==0.2.27