py-adtools 0.1.9__tar.gz → 0.1.10__tar.gz

{py_adtools-0.1.9 → py_adtools-0.1.10}/PKG-INFO

@@ -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.9 → py_adtools-0.1.10}/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,

{py_adtools-0.1.9 → py_adtools-0.1.10}/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
     ):

{py_adtools-0.1.9 → py_adtools-0.1.10}/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

{py_adtools-0.1.9 → py_adtools-0.1.10}/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 → py_adtools-0.1.10}/py_adtools.egg-info/PKG-INFO

@@ -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.9 → py_adtools-0.1.10}/setup.py

@@ -5,7 +5,7 @@ with open('README.md', 'r', encoding='utf-8') as fh:
 
 setup(
     name='py-adtools',
-    version='0.1.9',
+    version='0.1.10',
     author='Rui Zhang',
     author_email='rzhang.cs@gmail.com',
     description='Useful tools for parsing and evaluating Python programs for LLM-based algorithm design.',