apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-07-13 11:22:14 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix: add validation for URL and params

Browse source
This commit is contained in:
Abhishek Kumar 2026-06-02 12:45:13 +05:30
parent 858c474139
commit 7a3b1c4a4b
11 changed files with 327 additions and 129 deletions
Download patch file Download diff file Expand all files Collapse all files

9
api/routes/knowledge_base.py
View file

@ -370,7 +370,10 @@ async def search_chunks(
try:
# Import here to avoid circular dependency
from api.services.configuration.registry import ServiceProviders
from api.services.gen_ai import AzureOpenAIEmbeddingService, OpenAIEmbeddingService
from api.services.gen_ai import (
AzureOpenAIEmbeddingService,
OpenAIEmbeddingService,
)
# Try to get user's embeddings configuration
user_config = await db_client.get_user_configurations(user.id)
@ -385,7 +388,9 @@ async def search_chunks(
embeddings_model = user_config.embeddings.model
embeddings_provider = getattr(user_config.embeddings, "provider", None)
embeddings_endpoint = getattr(user_config.embeddings, "endpoint", None)
embeddings_api_version = getattr(user_config.embeddings, "api_version", None)
embeddings_api_version = getattr(
user_config.embeddings, "api_version", None
)
# Initialize embedding service based on provider
if embeddings_provider == ServiceProviders.AZURE.value and embeddings_endpoint: