fixes trustee data
This commit is contained in:
parent
c3f8be68b2
commit
f89eacfc47
1 changed files with 15 additions and 5 deletions
|
|
@ -2528,6 +2528,8 @@ def _registerCoreTools(registry: ToolRegistry, services):
|
||||||
try:
|
try:
|
||||||
import matplotlib
|
import matplotlib
|
||||||
matplotlib.use("Agg")
|
matplotlib.use("Agg")
|
||||||
|
import logging as _mpllog
|
||||||
|
_mpllog.getLogger("matplotlib").setLevel(_mpllog.WARNING)
|
||||||
import matplotlib.pyplot as plt
|
import matplotlib.pyplot as plt
|
||||||
import io
|
import io
|
||||||
|
|
||||||
|
|
@ -2607,11 +2609,19 @@ def _registerCoreTools(registry: ToolRegistry, services):
|
||||||
chatService = services.chat
|
chatService = services.chat
|
||||||
sanitizedTitle = _re.sub(r'[^\w._-]', '_', title, flags=_re.UNICODE).strip('_') or "chart"
|
sanitizedTitle = _re.sub(r'[^\w._-]', '_', title, flags=_re.UNICODE).strip('_') or "chart"
|
||||||
fileName = f"{sanitizedTitle}.png"
|
fileName = f"{sanitizedTitle}.png"
|
||||||
fid = chatService.saveFile(
|
|
||||||
fileName=fileName,
|
if hasattr(chatService.interfaceDbComponent, "saveGeneratedFile"):
|
||||||
fileData=pngData,
|
fileItem = chatService.interfaceDbComponent.saveGeneratedFile(pngData, fileName, "image/png")
|
||||||
mimeType="image/png",
|
else:
|
||||||
)
|
fileItem, _ = chatService.interfaceDbComponent.saveUploadedFile(pngData, fileName)
|
||||||
|
|
||||||
|
fid = fileItem.id if hasattr(fileItem, "id") else fileItem.get("id", "?") if isinstance(fileItem, dict) else "?"
|
||||||
|
fiId = context.get("featureInstanceId") or (services.featureInstanceId if services else "")
|
||||||
|
if fiId and fid != "?":
|
||||||
|
chatService.interfaceDbComponent.updateFile(fid, {"featureInstanceId": fiId})
|
||||||
|
tempFolderId = _getOrCreateTempFolder(chatService)
|
||||||
|
if tempFolderId and fid != "?":
|
||||||
|
chatService.interfaceDbComponent.updateFile(fid, {"folderId": tempFolderId})
|
||||||
|
|
||||||
sideEvents = [{"type": "fileCreated", "data": {
|
sideEvents = [{"type": "fileCreated", "data": {
|
||||||
"fileId": fid, "fileName": fileName,
|
"fileId": fid, "fileName": fileName,
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue