fix(frontend): Refactor getStringOrJson function to simplify value ha…

…ndling
Agenta-AI · Aug 21, 2024 · 1e0b1ab · 1e0b1ab
1 parent 6ebd073
commit 1e0b1ab
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/agenta-web/src/components/Playground/Views/TestView.tsx b/agenta-web/src/components/Playground/Views/TestView.tsx
@@ -568,10 +568,10 @@ const App: React.FC<TestViewProps> = ({
             // String, FuncResponse or BaseResponse
             if (typeof result === "string") {
                 res = {version: "2.0", data: result} as BaseResponse
-                setResultForIndex(getStringOrJson(res), index)
+                setResultForIndex(getStringOrJson(res.data), index)
             } else if (isFuncResponse(result)) {
                 res = {version: "2.0", data: result.message} as BaseResponse
-                setResultForIndex(getStringOrJson(res), index)
+                setResultForIndex(getStringOrJson(res.data), index)
 
                 const {message, cost, latency, usage} = result
                 setAdditionalDataList((prev) => {
@@ -581,7 +581,7 @@ const App: React.FC<TestViewProps> = ({
                 })
             } else if (isBaseResponse(result)) {
                 res = result as BaseResponse
-                setResultForIndex(getStringOrJson(res), index)
+                setResultForIndex(getStringOrJson(res.data), index)
 
                 const {data, trace} = result
                 setAdditionalDataList((prev) => {

diff --git a/agenta-web/src/lib/helpers/utils.ts b/agenta-web/src/lib/helpers/utils.ts
@@ -338,11 +338,7 @@ export const getInitials = (str: string, limit = 2) => {
 }
 
 export const getStringOrJson = (value: any) => {
-    return typeof value === "string"
-        ? value
-        : typeof value?.data === "string"
-          ? value?.data
-          : JSON.stringify(value.data, null, 2)
+    return typeof value === "string" ? value : JSON.stringify(value, null, 2)
 }
 
 export const filterVariantParameters = ({