feat: 新增股票数据波形图和截图功能
refactor: 重构数据库和LLM分析器逻辑 fix: 修复爬虫解析和UI显示问题 docs: 更新配置文件和注释 style: 优化代码格式和日志输出
This commit is contained in:
32
test_api.py
Normal file
32
test_api.py
Normal file
@@ -0,0 +1,32 @@
|
||||
import os
|
||||
from openai import OpenAI
|
||||
|
||||
# 设置 API key 为环境变量
|
||||
os.environ["NVIDIA_API_KEY"] = "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"
|
||||
|
||||
client = OpenAI(
|
||||
base_url="https://integrate.api.nvidia.com/v1",
|
||||
api_key=os.environ["NVIDIA_API_KEY"]
|
||||
)
|
||||
|
||||
print("正在测试 API 连接...")
|
||||
print("发送消息: 天气")
|
||||
print("-" * 50)
|
||||
|
||||
completion = client.chat.completions.create(
|
||||
model="deepseek-ai/deepseek-r1",
|
||||
messages=[{"role": "user", "content": "天气"}],
|
||||
temperature=0.6,
|
||||
top_p=0.7,
|
||||
max_tokens=4096,
|
||||
stream=False
|
||||
)
|
||||
|
||||
reasoning = getattr(completion.choices[0].message, "reasoning_content", None)
|
||||
if reasoning:
|
||||
print("推理过程:")
|
||||
print(reasoning)
|
||||
print("-" * 50)
|
||||
|
||||
print("回答:")
|
||||
print(completion.choices[0].message.content)
|
||||
Reference in New Issue
Block a user