update our websever

Author: dtam

guardrails_api/app.py

@@ -127,7 +127,7 @@ def create_app(
         pg_client.initialize(app)
 
     cache_client = CacheClient()
-    cache_client.initialize(app)
+    cache_client.initialize()
 
     from guardrails_api.api.root import router as root_router
     from guardrails_api.api.guards import router as guards_router, guard_client

guardrails_api/cli/start.py

@@ -3,7 +3,7 @@
 from guardrails_api.cli.cli import cli
 from guardrails_api.app import create_app
 from guardrails_api.utils.configuration import valid_configuration
-
+import uvicorn
 
 @cli.command("start")
 def start(
@@ -25,4 +25,5 @@ def start(
     env = env or None
     config = config or None
     valid_configuration(config)
-    create_app(env, config, port).run(port=port)
+    app = create_app(env, config, port)
+    uvicorn.run(app, port=port)

pyproject.toml

@@ -11,7 +11,6 @@ keywords = ["Guardrails", "Guardrails AI", "Guardrails API", "Guardrails API"]
 requires-python = ">= 3.8.1"
 dependencies = [
     "guardrails-ai>=0.5.10",
-    "Werkzeug>=3.0.3,<4",
     "jsonschema>=4.22.0,<5",
     "referencing>=0.35.1,<1",
     "boto3>=1.34.115,<2",
@@ -27,6 +26,7 @@ dependencies = [
     "aiocache>=0.11.1",
     "fastapi>=0.114.1",
     "SQLAlchemy>=2.0.34",
+    "uvicorn>=0.30.6",
 ]
 
 [tool.setuptools.dynamic]
@@ -42,7 +42,6 @@ dev = [
     "coverage",
     "pytest-mock",
     "gunicorn>=22.0.0,<23",
-    "uvicorn",
 ]
 
 [tool.pytest.ini_options]

setup.py

@@ -16,8 +16,7 @@
     packages=find_packages(),
     python_requires=">=3.8, <4",
     install_requires=[
-        "guardrails-ai>=0.5.6",
-        "Werkzeug>=3.0.3,<4",
+        "guardrails-ai>=0.5.10",
         "jsonschema>=4.22.0,<5",
         "referencing>=0.35.1,<1",
         "boto3>=1.34.115,<2",
@@ -33,6 +32,7 @@
         "aiocache>=0.11.1",
         "fastapi>=0.114.1",
         "SQLAlchemy>=2.0.34",
+        "uvicorn>=0.30.6",
     ],
     package_data={"guardrails_api": ["py.typed", "open-api-spec.json"]},
 )