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

49
stock_analyzer.py
View File

@@ -237,25 +237,40 @@ class StockAnalyzer:
""" """
# OpenAI API 调用 # OpenAI API 调用
response = requests.post( api_urls = [
f"{self.API_URL}/chat/completions", f"{self.API_URL}/chat/completions",
headers={ f"{self.API_URL}/v1/chat/completions"
"Authorization": f"Bearer {self.API_KEY}", ]
"Content-Type": "application/json"
}, last_error = None
json={ for api_url in api_urls:
"model": os.getenv('API_MODEL', 'gpt-3.5-turbo'), try:
"messages": [{"role": "user", "content": prompt}] response = requests.post(
}, api_url,
timeout=30 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}"
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})"
except Exception as e: except Exception as e:
print(f"AI 分析发生错误: {str(e)}") print(f"AI 分析发生错误: {str(e)}")
return f"AI 分析过程中发生错误: {str(e)}" return f"AI 分析过程中发生错误: {str(e)}"