带函数的多代理 AutoGen - 使用代码示例分步_autogen 使用

一、说明

AutoGen 是一个开创性的多智能体对话框架，彻底改变了基础模型的使用方式。这个创新平台具有多功能、可对话的座席，能够通过自动座席聊天集成大型语言模型 （LLM）、工具和人类见解。AutoGen 的独特方法不仅简化了复杂的 LLM 工作流程，而且还最大限度地提高了其性能，标志着下一代 LLM 应用程序开发的飞跃。

​​

二、AutoGen 的本质：可对话和可定制的代理

AutoGen 的核心是其代理，其设计考虑了对话。这些智能体可以无缝交换信息，通过智能体间对话为协作解决任务铺平道路。这些代理可定制以合并 LLM、人工输入或两者兼而有之，体现了灵活性和适应性。该框架具有内置代理，如 AssistantAgent 和 UserProxyAgent，每个代理都具有独特的功能。利用 LLM 的 AssistantAgent 可以自主生成 Python 代码和建议，而充当人工代理的 UserProxyAgent 可以在必要时执行代码并触发基于 LLM 的响应。

2.1 简化任务自动化和人机交互

AutoGen的出色之处在于它能够自动执行多智能体通信，同时保持人为干预的选项。这种双重方法确保了任务得到有效处理，无论是通过自主代理交互还是在人工指导下。该框架的工具调用功能进一步提高了其效率，允许代理有效地与外部工具和 API 进行交互。

2.2 赋能动态对话

AutoGen 的多功能性扩展到支持各种对话模式，从完全自主的对话到人机交互问题解决。这种适应性在需要动态响应和复杂问题解决策略的应用中至关重要。

三、分步代码示例

在我们即将推出的用例中，我们将探索 Whisper 和 GPT-4 与 AutoGen 的 AssistantAgent 和 UserProxyAgent 的集成。本演示将重点处理视频文件，我们的目标是使用 Whisper 识别和转录口语。随后，我们将利用 GPT-4 的翻译功能将转录转换为不同的语言。最终目标是生成带有时间戳的字幕，反映创建字幕文件的过程，展示 AutoGen 在媒体翻译和可访问性增强方面的实际应用。

3.1 第 1 步：安装必要的库

首先，我们需要确保我们的环境具有所有必要的 Python 库。以下是在笔记本中运行以安装每个命令的命令：

openai：这是 OpenAI 的官方库，允许我们与他们的 API 服务进行交互。
openai-whisper：用于音频处理的专用库，专门使用 OpenAI 的 Whisper 模型进行转录。
moviepy：用于视频处理的多功能库，支持视频编辑任务。
pyautogen：AutoGen 的核心库，可促进多智能体对话式 AI。
若要安装这些库，请在笔记本中运行以下命令：

%%capture --no-stderr
!pip install moviepy~=1.0.3
# !pip install openai-whisper~=20230918
!pip install openai-whisper
!pip install openai~=1.3.5
!pip install "pyautogen>=0.2.3"
1
2
3
4
5
6

3.2 第 2 步：设置 API 终端节点

使用 API 密钥时，安全性和便利性至关重要。若要安全地设置 OpenAI API 密钥，请在笔记本中使用以下代码。替换为您实际的 OpenAI API 密钥：‘your-api-key’

import os
os.environ["OPENAI_API_KEY"] = "your-openai-api-key"
%env
1
2
3

以这种方式设置 API 密钥允许您的代码与 OpenAI 的服务无缝进行身份验证，以进行任何后续 API 调用，例如转录视频音频或翻译文本。
请记住，代码片段中提供的 API 密钥是一个占位符。在运行笔记本之前，应将其替换为实际的 OpenAI API 密钥。

3.3 第 3 步：导入库并设置配置

现在我们的环境已经设置好了，下一步是导入所需的库，并为我们的任务配置它们：

• whisper对于音频转录，
• moviepy.editor要处理视频文件，
• openai访问 OpenAI 功能，
• autogen作为我们的对话式 AI 框架。
  下面是执行此操作的代码：

import os

import whisper
from moviepy.editor import VideoFileClip
from openai import OpenAI

import autogen
config_list = [
    {
        "model": "gpt-4",
        "api_key": os.getenv("OPENAI_API_KEY"),
    }
]
1
2
3
4
5
6
7
8
9
10
11
12
13

   通过导入这些库，我们使我们的笔记本能够转录和处理视频内容，并与 OpenAI 强大的语言模型进行通信。这一点至关重要，因为它指定了要使用的模型（在本例中）以及在哪里查找 API 密钥（我们在上一步的环境变量中设置了该密钥）。config_listgpt-4

    通过此设置，我们现在可以开始处理视频和音频内容了。
1
2
3

3.4 第 4 步：创建转录和翻译的函数和配置

让我们深入研究应用程序的核心功能。我们将定义两个基本函数：

1 recognize_transcript_from_video：此函数获取从视频中提取的音频文件的路径，并使用 Whisper 模型将音频转录为文本。它处理音频片段以构建带有时间戳的转录本。
2 translate_text：该函数将转录后的文本发送到OpenAI的API，具体使用gpt-4模型，请求从源语言直接翻译到目标语言，适用于视频字幕。
我们的 AutoGen 代理、聊天机器人和 user_proxy 的配置是通过 llm_config 字典建立的。在这里，我们定义代理将使用的函数并设置通信参数，例如系统消息和代码执行配置。​​​​​​​

最后，我们启动 user_proxy 和聊天机器人之间的聊天，指示它们识别给定视频文件中的语音并将文本翻译成所需的语言。

提供的代码将是这些操作的框架，您可以根据特定要求对其进行扩展和自定义。下面是初始化和聊天启动的示例：

def recognize_transcript_from_video(audio_filepath):
    try:
        # Load model
        model = whisper.load_model("small")

        # Transcribe audio with detailed timestamps
        result = model.transcribe(audio_filepath, verbose=True)

        # Initialize variables for transcript
        transcript = []
        sentence = ""
        start_time = 0

        # Iterate through the segments in the result
        for segment in result["segments"]:
            # If new sentence starts, save the previous one and reset variables
            if segment["start"] != start_time and sentence:
                transcript.append(
                    {
                        "sentence": sentence.strip() + ".",
                        "timestamp_start": start_time,
                        "timestamp_end": segment["start"],
                    }
                )
                sentence = ""
                start_time = segment["start"]

            # Add the word to the current sentence
            sentence += segment["text"] + " "

        # Add the final sentence
        if sentence:
            transcript.append(
                {
                    "sentence": sentence.strip() + ".",
                    "timestamp_start": start_time,
                    "timestamp_end": result["segments"][-1]["end"],
                }
            )

        # Save the transcript to a file
        with open("transcription.txt", "w") as file:
            for item in transcript:
                sentence = item["sentence"]
                start_time, end_time = item["timestamp_start"], item["timestamp_end"]
                file.write(f"{start_time}s to {end_time}s: {sentence}\n")

        return transcript

    except FileNotFoundError:
        return "The specified audio file could not be found."
    except Exception as e:
        return f"An unexpected error occurred: {str(e)}"

def translate_text(input_text, source_language, target_language):
    client = OpenAI(api_key=key)

    response = client.chat.completions.create(
#        model="gpt-3.5-turbo",
        model="gpt-4",
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {
                "role": "user",
                "content": f"Directly translate the following {source_language} text to a pure {target_language} "
                f"video subtitle text without additional explanation.: '{input_text}'",
            },
        ],
        max_tokens=1500,
    )

    # Correctly accessing the response content
    translated_text = response.choices[0].message.content if response.choices else None
    return translated_text


def translate_transcript(source_language, target_language):
    with open("transcription.txt", "r") as f:
        lines = f.readlines()

    translated_transcript = []

    for line in lines:
        # Split each line into timestamp and text parts
        parts = line.strip().split(": ")
        if len(parts) == 2:
            timestamp, text = parts[0], parts[1]
            # Translate only the text part
            translated_text = translate_text(text, source_language, target_language)
            # Reconstruct the line with the translated text and the preserved timestamp
            translated_line = f"{timestamp}: {translated_text}"
            translated_transcript.append(translated_line)
        else:
            # If the line doesn't contain a timestamp, add it as is
            translated_transcript.append(line.strip())

    return "\n".join(translated_transcript)


llm_config = {
    "functions": [
        {
            "name": "recognize_transcript_from_video",
            "description": "recognize the speech from video and transfer into a txt file",
            "parameters": {
                "type": "object",
                "properties": {
                    "audio_filepath": {
                        "type": "string",
                        "description": "path of the video file",
                    }
                },
                "required": ["audio_filepath"],
            },
        },
        {
            "name": "translate_transcript",
            "description": "using translate_text function to translate the script",
            "parameters": {
                "type": "object",
                "properties": {
                    "source_language": {
                        "type": "string",
                        "description": "source language",
                    },
                    "target_language": {
                        "type": "string",
                        "description": "target language",
                    },
                },
                "required": ["source_language", "target_language"],
            },
        },
    ],
    "config_list": config_list,
    "timeout": 120,
}
source_language = "English"
target_language = "Spanish"
key = os.getenv("OPENAI_API_KEY")
target_video = "/content/LiquidSyllabus.mp4"

chatbot = autogen.AssistantAgent(
    name="chatbot",
    system_message="For coding tasks, only use the functions you have been provided with. Reply TERMINATE when the task is done.",
    llm_config=llm_config,
)

user_proxy = autogen.UserProxyAgent(
    name="user_proxy",
    is_termination_msg=lambda x: x.get("content", "") and x.get("content", "").rstrip().endswith("TERMINATE"),
    human_input_mode="NEVER",
    max_consecutive_auto_reply=10,
    code_execution_config={
        "work_dir": "coding_2",
        "use_docker": False,
    },  # Please set use_docker=True if docker is available to run the generated code. Using docker is safer than running the generated code directly.
)

user_proxy.register_function(
    function_map={
        "recognize_transcript_from_video": recognize_transcript_from_video,
        "translate_transcript": translate_transcript,
    }
)
user_proxy.initiate_chat(
    chatbot,
    message=f"For the video located in {target_video}, recognize the speech and transfer it into a script file, "
    f"then translate from {source_language} text to a {target_language} video subtitle text. ",
)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171

四、结论

总之，我们开始的旅程展示了 AutoGen 多智能体对话框架的强大功能，利用尖端的 AI 将原始视频内容转换为可访问和翻译的字幕。通过利用 Whisper 进行转录，利用 GPT-4 进行翻译，我们弥合了不同语言和媒介之间的差距，展示了 AI 的实际和变革性应用。此用例说明了用户代理和助手等代理如何协同工作以自动执行复杂任务，为内容本地化和可访问性方面的创新解决方案铺平了道路。随着人工智能的不断发展，这种技术使全球通信更具包容性和互联性的潜力确实是无限的。