From f45a973f512c403b2a19c8768449554f99628dd8 Mon Sep 17 00:00:00 2001
From: Hemang <hemang@invariantlabs.ai>
Date: Tue, 1 Apr 2025 10:04:00 +0200
Subject: [PATCH] Small formatting changes.

---
 .env                        |  2 +-
 gateway/.env                |  6 +++---
 gateway/routes/anthropic.py | 10 +++++-----
 gateway/routes/gemini.py    |  8 ++++----
 gateway/routes/open_ai.py   |  7 ++++---
 5 files changed, 17 insertions(+), 16 deletions(-)

diff --git a/.env b/.env
index af1265e..6fa5ec4 100644
--- a/.env
+++ b/.env
@@ -3,4 +3,4 @@
 # If you want to push to a local instance of explorer, then specify the app-api docker container name like: 
 # http://<app-api-docker-container-name>:8000 to push to the local explorer instance.
 INVARIANT_API_URL=https://explorer.invariantlabs.ai
-GUADRAILS_API_URL=https://explorer.invariantlabs.ai
+GUADRAILS_API_URL=https://explorer.invariantlabs.ai
\ No newline at end of file
diff --git a/gateway/.env b/gateway/.env
index 78d3391..c84df8b 100644
--- a/gateway/.env
+++ b/gateway/.env
@@ -1,4 +1,4 @@
 POSTGRES_USER=postgres
-    POSTGRES_PASSWORD=postgres
-    POSTGRES_DB=invariantmonitor
-    POSTGRES_HOST=database
\ No newline at end of file
+POSTGRES_PASSWORD=postgres
+POSTGRES_DB=invariantmonitor
+POSTGRES_HOST=database
\ No newline at end of file
diff --git a/gateway/routes/anthropic.py b/gateway/routes/anthropic.py
index 2f3e243..9e5187b 100644
--- a/gateway/routes/anthropic.py
+++ b/gateway/routes/anthropic.py
@@ -5,20 +5,20 @@ import json
 from typing import Any, Optional
 
 import httpx
-from regex import R
-from common.config_manager import GatewayConfig, GatewayConfigManager
 from fastapi import APIRouter, Depends, Header, HTTPException, Request, Response
 from starlette.responses import StreamingResponse
+
+from common.authorization import extract_authorization_from_headers
+from common.config_manager import GatewayConfig, GatewayConfigManager
 from common.constants import (
     CLIENT_TIMEOUT,
     IGNORED_HEADERS,
 )
-from integrations.explorer import create_annotations_from_guardrails_errors, push_trace
+from common.request_context_data import RequestContextData
 from converters.anthropic_to_invariant import (
     convert_anthropic_to_invariant_message_format,
 )
-from common.authorization import extract_authorization_from_headers
-from common.request_context_data import RequestContextData
+from integrations.explorer import create_annotations_from_guardrails_errors, push_trace
 from integrations.guardrails import (
     ExtraItem,
     InstrumentedResponse,
diff --git a/gateway/routes/gemini.py b/gateway/routes/gemini.py
index 59d0874..ae86bf4 100644
--- a/gateway/routes/gemini.py
+++ b/gateway/routes/gemini.py
@@ -5,16 +5,18 @@ import json
 from typing import Any, Literal, Optional
 
 import httpx
-from common.config_manager import GatewayConfig, GatewayConfigManager
 from fastapi import APIRouter, Depends, HTTPException, Query, Request, Response
 from fastapi.responses import StreamingResponse
+
+from common.authorization import extract_authorization_from_headers
+from common.config_manager import GatewayConfig, GatewayConfigManager
 from common.constants import (
     CLIENT_TIMEOUT,
     IGNORED_HEADERS,
 )
-from common.authorization import extract_authorization_from_headers
 from common.request_context_data import RequestContextData
 from converters.gemini_to_invariant import convert_request, convert_response
+from integrations.explorer import create_annotations_from_guardrails_errors, push_trace
 from integrations.guardrails import (
     ExtraItem,
     InstrumentedResponse,
@@ -23,8 +25,6 @@ from integrations.guardrails import (
     preload_guardrails,
     check_guardrails,
 )
-from integrations.explorer import create_annotations_from_guardrails_errors, push_trace
-from integrations.guardrails import check_guardrails, preload_guardrails
 
 gateway = APIRouter()
 
diff --git a/gateway/routes/open_ai.py b/gateway/routes/open_ai.py
index 6ef3808..e3cec41 100644
--- a/gateway/routes/open_ai.py
+++ b/gateway/routes/open_ai.py
@@ -5,13 +5,16 @@ import json
 from typing import Any, Optional
 
 import httpx
-from common.config_manager import GatewayConfig, GatewayConfigManager
 from fastapi import APIRouter, Depends, Header, HTTPException, Request, Response
 from fastapi.responses import StreamingResponse
+
+from common.authorization import extract_authorization_from_headers
+from common.config_manager import GatewayConfig, GatewayConfigManager
 from common.constants import (
     CLIENT_TIMEOUT,
     IGNORED_HEADERS,
 )
+from common.request_context_data import RequestContextData
 from integrations.explorer import create_annotations_from_guardrails_errors, push_trace
 from integrations.guardrails import (
     ExtraItem,
@@ -20,8 +23,6 @@ from integrations.guardrails import (
     check_guardrails,
     preload_guardrails,
 )
-from common.authorization import extract_authorization_from_headers
-from common.request_context_data import RequestContextData
 
 gateway = APIRouter()