py-adtools 0.1.9__py3-none-any.whl → 0.1.10__py3-none-any.whl

adtools/evaluator.py

@@ -160,10 +160,13 @@ class PyEvaluator(ABC):
             with open(os.devnull, 'w') as devnull:
                 os.dup2(devnull.fileno(), sys.stdout.fileno())
                 os.dup2(devnull.fileno(), sys.stderr.fileno())
-
-        # Evaluate and put the results to the queue
-        res = self.evaluate(program_str, **kwargs)
-        result_queue.put(res)
+        try:
+            # Evaluate and put the results to the queue
+            res = self.evaluate(program_str, **kwargs)
+            result_queue.put(res)
+        except:
+            traceback.print_exc()
+            result_queue.put(None)
 
     def secure_evaluate(
             self,

adtools/lm/openai_api.py

@@ -6,7 +6,7 @@ Commercial use of this software or its derivatives requires prior written permis
 """
 
 import os
-from typing import List
+from typing import List, Optional
 
 import openai.types.chat
 
@@ -46,8 +46,8 @@ class OpenAIAPI(LanguageModel):
     def chat_completion(
             self,
             message: str | List[openai.types.chat.ChatCompletionMessageParam],
-            max_tokens: int,
-            timeout_seconds: float,
+            max_tokens: Optional[int] = None,
+            timeout_seconds: Optional[float] = None,
             *args,
             **kwargs
     ):

adtools/lm/vllm_server.py

@@ -75,6 +75,7 @@ class VLLMServer(LanguageModel):
             env_variable_dict: Environment variables to use for vLLM server, e.g., {'KEY': 'VALUE'}.
             vllm_serve_args: Arguments to pass to vLLM server, e.g., ['--enable-reasoning'].
             vllm_serve_kwargs: Keyword arguments to pass to vLLM server, e.g., {'--reasoning-parser': 'deepseek-r1'}.
+            chat_template_kwargs: Keyword arguments to pass to chat template, e.g., {'enable_thinking': False}.
 
         Example:
             # deploy a model on GPU 0 and 1

adtools/py_code.py

@@ -163,11 +163,14 @@ class PyProgram:
         return program
 
     @classmethod
-    def from_text(cls, text: str) -> 'PyProgram':
-        tree = ast.parse(text)
-        visitor = _ProgramVisitor(text)
-        visitor.visit(tree)
-        return visitor.return_program()
+    def from_text(cls, text: str) -> Optional['PyProgram']:
+        try:
+            tree = ast.parse(text)
+            visitor = _ProgramVisitor(text)
+            visitor.visit(tree)
+            return visitor.return_program()
+        except:
+            return None
 
 
 class _ProgramVisitor(ast.NodeVisitor):

{py_adtools-0.1.9.dist-info → py_adtools-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: py-adtools
-Version: 0.1.9
+Version: 0.1.10
 Summary: Useful tools for parsing and evaluating Python programs for LLM-based algorithm design.
 Home-page: https://github.com/RayZhhh/py-adtools
 Author: Rui Zhang

py_adtools-0.1.10.dist-info/RECORD

@@ -0,0 +1,13 @@
+adtools/__init__.py,sha256=2XOq8zj8mpejJFavtARmst_hih6qAtBPKBoNB0UWeqs,183
+adtools/evaluator.py,sha256=kgM06gIbiOWtjghmAqT1q-lrtwxn2CU5840bGueN4-A,26443
+adtools/evaluator_pool.py,sha256=zfQ7DgAjpByqPM5SE1tRQT_HGXU5uwNntvguzrXaPEk,3258
+adtools/py_code.py,sha256=1GvJAfdHGdJ6z9J0uamxi8D13u8ydzcmunaESfKoNos,14356
+adtools/lm/__init__.py,sha256=PZf5Lraly9rAWz-cxOSLCvZ9OZ4EI8aQEluetvNX8LM,146
+adtools/lm/lm_base.py,sha256=KtO7KTrrMW7oWN-BhncoIOsbOVQsSc-0gNCYtvR6Sog,1105
+adtools/lm/openai_api.py,sha256=6F2HGJL3fyzAixfkPqGEPd-9kns8bMp1mqWkRocmAig,2434
+adtools/lm/vllm_server.py,sha256=tn4NBVEAXoLwrS8_BYLK4xpWfcELrzzoKnBI2jEorOE,13054
+py_adtools-0.1.10.dist-info/licenses/LICENSE,sha256=E5GGyecx3y5h2gcEGQloF-rDY9wbaef5IHjRsvtFbt8,1065
+py_adtools-0.1.10.dist-info/METADATA,sha256=9YWOFbdHufsg2SzjRm94bBwsR8K1u9i_IVVummPo3PM,6387
+py_adtools-0.1.10.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
+py_adtools-0.1.10.dist-info/top_level.txt,sha256=X2kKzmJFDAKR2FWCij5pfMG9pVVjVUomyl4e-1VLXIk,8
+py_adtools-0.1.10.dist-info/RECORD,,

py_adtools-0.1.9.dist-info/RECORD

@@ -1,13 +0,0 @@
-adtools/__init__.py,sha256=2XOq8zj8mpejJFavtARmst_hih6qAtBPKBoNB0UWeqs,183
-adtools/evaluator.py,sha256=v-tdE5emOa7o6615YN2Av7fuh5W5hIlHoSjc-rTO1uU,26334
-adtools/evaluator_pool.py,sha256=zfQ7DgAjpByqPM5SE1tRQT_HGXU5uwNntvguzrXaPEk,3258
-adtools/py_code.py,sha256=FZfkp-IZ4zpOjrWe6svKNJsQhVANaTTkE0l0mc4aMW8,14277
-adtools/lm/__init__.py,sha256=PZf5Lraly9rAWz-cxOSLCvZ9OZ4EI8aQEluetvNX8LM,146
-adtools/lm/lm_base.py,sha256=KtO7KTrrMW7oWN-BhncoIOsbOVQsSc-0gNCYtvR6Sog,1105
-adtools/lm/openai_api.py,sha256=LcfLkNOBrJTdsp0zcUjaCelIcQK5XknpHWrlB0S67_k,2390
-adtools/lm/vllm_server.py,sha256=BPZoTS77wNJDcJ_0FO2QFyZTf6WR0isYKMuTctqKEU8,12942
-py_adtools-0.1.9.dist-info/licenses/LICENSE,sha256=E5GGyecx3y5h2gcEGQloF-rDY9wbaef5IHjRsvtFbt8,1065
-py_adtools-0.1.9.dist-info/METADATA,sha256=Fnu9fmNWBgI8l4nuvrIwYHXGUu4vweTH8b_uziAB4sM,6386
-py_adtools-0.1.9.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
-py_adtools-0.1.9.dist-info/top_level.txt,sha256=X2kKzmJFDAKR2FWCij5pfMG9pVVjVUomyl4e-1VLXIk,8
-py_adtools-0.1.9.dist-info/RECORD,,