rkihacker commited on
Commit
98758f0
·
verified ·
1 Parent(s): bef12d1

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -7
main.py CHANGED
@@ -10,7 +10,7 @@ last_refresh = 0
10
 
11
  app = Flask(__name__)
12
 
13
- # DeepInfra required headers
14
  DEEPINFRA_HEADERS = {
15
  "accept": "text/event-stream",
16
  "content-type": "application/json",
@@ -62,13 +62,9 @@ def proxy_deepinfra():
62
  try:
63
  print(f"[INFO] Forwarding POST to {target_url} via {proxy}")
64
 
65
- # Start with required DeepInfra headers, merge client’s headers on top
66
  forward_headers = dict(DEEPINFRA_HEADERS)
67
- for k, v in request.headers:
68
- if k.lower() != "host":
69
- forward_headers[k] = v
70
 
71
- # Stream request
72
  upstream = requests.post(
73
  url=target_url,
74
  headers=forward_headers,
@@ -76,7 +72,7 @@ def proxy_deepinfra():
76
  params=request.args,
77
  proxies=proxies,
78
  stream=True,
79
- timeout=300,
80
  )
81
 
82
  def generate():
 
10
 
11
  app = Flask(__name__)
12
 
13
+ # Fixed headers for DeepInfra requests
14
  DEEPINFRA_HEADERS = {
15
  "accept": "text/event-stream",
16
  "content-type": "application/json",
 
62
  try:
63
  print(f"[INFO] Forwarding POST to {target_url} via {proxy}")
64
 
65
+ # 🚫 Ignore user headers completely, always use fixed DeepInfra headers
66
  forward_headers = dict(DEEPINFRA_HEADERS)
 
 
 
67
 
 
68
  upstream = requests.post(
69
  url=target_url,
70
  headers=forward_headers,
 
72
  params=request.args,
73
  proxies=proxies,
74
  stream=True,
75
+ timeout=120,
76
  )
77
 
78
  def generate():