weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update api.py

This commit is contained in:
imClumsyPanda 2023-05-13 11:02:27 +08:00
parent 6e022d0d45
commit 6e7078cf24
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api.py
View File

@ -2,15 +2,12 @@ import argparse
import json import json
import os import os
import shutil import shutil
import subprocess
import tempfile
from typing import List, Optional from typing import List, Optional
import nltk import nltk
import pydantic import pydantic
import uvicorn import uvicorn
from fastapi import Body, FastAPI, File, Form, Query, UploadFile, WebSocket from fastapi import Body, FastAPI, File, Form, Query, UploadFile, WebSocket
from fastapi.openapi.utils import get_openapi
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel from pydantic import BaseModel
from typing_extensions import Annotated from typing_extensions import Annotated
@ -144,7 +141,7 @@ async def upload_files(
async def list_docs( async def list_docs(
knowledge_base_id: Optional[str] = Query(description="Knowledge Base Name", example="kb1") knowledge_base_id: Optional[str] = Query(default=None, description="Knowledge Base Name", example="kb1")
): ):
if knowledge_base_id: if knowledge_base_id:
local_doc_folder = get_folder_path(knowledge_base_id) local_doc_folder = get_folder_path(knowledge_base_id)