Timothy J. Baek 10 місяців тому
батько
коміт
f14ca48334

+ 10 - 10
src/lib/components/workspace/Functions/FunctionEditor.svelte

@@ -39,6 +39,7 @@ class Filter:
     def __init__(self):
         # Indicates custom file handling logic. This flag helps disengage default routines in favor of custom
         # implementations, informing the WebUI to defer file-related operations to designated methods within this class.
+        # Alternatively, you can remove the files directly from the body in from the inlet hook
         self.file_handler = True
 
         # Initialize 'valves' with specific configurations. Using 'Valves' instance helps encapsulate settings,
@@ -50,16 +51,15 @@ class Filter:
         # Modify the request body or validate it before processing by the chat completion API.
         # This function is the pre-processor for the API where various checks on the input can be performed.
         # It can also modify the request before sending it to the API.
+        print(f"inlet:{__name__}")
+        print(f"inlet:body:{body}")
+        print(f"inlet:user:{user}")
 
-        print("inlet")
-        print(body)
-        print(user)
-
-        if user.get("role", "admin") in ["user"]:
+        if user.get("role", "admin") in ["user", "admin"]:
             messages = body.get("messages", [])
-            if len(messages) > self.max_turns:
+            if len(messages) > self.valves.max_turns:
                 raise Exception(
-                    f"Conversation turn limit exceeded. Max turns: {self.max_turns}"
+                    f"Conversation turn limit exceeded. Max turns: {self.valves.max_turns}"
                 )
 
         return body
@@ -68,9 +68,9 @@ class Filter:
         # Modify or analyze the response body after processing by the API.
         # This function is the post-processor for the API, which can be used to modify the response
         # or perform additional checks and analytics.
-        print(f"outlet")
-        print(body)
-        print(user)
+        print(f"outlet:{__name__}")
+        print(f"outlet:body:{body}")
+        print(f"outlet:user:{user}")
 
         return body`;