mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-02 21:19:12 +00:00
Make URI field optional for now to handle previously saved documents
For files not synced after the previous release, context uri is unset. This results in failure to save chat messages that retrieve documents as the uri field cannot be unset so pre save validation fails. We'd use a db migration to handle this but this is a quick mitigation for now.
This commit is contained in:
@@ -23,7 +23,7 @@ logger = logging.getLogger(__name__)
|
||||
class Context(PydanticBaseModel):
|
||||
compiled: str
|
||||
file: str
|
||||
uri: str
|
||||
uri: Optional[str] = None
|
||||
query: Optional[str] = None
|
||||
|
||||
@model_validator(mode="after")
|
||||
|
||||
@@ -646,7 +646,11 @@ def generate_chatml_messages_with_context(
|
||||
|
||||
if not is_none_or_empty(chat.context):
|
||||
references = "\n\n".join(
|
||||
{f"# URI: {item.uri}\n## {item.compiled}\n" for item in chat.context or [] if isinstance(item, dict)}
|
||||
{
|
||||
f"# URI: {item.uri or item.file}\n## {item.compiled}\n"
|
||||
for item in chat.context or []
|
||||
if isinstance(item, dict)
|
||||
}
|
||||
)
|
||||
message_context += [{"type": "text", "text": f"{prompts.notes_conversation.format(references=references)}"}]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user