https://docs.ag2.ai/docs/Getting-Started
https://docs.ag2.ai/docs/autogen-studio/usage
pip install autogen
Ví dụ 01 – Khởi động
import os
from autogen import AssistantAgent, UserProxyAgent
llm_config = {"model": "gpt-4o-mini", "api_key": os.environ["OPENAI_API_KEY"]}
assistant = AssistantAgent("assistant", llm_config=llm_config)
user_proxy = UserProxyAgent("user_proxy", code_execution_config=False)
# Start the chat
user_proxy.initiate_chat(
assistant,
message="Tell me a joke about NVDA and TESLA stock prices.",
)
Giải thích:
Đoạn mã nguồn này sử dụng thư viện autogen
để tạo một cuộc trò chuyện giữa một trợ lý AI và người dùng.
Phân tích từng phần:
import os
: Dòng này nhập moduleos
của Python, cho phép tương tác với hệ điều hành, ví dụ như đọc biến môi trường.from autogen import AssistantAgent, UserProxyAgent
: Dòng này nhập hai lớpAssistantAgent
vàUserProxyAgent
từ thư việnautogen
. Hai lớp này đại diện cho trợ lý AI và người dùng trong cuộc trò chuyện.llm_config = {"model": "gpt-4o-mini", "api_key": os.environ["OPENAI_API_KEY"]}
:- Biến
llm_config
là một từ điển chứa các cấu hình cho mô hình ngôn ngữ lớn (LLM). "model": "gpt-4o-mini"
chỉ định mô hình sẽ được sử dụng là GPT-4o-mini."api_key": os.environ["OPENAI_API_KEY"]}
lấy khóa API OpenAI từ biến môi trườngOPENAI_API_KEY
. Điều này cho phép mã nguồn kết nối với dịch vụ OpenAI API.
- Biến
assistant = AssistantAgent("assistant", llm_config=llm_config)
: Dòng này tạo một instance củaAssistantAgent
với tên là “assistant” và sử dụng cấu hìnhllm_config
đã được định nghĩa.user_proxy = UserProxyAgent("user_proxy", code_execution_config=False)
: Dòng này tạo một instance củaUserProxyAgent
với tên là “user_proxy”. Tham sốcode_execution_config=False
ngăn không cho agent này thực thi mã.user_proxy.initiate_chat(assistant, message="Tell me a joke about NVDA and TESLA stock prices.")
:- Dòng này bắt đầu cuộc trò chuyện giữa
user_proxy
vàassistant
. user_proxy
gửi tin nhắn “Tell me a joke about NVDA and TESLA stock prices.” đếnassistant
.
- Dòng này bắt đầu cuộc trò chuyện giữa
Tóm lại:
Đoạn mã này tạo ra một chatbot đơn giản sử dụng mô hình GPT-4o-mini của OpenAI. Người dùng (được đại diện bởi user_proxy
) yêu cầu chatbot kể một câu chuyện cười về giá cổ phiếu của NVDA và TESLA. Chatbot (được đại diện bởi assistant
) sẽ sử dụng mô hình GPT-4o-mini để tạo ra câu trả lời.
Lưu ý:
Bạn cần có khóa API OpenAI và lưu nó vào biến môi trường OPENAI_API_KEY
.
Để chạy mã này, bạn cần cài đặt thư viện autogen
(pip install autogen
).
Ollama
ollama pull codellama
CodeLLama
https://ollama.com/library/codellama
https://ollama.com/library/codellama
ollama pull codellama