AI Agents入门教程之不同的智能体架构

class Agent:    def __init__(self, system=""):        self.system = system        self.messages = []        if self.system:            self.messages.append({"role": "system", "content": system})    def __call__(self, message):        self.messages.append({"role": "user", "content": message})        result = self.execute()        self.messages.append({"role": "assistant", "content": result})        return result    def execute(self):        completion = client.chat.completions.create(                        model="gpt-4o",                         temperature=0,                        messages=self.messages)        return completion.choices[0].message.content

import openaiimport reimport httpximport osprompt = """You run in a loop of Thought, Action, PAUSE, Observation.At the end of the loop you output an AnswerUse Thought to describe your thoughts about the question you have been asked.Use Action to run one of the actions available to you - then return PAUSE.Observation will be the result of running those actions.Your available actions are:calculate:e.g. calculate: 4 * 7 / 3Runs a calculation and returns the number - uses Python so be sure to use floating point syntax if necessaryaverage_dog_weight:e.g. average_dog_weight: Colliereturns average weight of a dog when given the breedExample session:Question: How much does a Bulldog weigh?Thought: I should look the dogs weight using average_dog_weightAction: average_dog_weight: BulldogPAUSEYou will be called again with this:Observation: A Bulldog weights 51 lbsYou then output:Answer: A bulldog weights 51 lbs""".strip()def calculate(what):    return eval(what)def average_dog_weight(name):    if name in "Scottish Terrier":         return("Scottish Terriers average 20 lbs")    elif name in "Border Collie":        return("a Border Collies average weight is 37 lbs")    elif name in "Toy Poodle":        return("a toy poodles average weight is 7 lbs")    else:        return("An average dog weights 50 lbs")known_actions = {    "calculate": calculate,    "average_dog_weight": average_dog_weight}prompt = """You run in a loop of Thought, Action, PAUSE, Observation.At the end of the loop you output an AnswerUse Thought to describe your thoughts about the question you have been asked.Use Action to run one of the actions available to you - then return PAUSE.Observation will be the result of running those actions.

abot = Agent(prompt)def query(question, max_turns=5):    i = 0    bot = Agent(prompt)    next_prompt = question    while i < max_turns:        i += 1        result = bot(next_prompt)        print(result)        actions = [            action_re.match(a)             for a in result.split('\n')             if action_re.match(a)        ]        if actions:            # There is an action to run            action, action_input = actions[0].groups()            if action not in known_actions:                raise Exception("Unknown action: {}: {}".format(action, action_input))            print(" -- running {} {}".format(action, action_input))            observation = known_actions[action](action_input)            print("Observation:", observation)            next_prompt = "Observation: {}".format(observation)        else:            returnquestion = """I have 2 dogs, a border collie and a scottish terrier. \What is their combined weight"""query(question)Thought: I need to find the average weight of a Border Collie and a Scottish Terrier, then add them together to get the combined weight.Action: average_dog_weight: Border ColliePAUSE -- running average_dog_weight Border CollieObservation: a Border Collies average weight is 37 lbsAction: average_dog_weight: Scottish TerrierPAUSE -- running average_dog_weight Scottish TerrierObservation: Scottish Terriers average 20 lbsThought: Now that I have the average weights of both dogs, I can calculate their combined weight by adding them together.Action: calculate: 37 + 20PAUSE -- running calculate 37 + 20Observation: 57Answer: The combined weight of a Border Collie and a Scottish Terrier is 57 lbst