fix tools

docker-compose.yaml

@@ -83,8 +83,6 @@ services:
       interval: 10s
       timeout: 5s
       retries: 5
-    mem_limit: 4g
-    cpus: "1.5"
     security_opt: [ "no-new-privileges=true" ]
 
   migrate:
@@ -117,8 +115,6 @@ services:
       interval: 15s
       timeout: 5s
       retries: 5
-    mem_limit: 512m
-    cpus: "0.5"
     security_opt: [ "no-new-privileges=true" ]
 
   qdrant:
@@ -136,8 +132,6 @@ services:
       interval: 15s
       timeout: 5s
       retries: 5
-    mem_limit: 4g
-    cpus: "2"
     security_opt: [ "no-new-privileges=true" ]
     cap_drop: [ ALL ]
 
@@ -161,8 +155,6 @@ services:
       retries: 5
     ports:
       - "8000:8000"
-    mem_limit: 768m
-    cpus: "1"
 
   proxy:
     build:
@@ -181,8 +173,6 @@ services:
       - ./tools:/app/tools:ro
     ports:
       - "8001:8001"
-    mem_limit: 256m
-    cpus: "0.5"
 
   # ------------------------------------------------------------ Celery workers
   worker:

tools/add_user.py

@@ -13,7 +13,9 @@ if __name__ == "__main__":
     args = args.parse_args()
 
     with make_session() as session:
-        user = User(email=args.email, password=args.password, name=args.name)
+        user = User.create_with_password(
+            email=args.email, password=args.password, name=args.name
+        )
         session.add(user)
         session.commit()
 

tools/simple_proxy.py

@@ -76,6 +76,7 @@ async def proxy_request(request: Request) -> Response:
     # Get request body
     body = await request.body()
     headers = dict(request.headers)
+    headers.pop("host", None)
 
     async with httpx.AsyncClient() as client:
         try:
@@ -88,12 +89,13 @@ async def proxy_request(request: Request) -> Response:
             )
 
             # Forward response
-            return Response(
+            resp = Response(
                 content=response.content,
                 status_code=response.status_code,
                 headers=dict(response.headers),
                 media_type=response.headers.get("content-type"),
             )
+            return resp
 
         except httpx.RequestError as e:
             print(f"Request failed: {e}")