fix openai start zone generation

neph1 · Sep 29, 2023 · decaeea · decaeea
1 parent 774a253
commit decaeea
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tale/llm/llm_utils.py b/tale/llm/llm_utils.py
@@ -516,9 +516,11 @@ def generate_start_zone(self, location_desc: str, story_type: str, story_context
         if self.backend == 'kobold_cpp':
             request_body = self._kobold_generation_prompt(request_body)
             request_body['prompt'] = prompt
+            request_body['max_length'] = 750
         elif self.backend == 'openai':
             request_body['messages'][1]['content'] = prompt
-        request_body['max_length'] = 750
+            request_body['max_tokens'] = 750
+
         result = self.io_util.synchronous_request(request_body)
         try:
             json_result = json.loads(parse_utils.sanitize_json(result))