(input, command, results, api_key, api_type, api_endpoint)
| 718 | |
| 719 | |
| 720 | def run_task(input, command, results, api_key, api_type, api_endpoint): |
| 721 | id = command["id"] |
| 722 | args = command["args"] |
| 723 | task = command["task"] |
| 724 | deps = command["dep"] |
| 725 | if deps[0] != -1: |
| 726 | dep_tasks = [results[dep] for dep in deps] |
| 727 | else: |
| 728 | dep_tasks = [] |
| 729 | |
| 730 | logger.debug(f"Run task: {id} - {task}") |
| 731 | logger.debug("Deps: " + json.dumps(dep_tasks)) |
| 732 | |
| 733 | if deps[0] != -1: |
| 734 | if "image" in args and "<GENERATED>-" in args["image"]: |
| 735 | resource_id = int(args["image"].split("-")[1]) |
| 736 | if "generated image" in results[resource_id]["inference result"]: |
| 737 | args["image"] = results[resource_id]["inference result"]["generated image"] |
| 738 | if "audio" in args and "<GENERATED>-" in args["audio"]: |
| 739 | resource_id = int(args["audio"].split("-")[1]) |
| 740 | if "generated audio" in results[resource_id]["inference result"]: |
| 741 | args["audio"] = results[resource_id]["inference result"]["generated audio"] |
| 742 | if "text" in args and "<GENERATED>-" in args["text"]: |
| 743 | resource_id = int(args["text"].split("-")[1]) |
| 744 | if "generated text" in results[resource_id]["inference result"]: |
| 745 | args["text"] = results[resource_id]["inference result"]["generated text"] |
| 746 | |
| 747 | text = image = audio = None |
| 748 | for dep_task in dep_tasks: |
| 749 | if "generated text" in dep_task["inference result"]: |
| 750 | text = dep_task["inference result"]["generated text"] |
| 751 | logger.debug("Detect the generated text of dependency task (from results):" + text) |
| 752 | elif "text" in dep_task["task"]["args"]: |
| 753 | text = dep_task["task"]["args"]["text"] |
| 754 | logger.debug("Detect the text of dependency task (from args): " + text) |
| 755 | if "generated image" in dep_task["inference result"]: |
| 756 | image = dep_task["inference result"]["generated image"] |
| 757 | logger.debug("Detect the generated image of dependency task (from results): " + image) |
| 758 | elif "image" in dep_task["task"]["args"]: |
| 759 | image = dep_task["task"]["args"]["image"] |
| 760 | logger.debug("Detect the image of dependency task (from args): " + image) |
| 761 | if "generated audio" in dep_task["inference result"]: |
| 762 | audio = dep_task["inference result"]["generated audio"] |
| 763 | logger.debug("Detect the generated audio of dependency task (from results): " + audio) |
| 764 | elif "audio" in dep_task["task"]["args"]: |
| 765 | audio = dep_task["task"]["args"]["audio"] |
| 766 | logger.debug("Detect the audio of dependency task (from args): " + audio) |
| 767 | |
| 768 | if "image" in args and "<GENERATED>" in args["image"]: |
| 769 | if image: |
| 770 | args["image"] = image |
| 771 | if "audio" in args and "<GENERATED>" in args["audio"]: |
| 772 | if audio: |
| 773 | args["audio"] = audio |
| 774 | if "text" in args and "<GENERATED>" in args["text"]: |
| 775 | if text: |
| 776 | args["text"] = text |
| 777 |
nothing calls this directly
no test coverage detected