enh: tool __id__ param support for cache dir

enh: tool id param support for cache dir
dcac1a3c · Timothy J. Baek · 6f9a31eb · dcac1a3c
Commit dcac1a3c authored Jun 18, 2024 by Timothy J. Baek
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

backend/main.py backend/main.py +7 -0

No files found.
--- a/backend/main.py
+++ b/backend/main.py
@@ -280,6 +280,13 @@ async def get_function_call_response(
                            "__files__": files,
                        }

+                    if "__id__" in sig.parameters:
+                        # Call the function with the '__id__' parameter included
+                        params = {
+                            **params,
+                            "__id__": tool_id,
+                        }
+
                    function_result = function(**params)
                except Exception as e:
                    print(e)