235 lines
13 KiB
Python
235 lines
13 KiB
Python
# Copyright (c) 2025 Patrick Motsch
|
|
# AI node definitions - map to methodAi actions.
|
|
|
|
from modules.shared.i18nRegistry import t
|
|
|
|
_AI_COMMON_PARAMS = [
|
|
{"name": "requireNeutralization", "type": "bool", "required": False,
|
|
"frontendType": "checkbox", "default": False,
|
|
"description": t("Eingaben fuer diesen Call neutralisieren")},
|
|
{"name": "allowedModels", "type": "array", "required": False,
|
|
"frontendType": "modelMultiSelect", "default": [],
|
|
"description": t("Erlaubte LLM-Modelle (leer = alle erlaubten)")},
|
|
]
|
|
|
|
AI_NODES = [
|
|
{
|
|
"id": "ai.prompt",
|
|
"category": "ai",
|
|
"label": t("Prompt"),
|
|
"description": t("Prompt eingeben und KI führt aus"),
|
|
"parameters": [
|
|
{"name": "aiPrompt", "type": "str", "required": True, "frontendType": "templateTextarea",
|
|
"description": t("KI-Prompt")},
|
|
{"name": "resultType", "type": "str", "required": False, "frontendType": "select",
|
|
"frontendOptions": {"options": ["txt", "json", "md", "csv", "xml", "html", "pdf", "docx", "xlsx", "pptx", "png", "jpg"]},
|
|
"description": t("Ausgabeformat"), "default": "txt"},
|
|
{"name": "documentList", "type": "DocumentList", "required": False, "frontendType": "hidden",
|
|
"description": t("Dokumente aus vorherigen Schritten"), "default": "",
|
|
"graphInherit": {"port": 0, "kind": "documentListWire"}},
|
|
{"name": "context", "type": "Any", "required": False, "frontendType": "contextBuilder",
|
|
"description": t("Daten aus vorherigen Schritten"), "default": "",
|
|
"graphInherit": {"port": 0, "kind": "primaryTextRef"}},
|
|
{"name": "simpleMode", "type": "bool", "required": False, "frontendType": "checkbox",
|
|
"description": t("Einfacher Modus"), "default": True},
|
|
] + _AI_COMMON_PARAMS,
|
|
"inputs": 1,
|
|
"outputs": 1,
|
|
"inputPorts": {0: {"accepts": [
|
|
"FormPayload", "DocumentList", "AiResult", "TextResult", "Transit", "LoopItem", "ActionResult",
|
|
]}},
|
|
"outputPorts": {0: {"schema": "AiResult"}},
|
|
"meta": {"icon": "mdi-robot", "color": "#9C27B0", "usesAi": True},
|
|
"_method": "ai",
|
|
"_action": "process",
|
|
},
|
|
{
|
|
"id": "ai.webResearch",
|
|
"category": "ai",
|
|
"label": t("Web-Recherche"),
|
|
"description": t("Recherche im Web"),
|
|
"parameters": [
|
|
{"name": "prompt", "type": "str", "required": True, "frontendType": "textarea",
|
|
"description": t("Recherche-Anfrage")},
|
|
{"name": "context", "type": "Any", "required": False, "frontendType": "contextBuilder",
|
|
"description": t("Daten aus vorherigen Schritten"), "default": "",
|
|
"graphInherit": {"port": 0, "kind": "primaryTextRef"}},
|
|
{"name": "documentList", "type": "DocumentList", "required": False, "frontendType": "hidden",
|
|
"description": t("Dokumente aus vorherigen Schritten"), "default": "",
|
|
"graphInherit": {"port": 0, "kind": "documentListWire"}},
|
|
] + _AI_COMMON_PARAMS,
|
|
"inputs": 1,
|
|
"outputs": 1,
|
|
"inputPorts": {0: {"accepts": [
|
|
"FormPayload", "Transit", "AiResult", "DocumentList", "ActionResult", "LoopItem", "TextResult",
|
|
]}},
|
|
"outputPorts": {0: {"schema": "AiResult"}},
|
|
"meta": {"icon": "mdi-magnify", "color": "#9C27B0", "usesAi": True},
|
|
"_method": "ai",
|
|
"_action": "webResearch",
|
|
},
|
|
{
|
|
"id": "ai.summarizeDocument",
|
|
"category": "ai",
|
|
"label": t("Dokument zusammenfassen"),
|
|
"description": t("Dokumentinhalt zusammenfassen"),
|
|
"parameters": [
|
|
{"name": "documentList", "type": "DocumentList", "required": True, "frontendType": "dataRef",
|
|
"description": t("Dokumente aus vorherigen Schritten"),
|
|
"graphInherit": {"port": 0, "kind": "documentListWire"}},
|
|
{"name": "resultType", "type": "str", "required": False, "frontendType": "select",
|
|
"frontendOptions": {"options": ["txt", "json", "md", "csv", "xml", "html", "pdf", "docx", "xlsx", "pptx", "png", "jpg"]},
|
|
"description": t("Ausgabeformat"), "default": "txt"},
|
|
{"name": "summaryLength", "type": "str", "required": False, "frontendType": "select",
|
|
"frontendOptions": {"options": ["brief", "medium", "detailed"]},
|
|
"description": t("Kurz, mittel oder ausführlich"), "default": "medium"},
|
|
{"name": "folderId", "type": "str", "required": False, "frontendType": "userFileFolder",
|
|
"description": t("Zielordner in Meine Dateien"),
|
|
"default": ""},
|
|
] + _AI_COMMON_PARAMS,
|
|
"inputs": 1,
|
|
"outputs": 1,
|
|
"inputPorts": {0: {"accepts": ["DocumentList", "Transit", "LoopItem"]}},
|
|
"outputPorts": {0: {"schema": "AiResult"}},
|
|
"meta": {"icon": "mdi-file-document-outline", "color": "#9C27B0", "usesAi": True},
|
|
"_method": "ai",
|
|
"_action": "summarizeDocument",
|
|
},
|
|
{
|
|
"id": "ai.translateDocument",
|
|
"category": "ai",
|
|
"label": t("Dokument übersetzen"),
|
|
"description": t("Dokument in Zielsprache übersetzen"),
|
|
"parameters": [
|
|
{"name": "documentList", "type": "DocumentList", "required": True, "frontendType": "dataRef",
|
|
"description": t("Dokumente aus vorherigen Schritten"),
|
|
"graphInherit": {"port": 0, "kind": "documentListWire"}},
|
|
{"name": "resultType", "type": "str", "required": False, "frontendType": "select",
|
|
"frontendOptions": {"options": ["txt", "json", "md", "csv", "xml", "html", "pdf", "docx", "xlsx", "pptx", "png", "jpg"]},
|
|
"description": t("Ausgabeformat"), "default": "txt"},
|
|
{"name": "targetLanguage", "type": "str", "required": True, "frontendType": "text",
|
|
"description": t("Zielsprache (z.B. de, en, French)")},
|
|
{"name": "folderId", "type": "str", "required": False, "frontendType": "userFileFolder",
|
|
"description": t("Zielordner in Meine Dateien"),
|
|
"default": ""},
|
|
] + _AI_COMMON_PARAMS,
|
|
"inputs": 1,
|
|
"outputs": 1,
|
|
"inputPorts": {0: {"accepts": ["DocumentList", "Transit", "LoopItem"]}},
|
|
"outputPorts": {0: {"schema": "AiResult"}},
|
|
"meta": {"icon": "mdi-translate", "color": "#9C27B0", "usesAi": True},
|
|
"_method": "ai",
|
|
"_action": "translateDocument",
|
|
},
|
|
{
|
|
"id": "ai.convertDocument",
|
|
"category": "ai",
|
|
"label": t("Dokument konvertieren"),
|
|
"description": t("Dokument in anderes Format konvertieren"),
|
|
"parameters": [
|
|
{"name": "documentList", "type": "DocumentList", "required": True, "frontendType": "dataRef",
|
|
"description": t("Dokumente aus vorherigen Schritten"),
|
|
"graphInherit": {"port": 0, "kind": "documentListWire"}},
|
|
{"name": "targetFormat", "type": "str", "required": True, "frontendType": "select",
|
|
"frontendOptions": {"options": ["docx", "pdf", "xlsx", "csv", "txt", "html", "json", "md"]},
|
|
"description": t("Zielformat")},
|
|
{"name": "folderId", "type": "str", "required": False, "frontendType": "userFileFolder",
|
|
"description": t("Zielordner in Meine Dateien"),
|
|
"default": ""},
|
|
] + _AI_COMMON_PARAMS,
|
|
"inputs": 1,
|
|
"outputs": 1,
|
|
"inputPorts": {0: {"accepts": ["DocumentList", "Transit", "LoopItem"]}},
|
|
"outputPorts": {0: {"schema": "DocumentList"}},
|
|
"meta": {"icon": "mdi-file-convert", "color": "#9C27B0", "usesAi": True},
|
|
"_method": "ai",
|
|
"_action": "convertDocument",
|
|
},
|
|
{
|
|
"id": "ai.generateDocument",
|
|
"category": "ai",
|
|
"label": t("Dokument generieren"),
|
|
"description": t("Dokument aus Prompt generieren"),
|
|
"parameters": [
|
|
{"name": "prompt", "type": "str", "required": True, "frontendType": "textarea",
|
|
"description": t("Generierungs-Prompt")},
|
|
{"name": "outputFormat", "type": "str", "required": False, "frontendType": "select",
|
|
"frontendOptions": {"options": ["docx", "pdf", "txt", "html", "md"]},
|
|
"description": t("Ausgabeformat"), "default": "docx"},
|
|
{"name": "title", "type": "str", "required": False, "frontendType": "text",
|
|
"description": t("Dokumenttitel (Metadaten / Dateiname)"), "default": ""},
|
|
{"name": "documentType", "type": "str", "required": False, "frontendType": "select",
|
|
"frontendOptions": {"options": ["letter", "memo", "proposal", "contract", "report", "email"]},
|
|
"description": t("Dokumentart (Inhaltshinweis fuer die KI)"), "default": "proposal"},
|
|
{"name": "folderId", "type": "str", "required": False, "frontendType": "userFileFolder",
|
|
"description": t("Zielordner in Meine Dateien"),
|
|
"default": ""},
|
|
{"name": "context", "type": "Any", "required": False, "frontendType": "contextBuilder",
|
|
"description": t("Daten aus vorherigen Schritten"), "default": "",
|
|
"graphInherit": {"port": 0, "kind": "primaryTextRef"}},
|
|
{"name": "documentList", "type": "DocumentList", "required": False, "frontendType": "hidden",
|
|
"description": t("Dokumente aus vorherigen Schritten"), "default": "",
|
|
"graphInherit": {"port": 0, "kind": "documentListWire"}},
|
|
] + _AI_COMMON_PARAMS,
|
|
"inputs": 1,
|
|
"outputs": 1,
|
|
"inputPorts": {0: {"accepts": [
|
|
"FormPayload", "Transit", "AiResult", "DocumentList", "ActionResult", "LoopItem", "TextResult",
|
|
]}},
|
|
"outputPorts": {0: {"schema": "DocumentList"}},
|
|
"meta": {"icon": "mdi-file-plus", "color": "#9C27B0", "usesAi": True},
|
|
"_method": "ai",
|
|
"_action": "generateDocument",
|
|
},
|
|
{
|
|
"id": "ai.generateCode",
|
|
"category": "ai",
|
|
"label": t("Code generieren"),
|
|
"description": t("Code aus Beschreibung generieren"),
|
|
"parameters": [
|
|
{"name": "prompt", "type": "str", "required": True, "frontendType": "textarea",
|
|
"description": t("Code-Generierungs-Prompt")},
|
|
{"name": "resultType", "type": "str", "required": False, "frontendType": "select",
|
|
"frontendOptions": {"options": ["py", "js", "ts", "html", "java", "cpp", "txt", "json", "csv", "xml"]},
|
|
"description": t("Datei-Endung der erzeugten Code-Datei"), "default": "py"},
|
|
{"name": "folderId", "type": "str", "required": False, "frontendType": "userFileFolder",
|
|
"description": t("Zielordner in Meine Dateien"),
|
|
"default": ""},
|
|
{"name": "context", "type": "Any", "required": False, "frontendType": "contextBuilder",
|
|
"description": t("Daten aus vorherigen Schritten"), "default": "",
|
|
"graphInherit": {"port": 0, "kind": "primaryTextRef"}},
|
|
{"name": "documentList", "type": "DocumentList", "required": False, "frontendType": "hidden",
|
|
"description": t("Dokumente aus vorherigen Schritten"), "default": "",
|
|
"graphInherit": {"port": 0, "kind": "documentListWire"}},
|
|
] + _AI_COMMON_PARAMS,
|
|
"inputs": 1,
|
|
"outputs": 1,
|
|
"inputPorts": {0: {"accepts": [
|
|
"FormPayload", "Transit", "AiResult", "DocumentList", "ActionResult", "LoopItem", "TextResult",
|
|
]}},
|
|
"outputPorts": {0: {"schema": "AiResult"}},
|
|
"meta": {"icon": "mdi-code-tags", "color": "#9C27B0", "usesAi": True},
|
|
"_method": "ai",
|
|
"_action": "generateCode",
|
|
},
|
|
{
|
|
"id": "ai.consolidate",
|
|
"category": "ai",
|
|
"label": t("KI-Konsolidierung"),
|
|
"description": t("Gesammelte Ergebnisse mit KI zusammenfassen, klassifizieren oder semantisch zusammenführen"),
|
|
"parameters": [
|
|
{"name": "mode", "type": "str", "required": False, "frontendType": "select",
|
|
"frontendOptions": {"options": ["summarize", "classify", "semanticMerge"]},
|
|
"description": t("Konsolidierungsmodus"), "default": "summarize"},
|
|
{"name": "prompt", "type": "str", "required": False, "frontendType": "textarea",
|
|
"description": t("Optionaler Prompt für die Konsolidierung"), "default": ""},
|
|
] + _AI_COMMON_PARAMS,
|
|
"inputs": 1,
|
|
"outputs": 1,
|
|
"inputPorts": {0: {"accepts": ["AggregateResult", "Transit"]}},
|
|
"outputPorts": {0: {"schema": "ConsolidateResult"}},
|
|
"meta": {"icon": "mdi-table-merge-cells", "color": "#9C27B0", "usesAi": True},
|
|
"_method": "ai",
|
|
"_action": "consolidate",
|
|
},
|
|
]
|