(20250120) Bug fix during mail retrieval.
This commit is contained in:
+10
-1
@@ -249,10 +249,19 @@ class LLMOutput(BaseModel):
|
||||
}
|
||||
|
||||
@property
|
||||
def json(self) -> dict | list | None:
|
||||
def output_json(self) -> dict | list | None:
|
||||
try: return json.from_string(self.output)
|
||||
except: return None
|
||||
|
||||
# ┏┓ ┏┓
|
||||
# ┃ ┓┏┏╋┏┓┏┳┓ ┣ ┓┏┏┓┏┏
|
||||
# ┗┛┗┻┛┗┗┛┛┗┗ ┻ ┗┻┛┗┗┛
|
||||
|
||||
def to_json(self) -> dict:
|
||||
dump = self.model_dump()
|
||||
dump["invocationId"] = str(dump["invocationId"])
|
||||
return dump
|
||||
|
||||
|
||||
# *****************************************************************************************************************
|
||||
# ***** ****
|
||||
|
||||
@@ -221,7 +221,7 @@ class CoreMessageModel(BaseModel):
|
||||
"chat": self.chat,
|
||||
"message": self.message,
|
||||
"snippet": self.snippet,
|
||||
"aiSnippet": self.aiSnippet,
|
||||
"aiSnippet": self.aiSnippet.to_json() if isinstance(self.aiSnippet, LLMOutput) else self.aiSnippet,
|
||||
"tags": self.tags
|
||||
}
|
||||
|
||||
@@ -240,7 +240,7 @@ class CoreMessageModel(BaseModel):
|
||||
"sender": self.sender,
|
||||
"chat": self.chat,
|
||||
"snippet": self.snippet,
|
||||
"aiSnippet": self.aiSnippet,
|
||||
"aiSnippet": self.aiSnippet.to_json() if isinstance(self.aiSnippet, LLMOutput) else self.aiSnippet,
|
||||
"tags": self.tags
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user