Add chapter7 Agent code
This commit is contained in:
25
docs/chapter7/Agent/demo.py
Normal file
25
docs/chapter7/Agent/demo.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from src.core import Agent
|
||||
from src.tools import add, count_letter_in_string, compare, get_current_datetime
|
||||
|
||||
from openai import OpenAI
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
client = OpenAI(
|
||||
api_key="your siliconflow api key",
|
||||
base_url="https://api.siliconflow.cn/v1",
|
||||
)
|
||||
|
||||
agent = Agent(
|
||||
client=client,
|
||||
model="Qwen/Qwen2.5-32B-Instruct",
|
||||
tools=[get_current_datetime, add, compare, count_letter_in_string],
|
||||
)
|
||||
|
||||
while True:
|
||||
# 使用彩色输出区分用户输入和AI回答
|
||||
prompt = input("\033[94mUser: \033[0m") # 蓝色显示用户输入提示
|
||||
if prompt == "exit":
|
||||
break
|
||||
response = agent.get_completion(prompt)
|
||||
print("\033[92mAssistant: \033[0m", response) # 绿色显示AI助手回答
|
||||
Reference in New Issue
Block a user