mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-08 05:39:13 +00:00
Fix sync to async issue when getting all valid vision configs
This commit is contained in:
@@ -761,6 +761,10 @@ class ConversationAdapters:
|
|||||||
def get_all_conversation_configs():
|
def get_all_conversation_configs():
|
||||||
return ChatModelOptions.objects.all()
|
return ChatModelOptions.objects.all()
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def aget_all_conversation_configs():
|
||||||
|
return await sync_to_async(list)(ChatModelOptions.objects.all())
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def get_vision_enabled_config():
|
def get_vision_enabled_config():
|
||||||
conversation_configurations = ConversationAdapters.get_all_conversation_configs()
|
conversation_configurations = ConversationAdapters.get_all_conversation_configs()
|
||||||
@@ -769,6 +773,14 @@ class ConversationAdapters:
|
|||||||
return config
|
return config
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def aget_vision_enabled_config():
|
||||||
|
conversation_configurations = await ConversationAdapters.aget_all_conversation_configs()
|
||||||
|
for config in conversation_configurations:
|
||||||
|
if config.vision_enabled:
|
||||||
|
return config
|
||||||
|
return None
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def get_openai_conversation_config():
|
def get_openai_conversation_config():
|
||||||
return OpenAIProcessorConversationConfig.objects.filter().first()
|
return OpenAIProcessorConversationConfig.objects.filter().first()
|
||||||
|
|||||||
@@ -595,7 +595,7 @@ async def send_message_to_model_wrapper(
|
|||||||
|
|
||||||
vision_available = conversation_config.vision_enabled
|
vision_available = conversation_config.vision_enabled
|
||||||
if not vision_available and uploaded_image_url:
|
if not vision_available and uploaded_image_url:
|
||||||
vision_enabled_config = ConversationAdapters.get_vision_enabled_config()
|
vision_enabled_config = await ConversationAdapters.aget_vision_enabled_config()
|
||||||
if vision_enabled_config:
|
if vision_enabled_config:
|
||||||
conversation_config = vision_enabled_config
|
conversation_config = vision_enabled_config
|
||||||
vision_available = True
|
vision_available = True
|
||||||
|
|||||||
Reference in New Issue
Block a user