2025.10.24

02 LLM基础与调用/2. 多轮问答agent.py

@@ -0,0 +1,36 @@
+import os # 导入环境变量
+from dotenv import load_dotenv
+from openai import OpenAI
+
+load_dotenv()
+
+def create_client():
+    api_key = os.getenv("OPENAI_API_KEY")
+    return OpenAI(api_key=api_key,base_url="https://api.deepseek.com")
+
+def chat_loop(agent_client):
+    messages = [
+                {"role":"system","content":"你是一个历史老师，你会耐心的教导我有关的事情。同时你的回答会相对精简，在五十字内。"},
+                {"role":"user","content":"汉朝存在了多久，其中哪个皇帝你认为最厉害？"}
+            ]
+    while 1:
+        response = agent_client.chat.completions.create(
+            model="deepseek-chat",
+            messages=messages
+        )
+        answer = response.choices[0].message.content
+        print(f'回答:{answer}')
+
+        # 询问是否还有其他问题
+        user_input =input('您还有其他想继续问的吗 | (exit退出)\n')
+        if user_input == "exit":
+            break
+
+        # 继续记录对话(涵盖用户追问 + ai上句回答)
+        messages.append({"role":"user","content":user_input})
+        messages.append({"role":"assistant","content":answer})
+
+
+if __name__ == '__main__':
+    client = create_client()
+    chat_loop(client)