import os import sys import yaml import json from pathlib import Path sys.path.append(str(Path('__file__').parent.parent)) from swarmai.__main__ import run_swarm """ Define some global parameters. This is a simple frontent for the swarm. The swarm has a config, the default output and entry-point. Default swarm config (for copilot =)): swarm: agents: # supported: manager, analyst, googler, crunchbase_searcher - type: manager n: 2 - type: analyst n: 2 - type: googler n: 2 - type: crunchbase_searcher # scraper can only have one job in parallel n: 1 timeout_min: 10 run_dir: ./tmp/swarm task: role: | professional venture capital agency, who has a proven track reckord of consistently funding successful startups global_goal: | A new startup just send us their pitch. Find if the startup is worth investing in. The startup is called Brainamics and it is in the space of brain computer interfaces. More information about them: 'https://brainamics.de', 'https://www.linkedin.com/company/thebrainamics/' goals: - Generate a comprehensive description of the startup. Describe their value proposition, the product, USP and business model of a startup. - Find any mentions of the startup in the news, social media, etc. Add links. - Find top 10 companies and startups in this field. Find out their locations, raised funding, value proposition, differentiation, etc. - Find top 5 investors in this field. Includ specific details in the format of 'company AAA (link) invested in company BBB (link) $XX in year YYYY' - Describe the market size, growth rate and trends of this field. - Main problems and challenges of the field. Create an extensive list of problems. What can stop the field from growing? What can stop the company from succeeding? - Briefly describe the technology for the non-tech audience. Include links to the main articles in the field. - What questions should we ask the startup to make a more informed decision? Avoid generic and obvious questions and focus on field/domain specific questions that can uncover problems with this specific startup. """ SWARM_CONFIG_PATH = "swarm_config.yaml" ALLOWED_AGENTS = ["manager", "analyst", "googler", "crunchbase_searcher"] SWARM_DEFAULT_RUN_FOLDER = (Path("__file__").parent / "tmp" / "swarm").resolve() SWARM_DEFAULT_JSON_OUTPUT = str(SWARM_DEFAULT_RUN_FOLDER / "output.json") SWARM_DEAFAULT_LOGS = str(SWARM_DEFAULT_RUN_FOLDER / "swarm.json") SWARM_DEFAULT_SHARED_MEMORY = str(SWARM_DEFAULT_RUN_FOLDER / "shared_memory") def get_swarm_config(): """ Load the swarm config from the default location. """ with open(SWARM_CONFIG_PATH) as f: swarm_config = yaml.load(f, Loader=yaml.FullLoader) return swarm_config def set_swarm_role(role_description): """ Set the role for the swarm. It's specified in the swarm_config.yaml file under: swarm.task.role """ if role_description=="": role_description = "professional venture capital agency, who has a proven track reckord of consistently funding successful startups" swarm_config = get_swarm_config() print(f"Setting role to: {role_description}") swarm_config["task"]["role"] = role_description with open(SWARM_CONFIG_PATH, "w") as f: yaml.dump(swarm_config, f) def get_swarm_role(): """ Get the role for the swarm. It's specified in the swarm_config.yaml file under: swarm.task.role """ swarm_config = get_swarm_config() return swarm_config["task"]["role"] def set_swarm_global_goal(global_goal): """ Set the global goal for the swarm. It's specified in the swarm_config.yaml file under: swarm.task.global_goal """ if global_goal=="": global_goal = "A new startup just send us their pitch. Find if the startup is worth investing in. The startup is called Brainamics and it is in the space of brain computer interfaces." swarm_config = get_swarm_config() print(f"Setting global goal to: {global_goal}") swarm_config["task"]["global_goal"] = global_goal with open(SWARM_CONFIG_PATH, "w") as f: yaml.dump(swarm_config, f) def get_swarm_global_goal(): """ Get the global goal for the swarm. It's specified in the swarm_config.yaml file under: swarm.task.global_goal """ swarm_config = get_swarm_config() return swarm_config["task"]["global_goal"] def set_swarm_goals(goals: list): """ Set the goals for the swarm. It's specified in the swarm_config.yaml file under: swarm.task.goals Default goals: - Generate a comprehensive description of the startup. Describe their value proposition, the product, USP and business model of a startup. - Find any mentions of the startup in the news, social media, etc. Add links. - Find top 10 companies and startups in this field. Find out their locations, raised funding, value proposition, differentiation, etc. - Find top 5 investors in this field. Includ specific details in the format of 'company AAA (link) invested in company BBB (link) $XX in year YYYY' - Describe the market size, growth rate and trends of this field. - Main problems and challenges of the field. Create an extensive list of problems. What can stop the field from growing? What can stop the company from succeeding? - Briefly describe the technology for the non-tech audience. Include links to the main articles in the field. - What questions should we ask the startup to make a more informed decision? Avoid generic and obvious questions and focus on field/domain specific questions that can uncover problems with this specific startup. """ try: if len(goals) == 0: raise ValueError("Goals can't be empty.") all_empty = True for idx, goal in enumerate(goals): if goal != "": all_empty = False break else: # remove empty goals goals.pop(idx) if not all_empty: raise ValueError("Goals can't be empty.") except ValueError: goals = [ "Generate a comprehensive description of the startup. Describe their value proposition, the product, USP and business model of a startup.", "Find any mentions of the startup in the news, social media, etc. Add links.", "Find top 10 companies and startups in this field. Find out their locations, raised funding, value proposition, differentiation, etc.", "Find top 5 investors in this field. Includ specific details in the format of 'company AAA (link) invested in company BBB (link) $XX in year YYYY'", "Describe the market size, growth rate and trends of this field.", "Main problems and challenges of the field. Create an extensive list of problems. What can stop the field from growing? What can stop the company from succeeding?", "Briefly describe the technology for the non-tech audience. Include links to the main articles in the field.", "What questions should we ask the startup to make a more informed decision? Avoid generic and obvious questions and focus on field/domain specific questions that can uncover problems with this specific startup." ] swarm_config = get_swarm_config() print(f"Setting goals to: {goals}") swarm_config["task"]["goals"] = goals with open(SWARM_CONFIG_PATH, "w") as f: yaml.dump(swarm_config, f) def get_swarm_goals(): """ Get the goals for the swarm. It's specified in the swarm_config.yaml file under: swarm.task.goals """ swarm_config = get_swarm_config() return swarm_config["task"]["goals"] def set_swarm_agents_config(agents_config: list): """ Set the agents config for the swarm. It's specified in the swarm_config.yaml file under: swarm.agents """ try: if len(agents_config) == 0: raise ValueError("No agents config specified.") for agent_config in agents_config: if "type" not in agent_config: raise ValueError(f"Agent config {agent_config} does not have a type specified.") if agent_config["type"] not in ALLOWED_AGENTS: raise ValueError(f"Agent type {agent_config['type']} is not supported. Supported agents: {ALLOWED_AGENTS}") if "n" not in agent_config: raise ValueError(f"Agent config {agent_config} does not have a number of agents specified.") if agent_config["n"] == '': raise ValueError(f"Agent config {agent_config} does not have a number of agents specified.") if agent_config["n"] < 0: raise ValueError(f"Agent config {agent_config} has a negative number of agents specified.") if agent_config["n"] > 100: raise ValueError(f"Agent config {agent_config} has a number of agents specified that is too large. Max number of agents is 10.") except ValueError as e: agents_config = [ {"type": "manager", "n": 2}, {"type": "analyst", "n": 2}, {"type": "googler", "n": 2}, ] swarm_config = get_swarm_config() print(f"Setting agents config to: {agents_config}") swarm_config["swarm"]["agents"] = agents_config with open(SWARM_CONFIG_PATH, "w") as f: yaml.dump(swarm_config, f) def get_swarm_agents_config(): """ Get the agents config for the swarm. It's specified in the swarm_config.yaml file under: swarm.agents """ swarm_config = get_swarm_config() return swarm_config["swarm"]["agents"] def read_swarm_output(): """ Read the output of the swarm. The file can sometimes be locked by the swarm, so we need to handle this. """ try: with open(SWARM_DEFAULT_JSON_OUTPUT) as f: final_out = "" output = json.load(f) for _, value in output.items(): final_out+="========================================\n" final_out+="========================================\n" for key, value in value.items(): final_out+=f"**{key}**:\n{value}\n\n" f.close() except Exception: final_out = "Swarm is starting up (needs ~2-3 minutes for first results and ~30 sec for first logs)..." return final_out def read_swarm_logs(): """ Read the logs of the swarm. The file can sometimes be locked by the swarm, so we need to handle this. """ try: with open(SWARM_DEAFAULT_LOGS) as f: # read last 100 lines logs = f.readlines()[-100:] final_out = "\n".join(logs) f.close() except Exception: final_out = "Swarm is starting up..." return final_out def execute_swarm(smarm_ui_status): _ = run_swarm() # when it finishes, it will set SWARM_IS_RUNNING to False raise TimeoutError("Swarm timed out. Please try again.")