From ff8182f49a016f204999f945bae2193111ec7fbd Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Sat, 13 May 2023 11:01:41 +0800 Subject: [PATCH] update api.py --- api.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/api.py b/api.py index 9c93c0e5..00d86005 100644 --- a/api.py +++ b/api.py @@ -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)