release: 0.1.0

Now it saves both prompts and completions
superheavytail · Jun 2, 2024 · 29e76ae · 29e76ae
1 parent 5998095
commit 29e76ae
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/batched_chatgpt/__init__.py b/batched_chatgpt/__init__.py
@@ -1,3 +1,3 @@
 from .chatgpt_utils import call_chatgpt
 
-__version__ = '0.0.5'
+__version__ = '0.1.0'
diff --git a/batched_chatgpt/chatgpt_utils.py b/batched_chatgpt/chatgpt_utils.py
@@ -93,7 +93,7 @@ def batched_multiprocess_auto_retry(
                 outputs[remain_indices[i * chunk_size + j]] = result
 
             # save the outputs which may be incomplete
-            pickle_bobj(outputs, pkl_path) if pkl_path else None
+            pickle_bobj({'prompts': items, 'completions': outputs}, pkl_path) if pkl_path else None
 
             time.sleep(sleep_between_chunk) if not all(outputs) else ...
     return outputs

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="batched_chatgpt",
-    version="0.0.5",
+    version="0.1.0",
     description="Easy calling chatgpt with batched instances",
     packages=find_packages(),
     author="superheavytail",