AnyLlmUploadDocument

Uploads pre-processed text data to the LLM's document catalog. The document data must have been prepared by AnyLlmUploadText, AnyLlmUploadFileEntry, AnyLlmUploadResults. Text documentation processed and uploaded by this method can also be full text searched.

Type

python

Pack

AnythingLLM

Source

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


def main():
    try:
        llmupload = demisto.incident()["CustomFields"].get("anythingllmupload", "").strip()
        if llmupload == "":
            raise Exception("No processed document found in anythingllmupload incident field ready for upload")

        args = json.loads(llmupload)
        title = args.get("title", "")
        files = demisto.context().get("File", "")
        if files == "":
            raise Exception(f"No File key found in context for document title [{title}]")

        if isinstance(files, dict):
            files = [files]

        for ff in files:
            if ff["Name"] == title:
                args["title"] = f"{ff['EntryID']}_{title}"
                execute_command("anyllm-document-upload-text", args)
                execute_command("setIncident", {"customFields": {"anythingllmupload": ""}, "version": -1})
                return

        raise Exception(f"Processed document {title} not found in File context key")
    except Exception as ex:
        demisto.error(traceback.format_exc())
        return_error(f"AnyLlmUploadDocument: error is - {ex}")


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

README