gpt-batch 0.1.2__tar.gz → 0.1.5__tar.gz

{gpt_batch-0.1.2 → gpt_batch-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpt_batch
-Version: 0.1.2
+Version: 0.1.5
 Summary: A package for batch processing with OpenAI API.
 Home-page: https://github.com/fengsxy/gpt_batch
 Author: Ted Yu
@@ -21,7 +21,7 @@ A simple tool to batch process messages using OpenAI's GPT models. `GPTBatcher`
 To get started with `GPTBatcher`, clone this repository to your local machine. Navigate to the repository directory and install the required dependencies (if any) by running:
 
 ```bash
-pip install -r requirements.txt
+pip install gpt_batch
 ```
 
 ## Quick Start
@@ -75,11 +75,4 @@ The `GPTBatcher` class can be customized with several parameters to adjust its p
 
 For more detailed documentation on the parameters and methods, refer to the class docstring.
 
-## License
-
-Specify your licensing information here.
-
-```
-
-This README provides clear instructions on how to install and use the `GPTBatcher`, along with detailed explanations of its configuration parameters. Adjust the "License" section as necessary based on your project's licensing terms.
 

{gpt_batch-0.1.2 → gpt_batch-0.1.5}/README.md

@@ -10,7 +10,7 @@ A simple tool to batch process messages using OpenAI's GPT models. `GPTBatcher`
 To get started with `GPTBatcher`, clone this repository to your local machine. Navigate to the repository directory and install the required dependencies (if any) by running:
 
 ```bash
-pip install -r requirements.txt
+pip install gpt_batch
 ```
 
 ## Quick Start
@@ -63,11 +63,3 @@ The `GPTBatcher` class can be customized with several parameters to adjust its p
 - **miss_index** (list): Tracks indices of requests that failed to process correctly.
 
 For more detailed documentation on the parameters and methods, refer to the class docstring.
-
-## License
-
-Specify your licensing information here.
-
-```
-
-This README provides clear instructions on how to install and use the `GPTBatcher`, along with detailed explanations of its configuration parameters. Adjust the "License" section as necessary based on your project's licensing terms.

{gpt_batch-0.1.2 → gpt_batch-0.1.5}/gpt_batch/batcher.py

@@ -57,22 +57,26 @@ class GPTBatcher:
         new_list = []
         num_workers = self.num_workers
         timeout_duration = self.timeout_duration
-        retry_attempts=2
-
+        retry_attempts = 2
+    
         executor = ThreadPoolExecutor(max_workers=num_workers)
         message_chunks = list(self.chunk_list(message_list, num_workers))
-        for chunk in tqdm(message_chunks, desc="Processing messages"):
-            future_to_message =  {executor.submit(self.get_attitude, message): message for message in chunk}
-            for _ in range(retry_attempts):
-                done, not_done = wait(future_to_message.keys(), timeout=timeout_duration)
-                for future in not_done:
-                    future.cancel()
-                new_list.extend(future.result() for future in done if future.done())
-                if len(not_done) == 0:
-                    break
-                future_to_message = {executor.submit(self.get_attitude, future_to_message[future]): future_to_message[future] for future, msg in not_done}
-        executor.shutdown(wait=False)
-        return new_list
+        try:
+            for chunk in tqdm(message_chunks, desc="Processing messages"):
+                future_to_message = {executor.submit(self.get_attitude, message): message for message in chunk}
+                for _ in range(retry_attempts):
+                    done, not_done = wait(future_to_message.keys(), timeout=timeout_duration)
+                    for future in not_done:
+                        future.cancel()
+                    new_list.extend(future.result() for future in done if future.done())
+                    if len(not_done) == 0:
+                        break
+                    future_to_message = {executor.submit(self.get_attitude, future_to_message[future]): future for future in not_done}
+        except Exception as e:
+            print(f"Error occurred: {e}")
+        finally:
+            executor.shutdown(wait=False)
+            return new_list
 
     def complete_attitude_list(self,attitude_list, max_length):
         completed_list = []

{gpt_batch-0.1.2 → gpt_batch-0.1.5}/gpt_batch.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpt-batch
-Version: 0.1.2
+Version: 0.1.5
 Summary: A package for batch processing with OpenAI API.
 Home-page: https://github.com/fengsxy/gpt_batch
 Author: Ted Yu
@@ -21,7 +21,7 @@ A simple tool to batch process messages using OpenAI's GPT models. `GPTBatcher`
 To get started with `GPTBatcher`, clone this repository to your local machine. Navigate to the repository directory and install the required dependencies (if any) by running:
 
 ```bash
-pip install -r requirements.txt
+pip install gpt_batch
 ```
 
 ## Quick Start
@@ -75,11 +75,4 @@ The `GPTBatcher` class can be customized with several parameters to adjust its p
 
 For more detailed documentation on the parameters and methods, refer to the class docstring.
 
-## License
-
-Specify your licensing information here.
-
-```
-
-This README provides clear instructions on how to install and use the `GPTBatcher`, along with detailed explanations of its configuration parameters. Adjust the "License" section as necessary based on your project's licensing terms.
 

{gpt_batch-0.1.2 → gpt_batch-0.1.5}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='gpt_batch',
-    version='0.1.2',
+    version='0.1.5',
     packages=find_packages(),
     install_requires=[
         'openai', 'tqdm'