(20250120) Mail retrieval bug fixed.

This commit is contained in:
2025-01-20 15:11:01 +05:30
parent 06058238e2
commit 54f62bc986
2 changed files with 9 additions and 2 deletions
+1
View File
@@ -259,6 +259,7 @@ class LLMOutput(BaseModel):
def to_json(self) -> dict: def to_json(self) -> dict:
dump = self.model_dump() dump = self.model_dump()
print("DUMP:", dump)
dump["invocationId"] = str(dump["invocationId"]) dump["invocationId"] = str(dump["invocationId"])
return dump return dump
+8 -2
View File
@@ -207,7 +207,7 @@ class CoreMessageModel(BaseModel):
@property @property
def full(self): def full(self):
return { full_json = {
"messageId": str(self.messageId), "messageId": str(self.messageId),
"ts": self.ts.isoformat(), "ts": self.ts.isoformat(),
"serviceType": self.serviceType, "serviceType": self.serviceType,
@@ -224,10 +224,13 @@ class CoreMessageModel(BaseModel):
"aiSnippet": self.aiSnippet.to_json() if isinstance(self.aiSnippet, LLMOutput) else self.aiSnippet, "aiSnippet": self.aiSnippet.to_json() if isinstance(self.aiSnippet, LLMOutput) else self.aiSnippet,
"tags": self.tags "tags": self.tags
} }
if full_json["aiSnippet"]:
full_json["aiSnippet"]["invocationId"] = str(full_json["aiSnippet"]["invocationId"])
return full_json
@property @property
def preview(self): def preview(self):
return { preview_json = {
"messageId": str(self.messageId), "messageId": str(self.messageId),
"ts": self.ts.isoformat(), "ts": self.ts.isoformat(),
"serviceType": self.serviceType, "serviceType": self.serviceType,
@@ -243,6 +246,9 @@ class CoreMessageModel(BaseModel):
"aiSnippet": self.aiSnippet.to_json() if isinstance(self.aiSnippet, LLMOutput) else self.aiSnippet, "aiSnippet": self.aiSnippet.to_json() if isinstance(self.aiSnippet, LLMOutput) else self.aiSnippet,
"tags": self.tags "tags": self.tags
} }
if preview_json["aiSnippet"]:
preview_json["aiSnippet"]["invocationId"] = str(preview_json["aiSnippet"]["invocationId"])
return preview_json
# ┓┏ ┓• ┓ • # ┓┏ ┓• ┓ •
# ┃┃┏┓┃┓┏┫┏┓╋┓┏┓┏┓ # ┃┃┏┓┃┓┏┫┏┓╋┓┏┓┏┓