26 lines
789 B
Python
26 lines
789 B
Python
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助手回答
|