diff --git a/README.md b/README.md index 9a60617..01bac54 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,17 @@ -让大模型为你打工！ +# autoMate -输入任务后就能上网找资料，然后参考优秀案例执行，最后给一个结果！你只需要像一个老板一样反馈结果好还是坏，autoMate会自动优化方案。 \ No newline at end of file +autoMate是一个让大模型为你打工的工具。它能够根据你输入的任务，在网上找资料并参考优秀案例来执行任务，并最终给出一个结果。你只需要像一个老板一样反馈结果好还是坏，autoMate会自动优化方案。 + + +# 角色 + +autoMate定义了三种角色： + +- 总经理：负责整体的任务管理和决策。 +- 计划者：负责制定任务的具体计划和策略。 +- 工人：负责执行具体的任务。 + + +# OKR工作体系 + +市面上的大模型是对任务理解不透彻，如何解决这个问题呢？我想到了OKR工作体系，把上层的O向下拆解成KR，并且中间不断对焦！我认为这是一个非常高效的工作体系，AutoMate 引入OKR就像是给整个团队配上了高级导航系统，各agent都能清清楚楚知道自己要完成的任务，同时不断与上级对焦能够避免任务失真。 \ No newline at end of file diff --git a/__pycache__/okr_define.cpython-39.pyc b/__pycache__/okr_define.cpython-39.pyc new file mode 100644 index 0000000..25356e6 Binary files /dev/null and b/__pycache__/okr_define.cpython-39.pyc differ diff --git a/agent/agent.py b/agent/agent.py new file mode 100644 index 0000000..ab0a2f2 --- /dev/null +++ b/agent/agent.py @@ -0,0 +1,47 @@ +import openai +import yaml + + +class Agent: + def __init__(self, role): + """ + Initialize an Agent object. + + Args: + role (str): The role of the agent. + + Returns: + None + """ + self.role = role + + + + def call(self, input_text): + """ + Make a call to the OpenAI chat API. + + Args: + input_text (str): The input text for the chat API. + + Returns: + str: The generated text from the chat API. + """ + # Load API configuration from YAML file + with open("config.yaml", "r") as file: + config = yaml.safe_load(file).get("openai") + + openai.api_key = config.get("api_key") + openai.base_url = config.get("api_url") + + response = openai.chat.completions.create( + model="gpt-3.5-turbo", # GPT-3.5 model + messages=[ + {"role": "user", "content": self.role +","+ input_text} + ], + temperature=0.2 + ) + + generated_text = response.choices[0].message.content + + return generated_text \ No newline at end of file diff --git a/agent/manager_agent.py b/agent/manager_agent.py new file mode 100644 index 0000000..b2abba0 --- /dev/null +++ b/agent/manager_agent.py @@ -0,0 +1,5 @@ +from agent.agent import Agent + +class ManagerAgent(Agent): + def __init__(self): + super().__init__("你是一名总经理，负责与用户沟通需求，制定OKR中的O，并对下属的工作成果进行评估") diff --git a/agent/plan_agent.py b/agent/plan_agent.py new file mode 100644 index 0000000..e89d280 --- /dev/null +++ b/agent/plan_agent.py @@ -0,0 +1,5 @@ +from agent.agent import Agent + +class WokerAgent(Agent): + def __init__(self): + super().__init__("你是一名计划拆解者，负责对OKR中的O进行拆解并制定KR，向总经理汇报") diff --git a/agent/woker_agent.py b/agent/woker_agent.py new file mode 100644 index 0000000..d18255d --- /dev/null +++ b/agent/woker_agent.py @@ -0,0 +1,5 @@ +from agent.agent import Agent + +class WokerAgent(Agent): + def __init__(self): + super().__init__("你是一名工作者，负责执行OKR中的KR，向总计划制定者汇报") diff --git a/main.py b/main.py index c24759b..69d8f86 100644 --- a/main.py +++ b/main.py @@ -1,79 +1,12 @@ -import os -import requests -import yaml -from selenium import webdriver -from selenium.webdriver.chrome.service import Service -from selenium.webdriver.common.by import By -from selenium.webdriver.chrome.options import Options -from webdriver_manager.chrome import ChromeDriverManager -from webdriver_manager.firefox import GeckoDriverManager -from webdriver_manager.microsoft import EdgeChromiumDriverManager -from webdriver_manager.opera import OperaDriverManager -from selenium.webdriver.edge.service import Service as EdgeService -import openai -from selenium.webdriver.common.by import By - class AutoMate: def __init__(self): pass - def main(self): - # Load browser configuration from YAML file - with open("config.yaml", "r") as file: - config = yaml.safe_load(file) - - # Check if webdriver is available - if not os.path.exists("webdriver.exe"): - # Download webdriver based on browser type - browser_type = config.get("browser_type") - driver = None - if browser_type == "chrome": - options = webdriver.ChromeOptions() - options.add_argument("--headless") # Enable headless mode - webdriver_manager = ChromeDriverManager() - driver = webdriver.Chrome(webdriver_manager.install(), options=options) - elif browser_type == "firefox": - options = webdriver.FirefoxOptions() - options.add_argument("--headless") # Enable headless mode - webdriver_manager = GeckoDriverManager() - driver = webdriver.Firefox(webdriver_manager.install(), options=options) - elif browser_type == "edge": - options = webdriver.EdgeOptions() - options.add_argument("--headless") # Enable headless mode - webdriver_manager = EdgeChromiumDriverManager() - driver = webdriver.Edge(service=EdgeService(webdriver_manager.install()), options=options) - elif browser_type == "opera": - options = webdriver.ChromeOptions() - options.add_argument("--headless") # Enable headless mode - webdriver_manager = OperaDriverManager() - driver = webdriver.Opera(webdriver_manager.install(), options=options) - driver.implicitly_wait(10) - browser_url = config.get("browser_url") - driver.get(browser_url) - driver.find_element(by=By.XPATH, value='//*[@id="kw"]').send_keys("陈") - driver.find_element(by=By.XPATH, value='//*[@id="su"]').click() - print(driver.page_source) - driver.quit() - - def call_chatgpt_api(self, input_text): - # Load API configuration from YAML file - with open("config.yaml", "r") as file: - config = yaml.safe_load(file).get("openai") - - openai.api_key = config.get("api_key") - openai.base_url = config.get("api_url") - - response = openai.chat.completions.create( - model="gpt-3.5-turbo", # GPT-3.5 model - messages=[ - {"role": "user", "content": input_text} - ], - temperature=0.7 - ) - - generated_text = response.choices[0].message.content - - return generated_text + + + def rule_define(self): + pass + diff --git a/okr_define.py b/okr_define.py new file mode 100644 index 0000000..77984fc --- /dev/null +++ b/okr_define.py @@ -0,0 +1,35 @@ +class OKR: + """ + Represents an Objectives and Key Results (OKR) tracker. + + Attributes: + objectives (dict): A dictionary to store the objectives and their progress. + key_results (dict): A dictionary to store the key results for each objective. + + Methods: + set_objective(objective): Sets a new objective with initial progress of 0. + set_key_result(objective, key_result): Sets a new key result for the given objective. + set_objective_progress(objective, progress): Updates the progress of the given objective. + set_key_result_progress(objective, key_result, progress): Updates the progress of the given key result. + """ + + def __init__(self): + self.objectives = {} + self.key_results = {} + + def set_objective(self, objective): + self.objectives[objective] = 0 + + def set_key_result(self, objective, key_result): + if objective in self.objectives: + if objective not in self.key_results: + self.key_results[objective] = [] + self.key_results[objective].append(key_result) + + def set_objective_progress(self, objective, progress): + if objective in self.objectives: + self.objectives[objective] = progress + + def set_key_result_progress(self, objective, key_result, progress): + if objective in self.key_results and key_result in self.key_results[objective]: + self.key_results[objective][key_result] = progress diff --git a/tools/tools_base.py b/tools/tools_base.py new file mode 100644 index 0000000..ba594a8 --- /dev/null +++ b/tools/tools_base.py @@ -0,0 +1,3 @@ +class ToolsBase: + def run(self): + pass diff --git a/tools/web_browser_tools.py b/tools/web_browser_tools.py new file mode 100644 index 0000000..cf7da4c --- /dev/null +++ b/tools/web_browser_tools.py @@ -0,0 +1,52 @@ +import os +import yaml +from selenium import webdriver +from selenium.webdriver.common.by import By +from webdriver_manager.chrome import ChromeDriverManager +from webdriver_manager.firefox import GeckoDriverManager +from webdriver_manager.microsoft import EdgeChromiumDriverManager +from webdriver_manager.opera import OperaDriverManager +from selenium.webdriver.edge.service import Service as EdgeService +from selenium.webdriver.common.by import By + +from tools.tools_base import ToolsBase + +class WebBrowser(ToolsBase): + + def run(self): + # Load browser configuration from YAML file + with open("config.yaml", "r") as file: + config = yaml.safe_load(file) + + # Check if webdriver is available + if not os.path.exists("webdriver.exe"): + # Download webdriver based on browser type + browser_type = config.get("browser_type") + driver = None + if browser_type == "chrome": + options = webdriver.ChromeOptions() + options.add_argument("--headless") # Enable headless mode + webdriver_manager = ChromeDriverManager() + driver = webdriver.Chrome(webdriver_manager.install(), options=options) + elif browser_type == "firefox": + options = webdriver.FirefoxOptions() + options.add_argument("--headless") # Enable headless mode + webdriver_manager = GeckoDriverManager() + driver = webdriver.Firefox(webdriver_manager.install(), options=options) + elif browser_type == "edge": + options = webdriver.EdgeOptions() + options.add_argument("--headless") # Enable headless mode + webdriver_manager = EdgeChromiumDriverManager() + driver = webdriver.Edge(service=EdgeService(webdriver_manager.install()), options=options) + elif browser_type == "opera": + options = webdriver.ChromeOptions() + options.add_argument("--headless") # Enable headless mode + webdriver_manager = OperaDriverManager() + driver = webdriver.Opera(webdriver_manager.install(), options=options) + driver.implicitly_wait(10) + browser_url = config.get("browser_url") + driver.get(browser_url) + driver.find_element(by=By.XPATH, value='//*[@id="kw"]').send_keys("陈") + driver.find_element(by=By.XPATH, value='//*[@id="su"]').click() + print(driver.page_source) + driver.quit()