(20250120) Bug fix during mail retrieval.
This commit is contained in:
+10
-1
@@ -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
|
||||||
|
|
||||||
|
|
||||||
# *****************************************************************************************************************
|
# *****************************************************************************************************************
|
||||||
# ***** ****
|
# ***** ****
|
||||||
|
|||||||
@@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user