fix for the new lm app response

Agenta-AI · Jan 11, 2024 · 2e7fc64 · 2e7fc64
1 parent 1b3f505
commit 2e7fc64
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agenta-backend/agenta_backend/services/llm_apps_service.py b/agenta-backend/agenta_backend/services/llm_apps_service.py
@@ -77,7 +77,9 @@ async def invoke_app(
             url, json=payload, timeout=httpx.Timeout(timeout=5, read=None, write=5)
         )
         response.raise_for_status()
-        return AppOutput(output=response.json(), status="success")
+
+        lm_app_response = response.json()
+        return AppOutput(output=lm_app_response["message"], status="success")
 
 
 async def run_with_retry(