added safety score. added proxy

Dockerfile

@@ -4,7 +4,7 @@ WORKDIR /app
 
 # Install dependencies
 COPY app/requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip cache purge && pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY app/ .

app/api_helpers.py

@@ -12,7 +12,7 @@ from openai import AsyncOpenAI
 
 from models import OpenAIRequest, OpenAIMessage
 from message_processing import (
-    convert_to_openai_format, 
+    convert_to_openai_format,
     convert_chunk_to_openai,
     extract_reasoning_by_tags,
     _create_safety_ratings_html

app/message_processing.py

@@ -5,7 +5,7 @@ import time
 import random # For more unique tool_call_id
 import urllib.parse
 from typing import List, Dict, Any, Tuple
-from app import config as app_config
+import config as app_config
 
 from google.genai import types
 from models import OpenAIMessage, ContentPartText, ContentPartImage

app/openai_handler.py

@@ -80,8 +80,15 @@ class ExpressClientWrapper:
         if 'extra_body' in payload:
             payload.update(payload.pop('extra_body'))
 
-        proxy = app_config.SOCKS_PROXY or app_config.HTTPS_PROXY
-        client_args = {'timeout': 300, 'proxies': proxy}
+        proxies = None
+        if app_config.SOCKS_PROXY:
+            proxies = {"all://": app_config.SOCKS_PROXY}
+        elif app_config.HTTPS_PROXY:
+            proxies = {"https://": app_config.HTTPS_PROXY}
+
+        client_args = {'timeout': 300}
+        if proxies:
+            client_args['proxies'] = proxies
         if app_config.SSL_CERT_FILE:
             client_args['verify'] = app_config.SSL_CERT_FILE
         async with httpx.AsyncClient(**client_args) as client:
@@ -110,8 +117,15 @@ class ExpressClientWrapper:
         if 'extra_body' in payload:
             payload.update(payload.pop('extra_body'))
 
-        proxy = app_config.SOCKS_PROXY or app_config.HTTPS_PROXY
-        client_args = {'timeout': 300, 'proxies': proxy}
+        proxies = None
+        if app_config.SOCKS_PROXY:
+            proxies = {"all://": app_config.SOCKS_PROXY}
+        elif app_config.HTTPS_PROXY:
+            proxies = {"https://": app_config.HTTPS_PROXY}
+
+        client_args = {'timeout': 300}
+        if proxies:
+            client_args['proxies'] = proxies
         if app_config.SSL_CERT_FILE:
             client_args['verify'] = app_config.SSL_CERT_FILE
         async with httpx.AsyncClient(**client_args) as client:
@@ -142,10 +156,15 @@ class OpenAIDirectHandler:
             f"projects/{project_id}/locations/{location}/endpoints/openapi"
         )
         
-        proxy = app_config.SOCKS_PROXY or app_config.HTTPS_PROXY
+        proxies = None
+        if app_config.SOCKS_PROXY:
+            proxies = {"all://": app_config.SOCKS_PROXY}
+        elif app_config.HTTPS_PROXY:
+            proxies = {"https://": app_config.HTTPS_PROXY}
+
         client_args = {}
-        if proxy:
-            client_args['proxies'] = proxy
+        if proxies:
+            client_args['proxies'] = proxies
         if app_config.SSL_CERT_FILE:
             client_args['verify'] = app_config.SSL_CERT_FILE
         

app/project_id_discovery.py

@@ -2,7 +2,7 @@ import aiohttp
 import json
 import re
 from typing import Dict, Optional
-from app import config
+import config
 
 # Global cache for project IDs: {api_key: project_id}
 PROJECT_ID_CACHE: Dict[str, str] = {}