Resetting utils subtree.

2024-12-05 14:47:16 +05:30
parent 6e6e6be87c
commit 75b4bf541b
151 changed files with 139 additions and 125374 deletions
@@ -331,7 +331,12 @@ class MailSyncModel(BaseModel):
            )
            llm_json = {
                "snippet": llm_response.content,
-                "usage": llm_response.usage_metadata
+                "usage": {
+                    "input": llm_response.usage_metadata["input_tokens"],
+                    "output": llm_response.usage_metadata["output_tokens"],
+                    "total": llm_response.usage_metadata["total_tokens"],
+                },
+                "rawUsage": llm_response.usage_metadata
            }
        client_response.data["aiSnippet"] = llm_json