fix (sdk): set default values to FuncResponse sdk type

Agenta-AI · Jun 6, 2024 · c50d0a9 · c50d0a9
1 parent 3b4d235
commit c50d0a9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agenta-cli/agenta/sdk/decorators/llm_entrypoint.py b/agenta-cli/agenta/sdk/decorators/llm_entrypoint.py
@@ -201,7 +201,9 @@ async def execute_function(self, func: Callable[..., Any], *args, **func_params)
             if isinstance(result, Dict):
                 return FuncResponse(**result, latency=round(latency, 4))
             if isinstance(result, str):
-                return FuncResponse(message=result, latency=round(latency, 4))  # type: ignore
+                return FuncResponse(
+                    message=result, usage=None, cost=None, latency=round(latency, 4)
+                )
             if isinstance(result, int) or isinstance(result, float):
                 return FuncResponse(
                     message=str(result),