flyteplugins-codegen 2.3.6__py3-none-any.whl → 2.3.7__py3-none-any.whl

flyteplugins/codegen/auto_coder_agent.py

@@ -73,7 +73,6 @@ class AutoCoderAgent:
         cache: CacheRequest for sandboxes: "auto", "override", or "disable". Defaults to "auto".
         backend: Execution backend: "litellm" (default) or "claude".
         agent_max_turns: Maximum agent turns when backend="claude". Defaults to 50.
-        block_network: Block all outbound network access in sandboxes. Defaults to False.
 
     Example::
 
@@ -120,7 +119,6 @@ class AutoCoderAgent:
     cache: str = "auto"
     backend: Literal["litellm", "claude"] = "litellm"
     agent_max_turns: int = 50
-    block_network: bool = False
 
     @syncify
     async def generate(
@@ -274,7 +272,6 @@ class AutoCoderAgent:
                 env_vars=self.env_vars,
                 secrets=self.secrets,
                 cache=self.cache,
-                block_network=self.block_network,
                 max_turns=self.agent_max_turns,
             )
 
@@ -390,7 +387,6 @@ class _CodeGenSession:
         self.env_vars = agent.env_vars
         self.secrets = agent.secrets
         self.cache = agent.cache
-        self.block_network = agent.block_network
         self.image_config = agent.image_config
         self.litellm_params = agent.litellm_params
 
@@ -611,7 +607,6 @@ class _CodeGenSession:
             env_vars=self.env_vars,
             secrets=self.secrets,
             cache=self.cache,
-            block_network=self.block_network,
             _attempt=attempt,
         )
 

flyteplugins/codegen/execution/agent.py

@@ -77,7 +77,6 @@ async def code_gen_eval_agent(
     env_vars: Optional[dict[str, str]] = None,
     secrets: Optional[list] = None,
     cache: str = "auto",
-    block_network: bool = False,
     max_turns: int = 50,
     language: str = "python",
 ) -> CodeGenEvalResult:
@@ -105,7 +104,6 @@ async def code_gen_eval_agent(
         env_vars: Optional environment variables to set in the sandbox
         secrets: Optional secrets to make available in the sandbox
         cache: Caching behavior for sandbox execution ("auto", "override", "disable")
-        block_network: Block all outbound network access in sandboxes. Defaults to False.
         max_turns: Maximum number of agent turns before stopping (default: 50)
         language: Programming language for code generation (default: "python")
 
@@ -191,7 +189,6 @@ async def code_gen_eval_agent(
             env_vars=env_vars,
             secrets=secrets,
             cache=cache,
-            block_network=block_network,
             _attempt=_exec_state["test_count"],
         )
 

flyteplugins/codegen/execution/docker.py

@@ -140,7 +140,6 @@ async def run_tests(
     env_vars: Optional[dict[str, str]] = None,
     secrets: Optional[list] = None,
     cache: str = "auto",
-    block_network: bool = False,
     _attempt: int = 1,
 ) -> RunResult:
     """Run pytest tests against code in an isolated container.
@@ -156,7 +155,6 @@ async def run_tests(
         env_vars: Environment variables available inside the container.
         secrets: Flyte secrets to mount.
         cache: Cache behaviour — `"auto"`, `"override"`, or `"disable"`.
-        block_network: Block all outbound network access. Defaults to False.
         _attempt: Differentiates repeated calls with the same base name.
 
     Returns:
@@ -182,7 +180,6 @@ async def run_tests(
         env_vars=env_vars,
         secrets=secrets,
         cache=cache,
-        block_network=block_network,
     )
 
     try:

{flyteplugins_codegen-2.3.6.dist-info → flyteplugins_codegen-2.3.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flyteplugins-codegen
-Version: 2.3.6
+Version: 2.3.7
 Summary: LLM-powered code generation and evaluation plugin for Flyte
 Author-email: Samhita Alla <samhita@union.ai>
 Requires-Python: >=3.10
@@ -178,7 +178,6 @@ result = await agent.generate.aio(prompt="...")
 | `cache`               | `str`             | `"auto"`       | CacheRequest for sandboxes: `"auto"`, `"override"`, or `"disable"` |
 | `backend`             | `str`             | `"litellm"`    | Execution backend: `"litellm"` or `"claude"`               |
 | `agent_max_turns`     | `int`             | `50`           | Max turns when `backend="claude"`                          |
-| `block_network`       | `bool`            | `False`        | Block all outbound network access in sandboxes. Set to `True` to block network access. |
 
 **`generate()` parameters (per-call):**
 

{flyteplugins_codegen-2.3.6.dist-info → flyteplugins_codegen-2.3.7.dist-info}/RECORD

@@ -1,17 +1,17 @@
 flyteplugins/codegen/__init__.py,sha256=jumoM0Po0Tx1KqatcdNi_Mk1MYCLcccnyjn6d1zurLQ,366
-flyteplugins/codegen/auto_coder_agent.py,sha256=G6F4OCJXKU-6Y16hNdodu8FtOVrOsdt2hI-lNHF40Ac,46282
+flyteplugins/codegen/auto_coder_agent.py,sha256=8i7vVqnuoS0r9jOsoZlMh5Xyv0hqTf8IXa3_aMxXu3M,46015
 flyteplugins/codegen/core/__init__.py,sha256=YjAN0PpvkhERFmlkEq78O2q92bmgyGAVttYXeDH_hyc,355
 flyteplugins/codegen/core/types.py,sha256=kxAoKMu9tOTRNPBfiFC-dxMQOmeEawJAVRNKj42_3js,12274
 flyteplugins/codegen/data/__init__.py,sha256=UDkKuvyBEZmsPaTR6yEexcQjLutMRxuV5-pWflGc_sI,670
 flyteplugins/codegen/data/extraction.py,sha256=3jjm4REJYzl3U8FWv1Cys4FOuh2gG1Hv90OpdeiBlgs,10273
 flyteplugins/codegen/data/schema.py,sha256=fAcT0upjXRFIDbLUolFiuPAJHWbVu5VmmBxWEW_-9uc,10300
 flyteplugins/codegen/execution/__init__.py,sha256=3i4bP_xNp8ZgsolRI7ccUtkIW-h-1zCXFrjWfltVu3U,192
-flyteplugins/codegen/execution/agent.py,sha256=eCIsQ09YU3OYRci1dZb8lRqxTPr21DHvW3LLPLfIKc4,27334
-flyteplugins/codegen/execution/docker.py,sha256=LSmg9g7MCiD-yGAMuQT0YCh_npma_dellSh0WIvJHak,7294
+flyteplugins/codegen/execution/agent.py,sha256=k6H54H-W5DFrDvg6ezCw4oKaUM7mN8g7E38QWGFLFeY,27170
+flyteplugins/codegen/execution/docker.py,sha256=vZtXhf4v5A0J2ZWPNT3aQIB5OzOPDEGfwGP7R7lJapE,7147
 flyteplugins/codegen/generation/__init__.py,sha256=GkLiXfJeVQmLlHf4R08qzgUa5wYG97KseGjRtLCFZhU,1065
 flyteplugins/codegen/generation/llm.py,sha256=FPBt1MlW_OjwtEEcnJhE62sYrbSK4tRB3UEk80cWvgQ,44229
 flyteplugins/codegen/generation/prompts.py,sha256=-Qt8M_6bkS4Jwm-bCQnCGR-oKL_mWhryCc9_OUnvXPs,6344
-flyteplugins_codegen-2.3.6.dist-info/METADATA,sha256=K571F-P8qBk9vBIZx4KOoI86Lcu_BwKaHTMNpHhSEyg,19246
-flyteplugins_codegen-2.3.6.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-flyteplugins_codegen-2.3.6.dist-info/top_level.txt,sha256=cgd779rPu9EsvdtuYgUxNHHgElaQvPn74KhB5XSeMBE,13
-flyteplugins_codegen-2.3.6.dist-info/RECORD,,
+flyteplugins_codegen-2.3.7.dist-info/METADATA,sha256=QW_daZomBExcGi-3f0yIwKlvuPhHmRVqYO1DH_tnGyc,19094
+flyteplugins_codegen-2.3.7.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+flyteplugins_codegen-2.3.7.dist-info/top_level.txt,sha256=cgd779rPu9EsvdtuYgUxNHHgElaQvPn74KhB5XSeMBE,13
+flyteplugins_codegen-2.3.7.dist-info/RECORD,,