Update app.py

app.py

@@ -2,6 +2,7 @@
 # https://username-spaceName.hf.space/proxy/anthropic/v1
 # Streaming is not supported
 
+
 import logging
 import json
 from flask import Flask, request, Response
@@ -18,24 +19,37 @@ def relay_request(path):
 
     relay_url = 'https://' + path
 
-    headers_ = {}
     
-    # Copy headers from the client request
-    for key, value in request.headers.items():
-        if key.lower() != 'host':
-            headers_[key] = value
-
-    # Proxy the request
-    response = requests.request(
-        method=request.method,
-        url=relay_url,
-        headers=headers_,
-        data=request.stream,
-        stream=True  # Stream the response back to the client
-    )
-
-    # Stream the response content back to the client
-    return Response(response.iter_content(chunk_size=1024), status=response.status_code, headers=response.headers.items())
+    headers_ = dict()
+    
+    
+    if 'Content-Type' in request.headers:
+        headers_['Content-Type'] = request.headers['Content-Type']
+    if 'x-api-key' in request.headers:
+        headers_['Authorization'] = "Bearer " + request.headers['x-api-key']
+    if 'Authorization' in request.headers:
+        headers_['Authorization'] = request.headers['Authorization']
+
+    if request.method == 'POST':
+        response = requests.request(
+            method='POST',
+            url=relay_url,
+            headers=headers_,
+            json=json.loads(request.get_data()),
+            allow_redirects=False 
+        )
+    else:
+        response = requests.request(
+            method=request.method,
+            url=relay_url,
+            headers=headers_,
+            allow_redirects=False 
+        )
+        
+
+
+    headers = [(name, value) for name, value in response.headers.items()]
+    return Response(response.content, response.status_code, headers)
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)