update api.py

This commit is contained in:
imClumsyPanda 2023-05-13 11:01:41 +08:00
parent 37b2fb18b7
commit ff8182f49a
1 changed files with 1 additions and 4 deletions

5
api.py
View File

@ -2,15 +2,12 @@ import argparse
import json
import os
import shutil
import subprocess
import tempfile
from typing import List, Optional
import nltk
import pydantic
import uvicorn
from fastapi import Body, FastAPI, File, Form, Query, UploadFile, WebSocket
from fastapi.openapi.utils import get_openapi
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from typing_extensions import Annotated
@ -144,7 +141,7 @@ async def upload_files(
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:
local_doc_folder = get_folder_path(knowledge_base_id)