Update deepseek api response timeout

This commit is contained in:
sheaven79 2025-02-01 21:04:46 -06:00 committed by GitHub
parent 1e37ab3fd4
commit f419a54d90
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -35,7 +35,7 @@ class APIAdapter:
"messages": [{"role": "user", "content": prompt}], "messages": [{"role": "user", "content": prompt}],
"temperature": temperature "temperature": temperature
} }
response = requests.post(self.api_endpoint, headers=headers, json=payload) response = requests.post(self.api_endpoint, headers=headers, json=payload, timeout=60)
else: # ollama else: # ollama
payload = { payload = {
"model": self.model, "model": self.model,
@ -1127,4 +1127,4 @@ if __name__ == '__main__':
window = CyberSecurityApp() window = CyberSecurityApp()
window.show() window.show()
sys.exit(app.exec_()) sys.exit(app.exec_())