From 5edc211392ebff06ac3687021f5ea6c04d75e065 Mon Sep 17 00:00:00 2001
From: Michael Poluektov <michael.poluektov@gmail.com>
Date: Mon, 19 Aug 2024 16:27:21 +0100
Subject: [PATCH] pass docstring to function

---
 backend/utils/tools.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/backend/utils/tools.py b/backend/utils/tools.py
index 12642ccfd..14519f1be 100644
--- a/backend/utils/tools.py
+++ b/backend/utils/tools.py
@@ -60,9 +60,10 @@ def get_tools(
             function_name = spec["name"]
 
             # convert to function that takes only model params and inserts custom params
-            callable = apply_extra_params_to_tool_function(
-                getattr(module, function_name), extra_params
-            )
+            original_func = getattr(module, function_name)
+            callable = apply_extra_params_to_tool_function(original_func, extra_params)
+            if hasattr(original_func, "__doc__"):
+                callable.__doc__ = original_func.__doc__
 
             # TODO: This needs to be a pydantic model
             tool_dict = {