AnyLlmWorkspaceUpdate

Executes the action taken on a workspace listed in the grid field. "Current" makes the workspace the active workspace and any workspace setting changes are updated in the LLM.

Type

python

Pack

AnythingLLM

Source

import demistomock as demisto  # noqa: F401
from CommonServerPython import *  # noqa: F401


def UpdateEmbeddings(workspace: str):
    # Had to duplicate automation since unable to execute_command with it
    wrk = execute_command("anyllm-workspace-get", {"workspace": workspace})
    embedds = []

    for embed in wrk["workspace"][0]["documents"]:
        gridrow = {"action": "", "title": json.loads(embed["metadata"])["title"], "pinned": embed["pinned"]}
        embedds.append(gridrow)

    grid = json.dumps({"anythingllmembeddings": embedds})
    execute_command("setIncident", {"customFields": grid, "version": -1})


def UpdateConversation(workspace: str):
    inci = demisto.incident()["CustomFields"]
    t = inci.get("anythingllmcurthread", "")
    threads = {}
    if t != "":
        threads = json.loads(t)
    thread = threads.get(workspace, "")

    if thread != "":
        response = execute_command(
            "anyllm-workspace-thread-chats", {"workspace": workspace, "thread": thread}, extract_contents=True
        )
        convo = ""
        for h in response.get("history", []):
            if h["role"] == "user":
                tm = datetime.fromtimestamp(h["sentAt"]).strftime("%Y %B %d %I:%M%p")
                content = h["content"].replace("\n", "")
                convo += f"##### {tm} |||MODE||| {content}\n"
            else:
                convo = convo.replace("|||MODE|||", f"[{h['type']}]:")
                convo += h["content"] + "\n"
                convo += "\n**Embedded Chunks Used**\n"
                for s in h["sources"]:
                    convo += f"* {s['score']:0.2f},  {s['title']}\n"
        execute_command("setIncident", {"customFields": {"anythingllmconversation": convo}})
    else:
        execute_command("setIncident", {"customFields": {"anythingllmconversation": ""}})


def main():
    try:
        args = demisto.args()
        oldgrid = json.loads(args.get("old", ""))
        newgrid = json.loads(args.get("new", ""))
        if len(oldgrid) == 0 or len(newgrid) == 0:
            return

        workspace = ""

        for old, new in zip(oldgrid, newgrid):
            if "action" in new and new["action"] == "Current" and old["action"] != "Current":
                workspace = new["name"]
                execute_command("setIncident", {"customFields": {"anythingllmworkspace": workspace}})
                break

        index = 0
        updated = False
        for old, new in zip(oldgrid, newgrid):
            if old == new:
                newgrid[index]["action"] = ""
            else:
                workspace = new["name"]
                settings = {
                    "openAiTemp": new["temperature"],  # 0.0 .. 1.0
                    "similarityThreshold": new["similarity"],  # vector DB similarity (0.0, 0.25, 0.50, 0.75)
                    "topN": new["topnresults"],  # top N similar results to return to chat context (1 - 12)
                }
                execute_command("anyllm-workspace-settings", {"workspace": workspace, "settings": settings})
            index += 1
            updated = True

        if workspace != "" and updated:
            grid = json.dumps({"anythingllmworkspacelist": newgrid})
            execute_command("setIncident", {"customFields": grid, "version": -1})
            UpdateEmbeddings(workspace)
            UpdateConversation(workspace)
    except Exception as ex:
        demisto.error(traceback.format_exc())
        return_error(f"AnyLlmWorkspaceUpdate: error is - {ex}")


if __name__ in ("__main__", "__builtin__", "builtins"):
    main()

README