|
|
#!/usr/bin/env python3
|
|
|
"""
|
|
|
快速DeepSeek v3.1测试脚本
|
|
|
独立运行，不依赖复杂导入
|
|
|
"""
|
|
|
|
|
|
import os
|
|
|
import requests
|
|
|
import json
|
|
|
import time
|
|
|
|
|
|
|
|
|
def quick_deepseek_test():
|
|
|
"""快速测试DeepSeek v3.1 API"""
|
|
|
print("🚀 DeepSeek v3.1 快速测试")
|
|
|
print("=" * 40)
|
|
|
|
|
|
# 检查API密钥
|
|
|
api_key = os.getenv('SILICONFLOW_API_KEY')
|
|
|
if not api_key:
|
|
|
print("❌ 错误: 未设置 SILICONFLOW_API_KEY 环境变量")
|
|
|
return
|
|
|
|
|
|
print(f"📡 连接模型: deepseek-ai/DeepSeek-V3.1")
|
|
|
print(f"🔑 API密钥: {api_key[:20]}...")
|
|
|
|
|
|
try:
|
|
|
# 测试连接
|
|
|
headers = {
|
|
|
"Authorization": f"Bearer {api_key}",
|
|
|
"Content-Type": "application/json"
|
|
|
}
|
|
|
|
|
|
data = {
|
|
|
"model": "deepseek-ai/DeepSeek-V3.1",
|
|
|
"messages": [
|
|
|
{
|
|
|
"role": "system",
|
|
|
"content": "你是CBMC形式化验证专家，为C代码生成CBMC规约。只输出CBMC代码。"
|
|
|
},
|
|
|
{
|
|
|
"role": "user",
|
|
|
"content": """为以下C函数生成CBMC规约：
|
|
|
|
|
|
int add_numbers(int a, int b) {
|
|
|
return a + b;
|
|
|
}
|
|
|
|
|
|
验证目标: functional_correctness
|
|
|
|
|
|
生成CBMC规约：
|
|
|
\\requires ...
|
|
|
\\ensures ...
|
|
|
\\assigns ..."""
|
|
|
}
|
|
|
],
|
|
|
"temperature": 0.1,
|
|
|
"max_tokens": 512,
|
|
|
"stream": False
|
|
|
}
|
|
|
|
|
|
print("\n⏳ 调用DeepSeek v3.1 API...")
|
|
|
start_time = time.time()
|
|
|
|
|
|
response = requests.post(
|
|
|
"https://api.siliconflow.cn/v1/chat/completions",
|
|
|
headers=headers,
|
|
|
json=data,
|
|
|
timeout=30
|
|
|
)
|
|
|
|
|
|
response_time = time.time() - start_time
|
|
|
|
|
|
if response.status_code == 200:
|
|
|
result = response.json()
|
|
|
content = result['choices'][0]['message']['content']
|
|
|
tokens_used = result.get('usage', {}).get('total_tokens', 0)
|
|
|
|
|
|
print(f"✅ API调用成功!")
|
|
|
print(f"⏱️ 响应时间: {response_time:.2f}s")
|
|
|
print(f"🔤 消耗Token: {tokens_used}")
|
|
|
|
|
|
print(f"\n📋 生成的CBMC规约:")
|
|
|
print("-" * 40)
|
|
|
print(content)
|
|
|
print("-" * 40)
|
|
|
|
|
|
# 提取CBMC规约
|
|
|
lines = content.split('\n')
|
|
|
cbmc_lines = []
|
|
|
for line in lines:
|
|
|
line = line.strip()
|
|
|
if (line.startswith(r'\requires') or
|
|
|
line.startswith(r'\ensures') or
|
|
|
line.startswith(r'\assigns')):
|
|
|
cbmc_lines.append(line)
|
|
|
|
|
|
if cbmc_lines:
|
|
|
print(f"\n🔧 提取的CBMC规约:")
|
|
|
print("-" * 30)
|
|
|
print('\n'.join(cbmc_lines))
|
|
|
print("-" * 30)
|
|
|
|
|
|
else:
|
|
|
print(f"❌ API调用失败: {response.status_code}")
|
|
|
print(f"错误信息: {response.text}")
|
|
|
|
|
|
except Exception as e:
|
|
|
print(f"❌ 测试失败: {str(e)}")
|
|
|
|
|
|
print(f"\n🎉 快速测试完成!")
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
quick_deepseek_test() |
