klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support constructing chat message as a list of dictionaries

Browse Source 
Research mode recently started passing iteration as list of message
content dicts. This change extends to storing it as is in DB.
This commit is contained in:
Debanjum
2025-05-21 22:57:56 -07:00
parent df9ab51fd0
commit 98b56316e4
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/khoj/processor/conversation/utils.py
View File

@@ -121,7 +121,7 @@ def construct_iteration_history(
index=idx + 1, index=idx + 1,
) )
previous_iterations_history.append(iteration_data) previous_iterations_history.append({"type": "text", "text": iteration_data})
return ( return (
[ [
@@ -341,7 +341,7 @@ Khoj: "{chat_response}"
def construct_structured_message( def construct_structured_message(
message: list[str] | str, message: list[dict] | str,
images: list[str], images: list[str],
model_type: str, model_type: str,
vision_enabled: bool, vision_enabled: bool,
@@ -355,11 +355,9 @@ def construct_structured_message(
ChatModel.ModelType.GOOGLE, ChatModel.ModelType.GOOGLE,
ChatModel.ModelType.ANTHROPIC, ChatModel.ModelType.ANTHROPIC,
]: ]:
message = [message] if isinstance(message, str) else message constructed_messages: List[dict[str, Any]] = (
[{"type": "text", "text": message}] if isinstance(message, str) else message
constructed_messages: List[dict[str, Any]] = [ )
{"type": "text", "text": message_part} for message_part in message
]
if not is_none_or_empty(attached_file_context): if not is_none_or_empty(attached_file_context):
constructed_messages.append({"type": "text", "text": attached_file_context}) constructed_messages.append({"type": "text", "text": attached_file_context})
@@ -368,6 +366,7 @@ def construct_structured_message(
constructed_messages.append({"type": "image_url", "image_url": {"url": image}}) constructed_messages.append({"type": "image_url", "image_url": {"url": image}})
return constructed_messages return constructed_messages
message = message if isinstance(message, str) else "\n\n".join(m["text"] for m in message)
if not is_none_or_empty(attached_file_context): if not is_none_or_empty(attached_file_context):
return f"{attached_file_context}\n\n{message}" return f"{attached_file_context}\n\n{message}"