Merge branch 'main' of http://jeason.online:3000/zhaojie/iov_data_analysis_agent

test.py

@@ -1,13 +1,22 @@
+# -*- coding: utf-8 -*-
+"""
+快速测试 LLM 连接是否正常
+"""
+
+import os
+from dotenv import load_dotenv
 from openai import OpenAI
 
+load_dotenv()
+
 client = OpenAI(
-    base_url="http://127.0.0.1:9999/v1",
-    api_key="sk-2187174de21548b0b8b0c92129700199"
+    base_url=os.getenv("OPENAI_BASE_URL", "http://127.0.0.1:9999/v1"),
+    api_key=os.getenv("OPENAI_API_KEY", ""),
 )
 
 response = client.chat.completions.create(
-    model="claude-sonnet-4-5",
-    messages=[{"role": "user", "content": "Hello"}]
+    model=os.getenv("OPENAI_MODEL", "gpt-3.5-turbo"),
+    messages=[{"role": "user", "content": "Hello"}],
 )
 
-print(response.choices[0].message.content)
+print(response.choices[0].message.content)