shiyue/stock-scanner
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update stock_analyzer.py

Browse Source
This commit is contained in:
Cassianvale
2025-03-04 12:01:58 +08:00
parent a3474bc201
commit 476568f328
1 changed files with 32 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
stock_analyzer.py
View File

@@ -237,24 +237,39 @@ class StockAnalyzer:
"""
# OpenAI API 调用
response = requests.post(
api_urls = [
f"{self.API_URL}/chat/completions",
headers={
"Authorization": f"Bearer {self.API_KEY}",
"Content-Type": "application/json"
},
json={
"model": os.getenv('API_MODEL', 'gpt-3.5-turbo'),
"messages": [{"role": "user", "content": prompt}]
},
timeout=30
)
f"{self.API_URL}/v1/chat/completions"
]
if response.status_code == 200:
return response.json()['choices'][0]['message']['content']
else:
print(f"API 错误: {response.status_code} - {response.text}")
return f"AI 分析暂时无法使用 (HTTP {response.status_code})"
last_error = None
for api_url in api_urls:
try:
response = requests.post(
api_url,
headers={
"Authorization": f"Bearer {self.API_KEY}",
"Content-Type": "application/json"
},
json={
"model": os.getenv('API_MODEL', 'gpt-3.5-turbo'),
"messages": [{"role": "user", "content": prompt}]
},
timeout=30
)
if response.status_code == 200:
return response.json()['choices'][0]['message']['content']
else:
last_error = f"API 错误: {response.status_code} - {response.text}"
continue
except Exception as e:
last_error = str(e)
continue
print(f"AI 分析暂时无法使用: {last_error}")
return f"AI 分析暂时无法使用: {last_error}"
except Exception as e:
print(f"AI 分析发生错误: {str(e)}")