store confidence_float in db (#446)

2024-06-07 10:57:53 -07:00
parent 12cfef09d4
commit 3801bcbf19
2 changed files with 20 additions and 6 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -854,7 +854,7 @@ class ForgeAgent:
        return json.dumps(
            [
                {
-                    "action": action.model_dump(exclude_none=True, exclude={"text"}),
+                    "action": action.model_dump(exclude_none=True, exclude={"text", "confidence_float"}),
                    "results": [
                        result.model_dump(
                            exclude_none=True,