feat: knowledge backend management

1.knowledge_type.py
2.knowledge backend api
This commit is contained in:
aries_ckt
2023-06-27 15:29:13 +08:00
parent 746da74116
commit 57da3c23a6
16 changed files with 153 additions and 100 deletions

View File

@@ -23,7 +23,6 @@ from pilot.configs.model_config import *
from pilot.model.llm_out.vicuna_base_llm import get_embeddings
from pilot.model.loader import ModelLoader
from pilot.server.chat_adapter import get_llm_chat_adapter
from knowledge.knowledge_controller import router
CFG = Config()
@@ -106,6 +105,7 @@ worker = ModelWorker(
)
app = FastAPI()
from pilot.openapi.knowledge.knowledge_controller import router
app.include_router(router)
origins = [
@@ -119,7 +119,7 @@ app.add_middleware(
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
allow_headers=["*"]
)
class PromptRequest(BaseModel):