Merge branch 'refs/heads/main' into eugene1/tlk-2622-instantiate-fast…

…api-app-once-when-building-test-fixtures

# Conflicts:
#	src/backend/main.py

src/backend/chat/enums.py

@@ -1,11 +1,10 @@
-from enum import Enum
+from enum import StrEnum
 
 
-class StreamEvent(str, Enum):
+class StreamEvent(StrEnum):
     """
     Stream Events returned by Cohere's chat stream response.
     """
-
     STREAM_START = "stream-start"
     SEARCH_QUERIES_GENERATION = "search-queries-generation"
     SEARCH_RESULTS = "search-results"

src/backend/config/routers.py

@@ -22,6 +22,7 @@ class RouterName(StrEnum):
     CONVERSATION = "conversation"
     DEPLOYMENT = "deployment"
     EXPERIMENTAL_FEATURES = "experimental_features"
+    ORGANIZATION = "organization"
     TOOL = "tool"
     USER = "user"
     AGENT = "agent"
@@ -90,6 +91,17 @@ class RouterName(StrEnum):
             Depends(validate_organization_header),
         ],
     },
+    RouterName.ORGANIZATION: {
+        "default": [
+            Depends(get_session),
+            Depends(validate_organization_header),
+        ],
+        "auth": [
+            Depends(get_session),
+            Depends(validate_authorization),
+            Depends(validate_organization_header),
+        ],
+    },
     RouterName.TOOL: {
         "default": [
             Depends(get_session),

src/backend/main.py

@@ -45,6 +45,9 @@
 ORIGINS = ["*"]
 
 
+_RELEASE_VERSION = "v1.1.5"
+
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # Retrieves all the Auth provider endpoints if authentication is enabled.
@@ -55,7 +58,12 @@ async def lifespan(app: FastAPI):
 
 
 def create_app() -> FastAPI:
-    app = FastAPI(lifespan=lifespan)
+    app = FastAPI(
+        title="Cohere Toolkit API",
+        description="This is the API for the Open Source Cohere Toolkit",
+        version=_RELEASE_VERSION,
+        lifespan=lifespan,
+    )
 
     routers = [
         auth_router,
@@ -151,7 +159,11 @@ async def health():
     return {"status": "OK"}
 
 
-@app.post("/migrate", dependencies=[Depends(verify_migrate_token)])
+@app.post(
+    "/migrate",
+    dependencies=[Depends(verify_migrate_token)],
+    include_in_schema=False,
+)
 async def apply_migrations():
     """
     Applies Alembic migrations - useful for serverless applications