fixes

premAI-io · Jun 10, 2024 · fc5263d · fc5263d
1 parent da4ca42
commit fc5263d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/prem_utils/connectors/anthropic.py b/prem_utils/connectors/anthropic.py
@@ -1,4 +1,5 @@
 from collections.abc import Sequence
+from uuid import uuid4
 
 from anthropic import (
     Anthropic,
@@ -91,7 +92,12 @@ def _get_content(self, response, tools=False):
                 "content": "",
                 "role": "assistant",
                 "tool_calls": [
-                    {"function_arguments": tool_message.input, "name": tool_message.name, "type": "function"}
+                    {
+                        "function_arguments": tool_message.input,
+                        "name": tool_message.name,
+                        "type": "function",
+                        "id": str(uuid4()),
+                    }
                     for tool_message in tool_messages
                 ],
             }