增加代码知识库;修复文档处理内容;增加API设置
This commit is contained in:
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user