(20250120) Bug fix during mail retrieval.

This commit is contained in:
2025-01-20 14:56:50 +05:30
parent f6636afe59
commit 06058238e2
2 changed files with 12 additions and 3 deletions
+10 -1
View File
@@ -249,10 +249,19 @@ class LLMOutput(BaseModel):
} }
@property @property
def json(self) -> dict | list | None: def output_json(self) -> dict | list | None:
try: return json.from_string(self.output) try: return json.from_string(self.output)
except: return None except: return None
# ┏┓ ┏┓
# ┃ ┓┏┏╋┏┓┏┳┓ ┣ ┓┏┏┓┏┏
# ┗┛┗┻┛┗┗┛┛┗┗ ┻ ┗┻┛┗┗┛
def to_json(self) -> dict:
dump = self.model_dump()
dump["invocationId"] = str(dump["invocationId"])
return dump
# ***************************************************************************************************************** # *****************************************************************************************************************
# ***** **** # ***** ****
+2 -2
View File
@@ -221,7 +221,7 @@ class CoreMessageModel(BaseModel):
"chat": self.chat, "chat": self.chat,
"message": self.message, "message": self.message,
"snippet": self.snippet, "snippet": self.snippet,
"aiSnippet": self.aiSnippet, "aiSnippet": self.aiSnippet.to_json() if isinstance(self.aiSnippet, LLMOutput) else self.aiSnippet,
"tags": self.tags "tags": self.tags
} }
@@ -240,7 +240,7 @@ class CoreMessageModel(BaseModel):
"sender": self.sender, "sender": self.sender,
"chat": self.chat, "chat": self.chat,
"snippet": self.snippet, "snippet": self.snippet,
"aiSnippet": self.aiSnippet, "aiSnippet": self.aiSnippet.to_json() if isinstance(self.aiSnippet, LLMOutput) else self.aiSnippet,
"tags": self.tags "tags": self.tags
} }