增加代码知识库;修复文档处理内容;增加API设置

This commit is contained in:
2026-05-16 20:20:10 +08:00
parent 69b49d28b2
commit 7aa3ce3294
119 changed files with 182273 additions and 793 deletions

View File

@@ -9,6 +9,7 @@ from app.db.session import get_db
from app.core.security import get_api_key_user
from app.core.config import settings
from app.services.embedding.embedding_factory import EmbeddingsFactory
from app.services.model_config import ModelConfigService
router = APIRouter()
@@ -36,7 +37,11 @@ def query_knowledge_base(
detail=f"Knowledge base {knowledge_base_id} not found",
)
embeddings = EmbeddingsFactory.create()
try:
model_profile = ModelConfigService.require_active_config(db, current_user.id)
except ValueError as exc:
raise HTTPException(status_code=400, detail=str(exc)) from exc
embeddings = EmbeddingsFactory.create(model_profile=model_profile)
vector_store = VectorStoreFactory.create(
store_type=settings.VECTOR_STORE_TYPE,
@@ -56,5 +61,7 @@ def query_knowledge_base(
return {"results": response}
except HTTPException:
raise
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
raise HTTPException(status_code=500, detail=str(e))