Spaces:
Sleeping
Sleeping
style(app): apply isort format
Browse files
app.py
CHANGED
|
@@ -1,20 +1,18 @@
|
|
| 1 |
import asyncio
|
|
|
|
| 2 |
from pathlib import Path
|
|
|
|
| 3 |
import markdown
|
| 4 |
-
from
|
|
|
|
|
|
|
| 5 |
from fastapi.middleware.cors import CORSMiddleware
|
| 6 |
from fastapi.templating import Jinja2Templates
|
| 7 |
-
from contextlib import asynccontextmanager
|
| 8 |
-
|
| 9 |
from manager.dialogue_manager import handle_dialogue
|
| 10 |
-
from
|
| 11 |
-
from
|
|
|
|
| 12 |
from schemas import AskReq, AskRes
|
| 13 |
-
from config import (
|
| 14 |
-
FALLBACK_MODEL_NAME, FALLBACK_MODEL_DIR,
|
| 15 |
-
EMBEDDER_MODEL_NAME, EMBEDDER_MODEL_DIR,
|
| 16 |
-
HF_TOKEN, BASE_DIR
|
| 17 |
-
)
|
| 18 |
|
| 19 |
templates = Jinja2Templates(directory="templates")
|
| 20 |
model_ready = False
|
|
|
|
| 1 |
import asyncio
|
| 2 |
+
from contextlib import asynccontextmanager
|
| 3 |
from pathlib import Path
|
| 4 |
+
|
| 5 |
import markdown
|
| 6 |
+
from config import (BASE_DIR, EMBEDDER_MODEL_DIR, EMBEDDER_MODEL_NAME,
|
| 7 |
+
FALLBACK_MODEL_DIR, FALLBACK_MODEL_NAME, HF_TOKEN)
|
| 8 |
+
from fastapi import FastAPI, HTTPException, Request
|
| 9 |
from fastapi.middleware.cors import CORSMiddleware
|
| 10 |
from fastapi.templating import Jinja2Templates
|
|
|
|
|
|
|
| 11 |
from manager.dialogue_manager import handle_dialogue
|
| 12 |
+
from models.model_loader import load_embedder, load_fallback_model
|
| 13 |
+
from rag.rag_manager import (add_docs, chroma_initialized,
|
| 14 |
+
load_game_docs_from_disk, set_embedder)
|
| 15 |
from schemas import AskReq, AskRes
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
templates = Jinja2Templates(directory="templates")
|
| 18 |
model_ready = False
|