From 7a07c69089698bf67bed727a3a6156052334cafa Mon Sep 17 00:00:00 2001
From: Josh XT <josh@devxt.com>
Date: Sun, 5 Jan 2025 11:46:30 -0500
Subject: [PATCH] Update chain

---
 agixt/endpoints/Chain.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/agixt/endpoints/Chain.py b/agixt/endpoints/Chain.py
index a249713bf129..4cbff1220bf3 100644
--- a/agixt/endpoints/Chain.py
+++ b/agixt/endpoints/Chain.py
@@ -50,6 +50,9 @@ async def get_chain(chain_name: str, user=Depends(verify_api_key)):
     "/api/chain/{chain_name}/run",
     tags=["Chain"],
     dependencies=[Depends(verify_api_key)],
+    response_model=str,
+    summary="Run chain",
+    description="Executes a chain with the specified name and returns the final output.",
 )
 async def run_chain(
     chain_name: str,
@@ -86,6 +89,9 @@ async def run_chain(
     "/api/chain/{chain_name}/run/step/{step_number}",
     tags=["Chain"],
     dependencies=[Depends(verify_api_key)],
+    response_model=str,
+    summary="Run chain step",
+    description="Executes a specific step within a chain and returns the output.",
 )
 async def run_chain_step(
     chain_name: str,