add seed parameter (#560)

truera · Nov 16, 2023 · 555ebad · 555ebad
1 parent 66a17c6
commit 555ebad
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/trulens_eval/trulens_eval/feedback/provider/openai.py b/trulens_eval/trulens_eval/feedback/provider/openai.py
@@ -64,6 +64,9 @@ def _create_chat_completion(
         if 'temperature' not in kwargs:
             kwargs['temperature'] = 0.0
 
+        if 'seed' not in kwargs:
+            kwargs['seed'] = 123
+
         if prompt is not None:
             completion = self.endpoint.client.chat.completions.create(messages=[{
                 "role": "system",