Forráskód Böngészése

enh: tool __model__ param support

Timothy J. Baek 10 hónapja
szülő
commit
dad7af6de1
1 módosított fájl, 7 hozzáadás és 0 törlés
  1. 7 0
      backend/main.py

+ 7 - 0
backend/main.py

@@ -280,6 +280,13 @@ async def get_function_call_response(
                             "__files__": files,
                         }
 
+                    if "__model__" in sig.parameters:
+                        # Call the function with the '__model__' parameter included
+                        params = {
+                            **params,
+                            "__model__": model,
+                        }
+
                     if "__id__" in sig.parameters:
                         # Call the function with the '__id__' parameter included
                         params = {