quantalogic 0.61.1__py3-none-any.whl → 0.61.3__py3-none-any.whl

{quantalogic-0.61.1.dist-info → quantalogic-0.61.3.dist-info}/RECORD

@@ -1,10 +1,10 @@
 quantalogic/__init__.py,sha256=qFbvfHOd_chAu536pH816E3uo6CdyAgXCpQOwMXXVnY,1076
 quantalogic/agent.py,sha256=VChZXFLEsIIrBtXVQZ-FGZ72GCUXDL3g9br8Vo1t5V8,75072
-quantalogic/agent_config.py,sha256=65dvgJ1rwC0eEysMtTv_Dsjln4WS0IvAIcn6r39Zq4k,8222
+quantalogic/agent_config.py,sha256=HNXHmu3BiT4q9i-L2wJKUh6PuiZ_ryTNaB7ga24kaJo,8334
 quantalogic/agent_factory.py,sha256=soActorasOqs5g6NmlyeEjRYbJceIGGg7wuUS-vA898,6878
 quantalogic/codeact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-quantalogic/codeact/agent.py,sha256=oajib7OsfBp230sg_IjmknqYsFz_lSURC39fj5brsP4,22454
-quantalogic/codeact/cli.py,sha256=FBqnqPUobq6ySuTL9oTLyXSenHC6V7wsCWAbNDK_xec,10810
+quantalogic/codeact/agent.py,sha256=lF5Cq9e_XrBrGwhEZoNNbV-FxAMM2S12_u2og5W4994,22445
+quantalogic/codeact/cli.py,sha256=Nxo5Biv08yvA72c860ikKI6Ebp0vlygoVrIMCpYsSxA,10867
 quantalogic/codeact/constants.py,sha256=_xJ2QwP5wUE9vwSDm0JTSMC4GVjXAaxHTwcZBjDCoKk,244
 quantalogic/codeact/events.py,sha256=4itxGsz54Te9vim13JC8QhacnLQ8cqtOOIyOWUrKFi0,1530
 quantalogic/codeact/llm_util.py,sha256=9HqN0prWyAgdt6FYtdzjWoIoPpDavHQ7EIIfJr2Fi8w,2749
@@ -12,9 +12,9 @@ quantalogic/codeact/prompts/error_format.j2,sha256=_8fNWxDr3U1Fq8SdzwRIBNsNG1WvP
 quantalogic/codeact/prompts/generate_action.j2,sha256=v0LXFxz6L2EwIPesu_mYSVqyNctjHOCIfdawHCb2rtc,1289
 quantalogic/codeact/prompts/generate_program.j2,sha256=05_1bAwIJLLBKt4o9c7sRWLHIoiHPyGl9OquieMZT5g,2048
 quantalogic/codeact/prompts/response_format.j2,sha256=TwO43dEG-3justNigpX4yyzGR1TGS3YDlpJQq8Z5Vf4,355
-quantalogic/codeact/tools_manager.py,sha256=l2cpHLuQP-MW0uP8d2e6IdMToeveaIOdilwDGPjPciY,4464
+quantalogic/codeact/tools_manager.py,sha256=Q_BGtqN3E8QGbobtaYMpQXqqTv_eb5HmIoZelleUMeI,4491
 quantalogic/codeact/utils.py,sha256=B7Xowk4id6Sgoyl0eBJsi6JiKa-BAwnj0a5z70vrV1M,5308
-quantalogic/coding_agent.py,sha256=i4QYY1Q7AXfJFCVfrowxukH33nuqByUOX6bt-hAcGDE,5504
+quantalogic/coding_agent.py,sha256=WFfabRwwPZFV3Pw3seLKpSrFE9Li4pz8Z8mCdsUIDi8,5532
 quantalogic/config.py,sha256=bmPI2rvQ9M8Zdl1H7K588JgJdnkxSE0L5_i5aBqsagA,564
 quantalogic/console_print_events.py,sha256=yDtfOr7s5r_gLTgwkl_XoKSkUqNRZhqqq4hwR_oJsUw,2050
 quantalogic/console_print_token.py,sha256=5IRVoPhwWZtSc4LpNoAsCQhCB_RnAW9chycGgyD3_5U,437
@@ -57,25 +57,6 @@ quantalogic/prompts/task_summary_prompt.j2,sha256=fcjV96nqi6jsfR8l6uyep20rCaOi-z
 quantalogic/prompts/tools_prompt.j2,sha256=ZjvTAZtAk-FmzDb1QuyDJg1hzo-FqayQ7wN_ytHAi2s,385
 quantalogic/prompts/variables_prompt.j2,sha256=N3OsMbzy3PfEUqYZVF_34wVidhh2ghVToSgdRCv2qgM,231
 quantalogic/prompts.py,sha256=zqmyosq4hdYpzHI-FG0m5CKV2wQnjgyMJf6Vo1ZaLW4,1982
-quantalogic/python_interpreter/__init__.py,sha256=cNYxH9jy5QEAPZo44zmJzy3Kci14e0lL2h20Cbrs7RQ,675
-quantalogic/python_interpreter/assignment_visitors.py,sha256=t6ttdYuwjcz8VtqJS5vMaaUUG6GbEhoNmnNDUj8dZiI,2771
-quantalogic/python_interpreter/base_visitors.py,sha256=aD1pc3DbdlPdjQ3yJTYhBqNpN6tant7NOHXShts8coo,765
-quantalogic/python_interpreter/class_visitors.py,sha256=FUgBVg_TmpAdLdLlYlPnruSm8iViijW3gTBQveJjkpI,862
-quantalogic/python_interpreter/comprehension_visitors.py,sha256=AdPn047QUeoG5D1VD4wU2JuXS2KBTwLOIGbApmx9s7A,8468
-quantalogic/python_interpreter/context_visitors.py,sha256=_4-Xfk0wzsla2rSIJZYEyYK4yIK7yRZWX6-HNBFSWTs,2137
-quantalogic/python_interpreter/control_flow_visitors.py,sha256=trJ0Led1pPWKyseQF7BbIF5LFopkOiRDSVHz9Hi7-hI,3662
-quantalogic/python_interpreter/exception_visitors.py,sha256=dbRuk2CaEZN-tYrzcRxZQh_UdLfpcW7wwrG8iQoGeE8,4571
-quantalogic/python_interpreter/exceptions.py,sha256=mL4G9a0PZrvW_hAjkweMohvN8ryZsZOiIOolZUDSq9Y,1296
-quantalogic/python_interpreter/execution.py,sha256=bggzv-GMnFe0Pkt00T4IlPUnDMsuqcHO1Qo0Wq6rEbs,8420
-quantalogic/python_interpreter/function_utils.py,sha256=1LCEqCDt0p0uBTeC6rcre84r_j06jc6HuWLZ5Fc8m8M,18133
-quantalogic/python_interpreter/function_visitors.py,sha256=WcGBWXZ_A7vekkrejL2XTuVPyqsKymK4svJr6-kR-OI,11317
-quantalogic/python_interpreter/import_visitors.py,sha256=AgqFM8SO1wuJluDRsDGBwm-FQ5K374pQrXgbXXEBgPI,1331
-quantalogic/python_interpreter/interpreter_core.py,sha256=PV_XFx0xhUG9ACvhz3FGat2FSaXK3C031HjRizHbox0,16646
-quantalogic/python_interpreter/literal_visitors.py,sha256=TX9-02rn093OFTDvKkhp0PbTsXqIU1voBfbxs-TXfu0,3715
-quantalogic/python_interpreter/misc_visitors.py,sha256=_olOaGD04icqKDqeEFUPIY5Nq0cH5pdGkq-8T6xbilM,6758
-quantalogic/python_interpreter/operator_visitors.py,sha256=WTf4EfwE88lMoTKjuL8UBNpbwJ8Z6DTdrwvVffVw-Ic,4048
-quantalogic/python_interpreter/scope.py,sha256=HSSbtDAMeDbE5LjSckTHvFlhcOUGv8UzxpI4NgUbp5U,257
-quantalogic/python_interpreter/visit_handlers.py,sha256=dGh7BxwGArnBnmwxVzpV6l_uXQ5J6esUm2xKtXne30M,2610
 quantalogic/quantlitellm.py,sha256=nf-awyOxP0ANoAPGHNvHfdLu8gNn65L39gl7x4saIQc,5550
 quantalogic/search_agent.py,sha256=tr0cwscJ4wu_G1aumjFyvGHQ0eQv5OL5sxj17s6Ocls,2470
 quantalogic/server/__init__.py,sha256=8sz_PYAUCrkM6JM5EAUeIzNM4NPW6j6UT72JVkc21WQ,91
@@ -91,6 +72,7 @@ quantalogic/task_runner.py,sha256=NB7TqNuwCstCAsTkjGcJSQRapNk8-iXe7d_2qf-fs1s,15
 quantalogic/tool_manager.py,sha256=vNA7aBKgdU3wpw_goom6i9rg_64pNZapNxvg4cUhhCI,6983
 quantalogic/tools/__init__.py,sha256=NU_M6VYYaAbSUtb2Qdu1lsYaDh0G3f_8jnrZTsBD0eY,2390
 quantalogic/tools/action_gen.py,sha256=M16voPq7tMR1qiCKTaZAusmVchFbBIluuBbLQH9kEG4,15580
+quantalogic/tools/action_gen_safe.py,sha256=7doUtPnV9oxBc02N_N-xcMxmfT9Nd6gHfkNU7MtwVUk,13996
 quantalogic/tools/agent_tool.py,sha256=MXCXxWHRch7VK4UWhtRP1jeI8Np9Ne2CUGo8vm1oZiM,3064
 quantalogic/tools/composio/__init__.py,sha256=Yo9ygNx0qQILVhIfRgqpO8fgnCgp5WoZMd3Hm5D20GY,429
 quantalogic/tools/composio/composio.py,sha256=icVHA_Scr1pViBhahGGBGBRBl9JSB3hGSqpgQzAIUH8,17627
@@ -188,7 +170,7 @@ quantalogic/tools/utils/generate_database_report.py,sha256=IU_XGTDNXfJXxzpHR1F4c
 quantalogic/tools/web_navigation/__init__.py,sha256=O7SkVqbGwN4zt8Sm3H8AHF9451FSgI5h0J3fDj1rFS4,142
 quantalogic/tools/web_navigation/web_tool.py,sha256=AxAxQLUNwCElqxP2ceOOjHVY80ck-Md-uNsjHdR9ErA,4721
 quantalogic/tools/wikipedia_search_tool.py,sha256=LXQSPH8961Efw2QNxKe-cD5ZiIYD3ufEgrxH4y5uB74,5180
-quantalogic/tools/write_file_tool.py,sha256=qyRhDwE77rakRb_CuAC00V0E2vqIRNxwKD4PHLGgyW8,5230
+quantalogic/tools/write_file_tool.py,sha256=cOheTr2ZTWNbPHlB1PRhc1btJ2mW8TvE5PLQFIdbUb8,5578
 quantalogic/utils/__init__.py,sha256=E442CJQuTohKzgI0Wrd4NZEpKascFjz6F4Vy8Y1c_0Y,634
 quantalogic/utils/ask_user_validation.py,sha256=kSr7TXPTpsLR9zgwpGWgvffx8-cKAC_rdFRdLqwC22A,1176
 quantalogic/utils/async_utils.py,sha256=FOizWRbHdsZwoD36dNErzunfwPlE7zDprS6RXcWuWSo,963
@@ -209,8 +191,8 @@ quantalogic/version_check.py,sha256=JyQFTNMDWtpHCLnN-BiakzB2cyXf6kUFsTjvmSruZi4,
 quantalogic/welcome_message.py,sha256=o4tHdgabNuIV9kbIDPgS3_2yzJhayK30oKad2UouYDc,3020
 quantalogic/xml_parser.py,sha256=bLLwIwO-VEHWF3heNS7nuPC8wgdYw9F_fVZZNW1figY,11728
 quantalogic/xml_tool_parser.py,sha256=hGHA1q20JUoTNTbZYmi4FTdA5I25-AGEIP8DwZgQCNA,3897
-quantalogic-0.61.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-quantalogic-0.61.1.dist-info/METADATA,sha256=k40t7hc00o09Fz4lEn4bqdk4XIOQRKIrRg6Jg3U0QoU,32998
-quantalogic-0.61.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-quantalogic-0.61.1.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
-quantalogic-0.61.1.dist-info/RECORD,,
+quantalogic-0.61.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+quantalogic-0.61.3.dist-info/METADATA,sha256=6VPi8J-lWUhZsEKVuCKFuJaWChJ2bHRO6qCQN47I_Cw,33029
+quantalogic-0.61.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+quantalogic-0.61.3.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
+quantalogic-0.61.3.dist-info/RECORD,,

quantalogic/python_interpreter/__init__.py

@@ -1,23 +0,0 @@
-# quantalogic/utils/__init__.py
-from .exceptions import BreakException, ContinueException, ReturnException, WrappedException, has_await
-from .execution import AsyncExecutionResult, execute_async, interpret_ast, interpret_code
-from .function_utils import AsyncFunction, Function, LambdaFunction
-from .interpreter_core import ASTInterpreter
-from .scope import Scope
-
-__all__ = [
-    'ASTInterpreter',
-    'execute_async',
-    'interpret_ast',
-    'interpret_code',
-    'AsyncExecutionResult',
-    'ReturnException',
-    'BreakException',
-    'ContinueException',
-    'WrappedException',
-    'has_await',
-    'Function',
-    'AsyncFunction',
-    'LambdaFunction',
-    'Scope',
-]

quantalogic/python_interpreter/assignment_visitors.py

@@ -1,63 +0,0 @@
-import ast
-from typing import Any
-
-from .interpreter_core import ASTInterpreter
-
-async def visit_Assign(self: ASTInterpreter, node: ast.Assign, wrap_exceptions: bool = True) -> None:
-    value: Any = await self.visit(node.value, wrap_exceptions=wrap_exceptions)
-    for target in node.targets:
-        if isinstance(target, ast.Subscript):
-            obj = await self.visit(target.value, wrap_exceptions=wrap_exceptions)
-            key = await self.visit(target.slice, wrap_exceptions=wrap_exceptions)
-            obj[key] = value
-        else:
-            await self.assign(target, value)
-
-async def visit_AugAssign(self: ASTInterpreter, node: ast.AugAssign, wrap_exceptions: bool = True) -> Any:
-    if isinstance(node.target, ast.Name):
-        current_val: Any = self.get_variable(node.target.id)
-    else:
-        current_val: Any = await self.visit(node.target, wrap_exceptions=wrap_exceptions)
-    right_val: Any = await self.visit(node.value, wrap_exceptions=wrap_exceptions)
-    op = node.op
-    if isinstance(op, ast.Add):
-        result: Any = current_val + right_val
-    elif isinstance(op, ast.Sub):
-        result = current_val - right_val
-    elif isinstance(op, ast.Mult):
-        result = current_val * right_val
-    elif isinstance(op, ast.Div):
-        result = current_val / right_val
-    elif isinstance(op, ast.FloorDiv):
-        result = current_val // right_val
-    elif isinstance(op, ast.Mod):
-        result = current_val % right_val
-    elif isinstance(op, ast.Pow):
-        result = current_val**right_val
-    elif isinstance(op, ast.BitAnd):
-        result = current_val & right_val
-    elif isinstance(op, ast.BitOr):
-        result = current_val | right_val
-    elif isinstance(op, ast.BitXor):
-        result = current_val ^ right_val
-    elif isinstance(op, ast.LShift):
-        result = current_val << right_val
-    elif isinstance(op, ast.RShift):
-        result = current_val >> right_val
-    else:
-        raise Exception("Unsupported augmented operator: " + str(op))
-    await self.assign(node.target, result)
-    return result
-
-async def visit_AnnAssign(self: ASTInterpreter, node: ast.AnnAssign, wrap_exceptions: bool = True) -> None:
-    value = await self.visit(node.value, wrap_exceptions=wrap_exceptions) if node.value else None
-    annotation = await self.visit(node.annotation, wrap_exceptions=True)
-    if isinstance(node.target, ast.Name):
-        self.type_hints[node.target.id] = annotation
-    if value is not None or node.simple:
-        await self.assign(node.target, value)
-
-async def visit_NamedExpr(self: ASTInterpreter, node: ast.NamedExpr, wrap_exceptions: bool = True) -> Any:
-    value = await self.visit(node.value, wrap_exceptions=wrap_exceptions)
-    await self.assign(node.target, value)
-    return value

quantalogic/python_interpreter/base_visitors.py

@@ -1,20 +0,0 @@
-import ast
-from typing import Any
-
-from .exceptions import WrappedException
-from .interpreter_core import ASTInterpreter
-
-async def visit_Module(self: ASTInterpreter, node: ast.Module, wrap_exceptions: bool = True) -> Any:
-    last_value = None
-    for stmt in node.body:
-        last_value = await self.visit(stmt, wrap_exceptions=True)
-    return last_value
-
-async def visit_Expr(self: ASTInterpreter, node: ast.Expr, wrap_exceptions: bool = True) -> Any:
-    return await self.visit(node.value, wrap_exceptions=wrap_exceptions)
-
-async def visit_Pass(self: ASTInterpreter, node: ast.Pass, wrap_exceptions: bool = True) -> None:
-    return None
-
-async def visit_TypeIgnore(self: ASTInterpreter, node: ast.TypeIgnore, wrap_exceptions: bool = True) -> None:
-    pass

quantalogic/python_interpreter/class_visitors.py

@@ -1,22 +0,0 @@
-import ast
-from typing import Any, Dict, List
-
-from .function_utils import Function
-from .interpreter_core import ASTInterpreter
-
-async def visit_ClassDef(self: ASTInterpreter, node: ast.ClassDef, wrap_exceptions: bool = True) -> Any:
-    base_frame = {}
-    self.env_stack.append(base_frame)
-    bases = [await self.visit(base, wrap_exceptions=True) for base in node.bases]
-    try:
-        for stmt in node.body:
-            await self.visit(stmt, wrap_exceptions=True)
-        class_dict = {k: v for k, v in self.env_stack[-1].items() if k not in ["__builtins__"]}
-        cls = type(node.name, tuple(bases), class_dict)
-        for name, value in class_dict.items():
-            if isinstance(value, Function):
-                value.defining_class = cls
-        self.env_stack[-2][node.name] = cls
-        return cls
-    finally:
-        self.env_stack.pop()

quantalogic/python_interpreter/comprehension_visitors.py

@@ -1,172 +0,0 @@
-import ast
-from typing import Any, Dict, List
-
-from .interpreter_core import ASTInterpreter
-from .exceptions import WrappedException
-
-async def visit_ListComp(self: ASTInterpreter, node: ast.ListComp, wrap_exceptions: bool = True) -> List[Any]:
-    result = []
-    base_frame = self.env_stack[-1].copy()
-    self.env_stack.append(base_frame)
-
-    async def rec(gen_idx: int):
-        if gen_idx == len(node.generators):
-            element = await self.visit(node.elt, wrap_exceptions=wrap_exceptions)
-            result.append(element)
-        else:
-            comp = node.generators[gen_idx]
-            iterable = await self.visit(comp.iter, wrap_exceptions=wrap_exceptions)
-            if hasattr(iterable, '__aiter__'):
-                async for item in iterable:
-                    new_frame = self.env_stack[-1].copy()
-                    self.env_stack.append(new_frame)
-                    await self.assign(comp.target, item)
-                    conditions = [await self.visit(if_clause, wrap_exceptions=True) for if_clause in comp.ifs]
-                    if all(conditions):
-                        await rec(gen_idx + 1)
-                    self.env_stack.pop()
-            else:
-                try:
-                    for item in iterable:
-                        new_frame = self.env_stack[-1].copy()
-                        self.env_stack.append(new_frame)
-                        await self.assign(comp.target, item)
-                        conditions = [await self.visit(if_clause, wrap_exceptions=True) for if_clause in comp.ifs]
-                        if all(conditions):
-                            await rec(gen_idx + 1)
-                        self.env_stack.pop()
-                except TypeError as e:
-                    lineno = getattr(node, "lineno", 1)
-                    col = getattr(node, "col_offset", 0)
-                    context_line = self.source_lines[lineno - 1] if self.source_lines and lineno <= len(self.source_lines) else ""
-                    raise WrappedException(f"Object {iterable} is not iterable", e, lineno, col, context_line) from e
-
-    await rec(0)
-    self.env_stack.pop()
-    return result
-
-async def visit_DictComp(self: ASTInterpreter, node: ast.DictComp, wrap_exceptions: bool = True) -> Dict[Any, Any]:
-    result = {}
-    base_frame = self.env_stack[-1].copy()
-    self.env_stack.append(base_frame)
-
-    async def rec(gen_idx: int):
-        if gen_idx == len(node.generators):
-            key = await self.visit(node.key, wrap_exceptions=True)
-            val = await self.visit(node.value, wrap_exceptions=True)
-            result[key] = val
-        else:
-            comp = node.generators[gen_idx]
-            iterable = await self.visit(comp.iter, wrap_exceptions=wrap_exceptions)
-            if hasattr(iterable, '__aiter__'):
-                async for item in iterable:
-                    new_frame = self.env_stack[-1].copy()
-                    self.env_stack.append(new_frame)
-                    await self.assign(comp.target, item)
-                    conditions = [await self.visit(if_clause, wrap_exceptions=True) for if_clause in comp.ifs]
-                    if all(conditions):
-                        await rec(gen_idx + 1)
-                    self.env_stack.pop()
-            else:
-                try:
-                    for item in iterable:
-                        new_frame = self.env_stack[-1].copy()
-                        self.env_stack.append(new_frame)
-                        await self.assign(comp.target, item)
-                        conditions = [await self.visit(if_clause, wrap_exceptions=True) for if_clause in comp.ifs]
-                        if all(conditions):
-                            await rec(gen_idx + 1)
-                        self.env_stack.pop()
-                except TypeError as e:
-                    lineno = getattr(node, "lineno", 1)
-                    col = getattr(node, "col_offset", 0)
-                    context_line = self.source_lines[lineno - 1] if self.source_lines and lineno <= len(self.source_lines) else ""
-                    raise WrappedException(f"Object {iterable} is not iterable", e, lineno, col, context_line) from e
-
-    await rec(0)
-    self.env_stack.pop()
-    return result
-
-async def visit_SetComp(self: ASTInterpreter, node: ast.SetComp, wrap_exceptions: bool = True) -> set:
-    result = set()
-    base_frame = self.env_stack[-1].copy()
-    self.env_stack.append(base_frame)
-
-    async def rec(gen_idx: int):
-        if gen_idx == len(node.generators):
-            result.add(await self.visit(node.elt, wrap_exceptions=True))
-        else:
-            comp = node.generators[gen_idx]
-            iterable = await self.visit(comp.iter, wrap_exceptions=wrap_exceptions)
-            if hasattr(iterable, '__aiter__'):
-                async for item in iterable:
-                    new_frame = self.env_stack[-1].copy()
-                    self.env_stack.append(new_frame)
-                    await self.assign(comp.target, item)
-                    conditions = [await self.visit(if_clause, wrap_exceptions=True) for if_clause in comp.ifs]
-                    if all(conditions):
-                        await rec(gen_idx + 1)
-                    self.env_stack.pop()
-            else:
-                try:
-                    for item in iterable:
-                        new_frame = self.env_stack[-1].copy()
-                        self.env_stack.append(new_frame)
-                        await self.assign(comp.target, item)
-                        conditions = [await self.visit(if_clause, wrap_exceptions=True) for if_clause in comp.ifs]
-                        if all(conditions):
-                            await rec(gen_idx + 1)
-                        self.env_stack.pop()
-                except TypeError as e:
-                    lineno = getattr(node, "lineno", 1)
-                    col = getattr(node, "col_offset", 0)
-                    context_line = self.source_lines[lineno - 1] if self.source_lines and lineno <= len(self.source_lines) else ""
-                    raise WrappedException(f"Object {iterable} is not iterable", e, lineno, col, context_line) from e
-
-    await rec(0)
-    self.env_stack.pop()
-    return result
-
-async def visit_GeneratorExp(self: ASTInterpreter, node: ast.GeneratorExp, wrap_exceptions: bool = True) -> Any:
-    base_frame: Dict[str, Any] = self.env_stack[-1].copy()
-    self.env_stack.append(base_frame)
-
-    async def gen():
-        async def rec(gen_idx: int):
-            if gen_idx == len(node.generators):
-                yield await self.visit(node.elt, wrap_exceptions=True)
-            else:
-                comp = node.generators[gen_idx]
-                iterable = await self.visit(comp.iter, wrap_exceptions=wrap_exceptions)
-                if hasattr(iterable, '__aiter__'):
-                    async for item in iterable:
-                        new_frame = self.env_stack[-1].copy()
-                        self.env_stack.append(new_frame)
-                        await self.assign(comp.target, item)
-                        conditions = [await self.visit(if_clause, wrap_exceptions=True) for if_clause in comp.ifs]
-                        if all(conditions):
-                            async for val in rec(gen_idx + 1):
-                                yield val
-                        self.env_stack.pop()
-                else:
-                    try:
-                        for item in iterable:
-                            new_frame = self.env_stack[-1].copy()
-                            self.env_stack.append(new_frame)
-                            await self.assign(comp.target, item)
-                            conditions = [await self.visit(if_clause, wrap_exceptions=True) for if_clause in comp.ifs]
-                            if all(conditions):
-                                async for val in rec(gen_idx + 1):
-                                    yield val
-                            self.env_stack.pop()
-                    except TypeError as e:
-                        lineno = getattr(node, "lineno", 1)
-                        col = getattr(node, "col_offset", 0)
-                        context_line = self.source_lines[lineno - 1] if self.source_lines and lineno <= len(self.source_lines) else ""
-                        raise WrappedException(f"Object {iterable} is not iterable", e, lineno, col, context_line) from e
-
-        async for val in rec(0):
-            yield val
-
-    self.env_stack.pop()
-    return gen()

quantalogic/python_interpreter/context_visitors.py

@@ -1,59 +0,0 @@
-import ast
-from typing import Any
-
-from .exceptions import ReturnException
-from .interpreter_core import ASTInterpreter
-
-async def visit_With(self: ASTInterpreter, node: ast.With, wrap_exceptions: bool = True) -> Any:
-    result = None
-    contexts = []
-    for item in node.items:
-        ctx = await self.visit(item.context_expr, wrap_exceptions=wrap_exceptions)
-        val = ctx.__enter__()
-        contexts.append((ctx, val))
-        if item.optional_vars:
-            await self.assign(item.optional_vars, val)
-    try:
-        for stmt in node.body:
-            result = await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-    except ReturnException as ret:
-        for ctx, _ in reversed(contexts):
-            ctx.__exit__(None, None, None)
-        raise ret
-    except Exception as e:
-        exc_type, exc_value, tb = type(e), e, e.__traceback__
-        for ctx, _ in reversed(contexts):
-            if not ctx.__exit__(exc_type, exc_value, tb):
-                raise
-        raise
-    else:
-        for ctx, _ in reversed(contexts):
-            ctx.__exit__(None, None, None)
-    return result
-
-async def visit_AsyncWith(self: ASTInterpreter, node: ast.AsyncWith, wrap_exceptions: bool = True) -> Any:
-    result = None
-    contexts = []
-    for item in node.items:
-        ctx = await self.visit(item.context_expr, wrap_exceptions=wrap_exceptions)
-        val = await ctx.__aenter__()
-        contexts.append((ctx, val))
-        if item.optional_vars:
-            await self.assign(item.optional_vars, val)
-    try:
-        for stmt in node.body:
-            result = await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-    except ReturnException as ret:
-        for ctx, _ in reversed(contexts):
-            await ctx.__aexit__(None, None, None)
-        raise ret
-    except Exception as e:
-        exc_type, exc_value, tb = type(e), e, e.__traceback__
-        for ctx, _ in reversed(contexts):
-            if not await ctx.__aexit__(exc_type, exc_value, tb):
-                raise
-        raise
-    else:
-        for ctx, _ in reversed(contexts):
-            await ctx.__aexit__(None, None, None)
-    return result

quantalogic/python_interpreter/control_flow_visitors.py

@@ -1,88 +0,0 @@
-import ast
-from typing import Any
-
-from .exceptions import BreakException, ContinueException, ReturnException
-from .interpreter_core import ASTInterpreter
-
-async def visit_If(self: ASTInterpreter, node: ast.If, wrap_exceptions: bool = True) -> Any:
-    if await self.visit(node.test, wrap_exceptions=wrap_exceptions):
-        branch = node.body
-    else:
-        branch = node.orelse
-    result = None
-    if branch:
-        for stmt in branch[:-1]:
-            await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-        result = await self.visit(branch[-1], wrap_exceptions=wrap_exceptions)
-    return result
-
-async def visit_While(self: ASTInterpreter, node: ast.While, wrap_exceptions: bool = True) -> None:
-    while await self.visit(node.test, wrap_exceptions=wrap_exceptions):
-        try:
-            for stmt in node.body:
-                await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-        except BreakException:
-            break
-        except ContinueException:
-            continue
-    for stmt in node.orelse:
-        await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-
-async def visit_For(self: ASTInterpreter, node: ast.For, wrap_exceptions: bool = True) -> None:
-    iter_obj: Any = await self.visit(node.iter, wrap_exceptions=wrap_exceptions)
-    broke = False
-    if hasattr(iter_obj, '__aiter__'):
-        async for item in iter_obj:
-            await self.assign(node.target, item)
-            try:
-                for stmt in node.body:
-                    await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-            except BreakException:
-                broke = True
-                break
-            except ContinueException:
-                continue
-    else:
-        for item in iter_obj:
-            await self.assign(node.target, item)
-            try:
-                for stmt in node.body:
-                    await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-            except BreakException:
-                broke = True
-                break
-            except ContinueException:
-                continue
-    if not broke:
-        for stmt in node.orelse:
-            await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-
-async def visit_AsyncFor(self: ASTInterpreter, node: ast.AsyncFor, wrap_exceptions: bool = True) -> None:
-    iterable = await self.visit(node.iter, wrap_exceptions=wrap_exceptions)
-    broke = False
-    async for value in iterable:
-        await self.assign(node.target, value)
-        try:
-            for stmt in node.body:
-                await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-        except BreakException:
-            broke = True
-            break
-        except ContinueException:
-            continue
-    if not broke:
-        for stmt in node.orelse:
-            await self.visit(stmt, wrap_exceptions=wrap_exceptions)
-
-async def visit_Break(self: ASTInterpreter, node: ast.Break, wrap_exceptions: bool = True) -> None:
-    raise BreakException()
-
-async def visit_Continue(self: ASTInterpreter, node: ast.Continue, wrap_exceptions: bool = True) -> None:
-    raise ContinueException()
-
-async def visit_Return(self: ASTInterpreter, node: ast.Return, wrap_exceptions: bool = True) -> None:
-    value: Any = await self.visit(node.value, wrap_exceptions=wrap_exceptions) if node.value is not None else None
-    raise ReturnException(value)
-
-async def visit_IfExp(self: ASTInterpreter, node: ast.IfExp, wrap_exceptions: bool = True) -> Any:
-    return await self.visit(node.body, wrap_exceptions=wrap_exceptions) if await self.visit(node.test, wrap_exceptions=wrap_exceptions) else await self.visit(node.orelse, wrap_exceptions=wrap_exceptions)

quantalogic/python_interpreter/exception_visitors.py

@@ -1,109 +0,0 @@
-import ast
-from typing import Any, Optional, Tuple
-
-from .exceptions import BaseExceptionGroup, ReturnException, WrappedException
-from .interpreter_core import ASTInterpreter
-
-async def visit_Try(self: ASTInterpreter, node: ast.Try, wrap_exceptions: bool = True) -> Any:
-    result: Any = None
-    try:
-        for stmt in node.body:
-            result = await self.visit(stmt, wrap_exceptions=False)
-    except ReturnException as ret:
-        raise ret
-    except Exception as e:
-        original_e = e.original_exception if isinstance(e, WrappedException) else e
-        for handler in node.handlers:
-            exc_type = await self._resolve_exception_type(handler.type)
-            if exc_type and isinstance(original_e, exc_type):
-                if handler.name:
-                    self.set_variable(handler.name, original_e)
-                handler_result = None
-                try:
-                    for stmt in handler.body:
-                        handler_result = await self.visit(stmt, wrap_exceptions=True)
-                except ReturnException as ret:
-                    raise ret
-                if handler_result is not None:
-                    result = handler_result
-                break
-        else:
-            raise
-    else:
-        for stmt in node.orelse:
-            result = await self.visit(stmt, wrap_exceptions=True)
-    finally:
-        for stmt in node.finalbody:
-            await self.visit(stmt, wrap_exceptions=True)
-    return result
-
-async def visit_TryStar(self: ASTInterpreter, node: ast.TryStar, wrap_exceptions: bool = True) -> Any:
-    result: Any = None
-    exc_info: Optional[Tuple] = None
-
-    try:
-        for stmt in node.body:
-            result = await self.visit(stmt, wrap_exceptions=False)
-    except BaseException as e:
-        exc_info = (type(e), e, e.__traceback__)
-        handled = False
-        if isinstance(e, BaseExceptionGroup):
-            remaining_exceptions = []
-            for handler in node.handlers:
-                if handler.type is None:
-                    exc_type = BaseException
-                elif isinstance(handler.type, ast.Name):
-                    exc_type = self.get_variable(handler.type.id)
-                else:
-                    exc_type = await self.visit(handler.type, wrap_exceptions=True)
-                matching_exceptions = [ex for ex in e.exceptions if isinstance(ex, exc_type)]
-                if matching_exceptions:
-                    if handler.name:
-                        self.set_variable(handler.name, BaseExceptionGroup("", matching_exceptions))
-                    for stmt in handler.body:
-                        result = await self.visit(stmt, wrap_exceptions=True)
-                    handled = True
-                remaining_exceptions.extend([ex for ex in e.exceptions if not isinstance(ex, exc_type)])
-            if remaining_exceptions and not handled:
-                raise BaseExceptionGroup("Uncaught exceptions", remaining_exceptions)
-            if handled:
-                exc_info = None
-        else:
-            for handler in node.handlers:
-                if handler.type is None:
-                    exc_type = BaseException
-                elif isinstance(handler.type, ast.Name):
-                    exc_type = self.get_variable(handler.type.id)
-                else:
-                    exc_type = await self.visit(handler.type, wrap_exceptions=True)
-                if exc_info and issubclass(exc_info[0], exc_type):
-                    if handler.name:
-                        self.set_variable(handler.name, exc_info[1])
-                    for stmt in handler.body:
-                        result = await self.visit(stmt, wrap_exceptions=True)
-                    exc_info = None
-                    handled = True
-                    break
-        if exc_info and not handled:
-            raise exc_info[1]
-    else:
-        for stmt in node.orelse:
-            result = await self.visit(stmt, wrap_exceptions=True)
-    finally:
-        for stmt in node.finalbody:
-            try:
-                await self.visit(stmt, wrap_exceptions=True)
-            except ReturnException:
-                raise
-            except Exception:
-                if exc_info:
-                    raise exc_info[1]
-                raise
-
-    return result
-
-async def visit_Raise(self: ASTInterpreter, node: ast.Raise, wrap_exceptions: bool = True) -> None:
-    exc = await self.visit(node.exc, wrap_exceptions=wrap_exceptions) if node.exc else None
-    if exc:
-        raise exc
-    raise Exception("Raise with no exception specified")