pass client kwargs

jerpint · Nov 20, 2023 · 53efd5f · 53efd5f
1 parent a87430d
commit 53efd5f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/buster/validators/validators.py b/buster/validators/validators.py
@@ -115,6 +115,7 @@ class DocumentsValidator:
     def __init__(
         self,
         completion_kwargs: Optional[dict] = None,
+        client_kwargs: Optional[dict] = None,
         system_prompt: Optional[str] = None,
         user_input_formatter: Optional[str] = None,
         max_calls: int = 30,
@@ -142,7 +143,7 @@ def __init__(
                 "temperature": 0,
             }
 
-        self.completer = ChatGPTCompleter(completion_kwargs=completion_kwargs)
+        self.completer = ChatGPTCompleter(completion_kwargs=completion_kwargs, client_kwargs=client_kwargs)
 
         self.max_calls = max_calls