flyteplugins-codegen 2.1.7__py3-none-any.whl → 2.1.8__py3-none-any.whl

flyteplugins/codegen/auto_coder_agent.py

@@ -73,6 +73,7 @@ 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::
 
@@ -119,6 +120,7 @@ class AutoCoderAgent:
     cache: str = "auto"
     backend: Literal["litellm", "claude"] = "litellm"
     agent_max_turns: int = 50
+    block_network: bool = False
 
     @syncify
     async def generate(
@@ -271,6 +273,7 @@ class AutoCoderAgent:
                 env_vars=self.env_vars,
                 secrets=self.secrets,
                 cache=self.cache,
+                block_network=self.block_network,
                 max_turns=self.agent_max_turns,
             )
 
@@ -386,6 +389,7 @@ 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
 
@@ -606,6 +610,7 @@ 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,6 +77,7 @@ 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:
@@ -104,6 +105,7 @@ 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")
 
@@ -189,6 +191,7 @@ 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,6 +140,7 @@ 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.
@@ -155,6 +156,7 @@ 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:
@@ -180,6 +182,7 @@ async def run_tests(
         env_vars=env_vars,
         secrets=secrets,
         cache=cache,
+        block_network=block_network,
     )
 
     try:

{flyteplugins_codegen-2.1.7.dist-info → flyteplugins_codegen-2.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flyteplugins-codegen
-Version: 2.1.7
+Version: 2.1.8
 Summary: LLM-powered code generation and evaluation plugin for Flyte
 Author-email: Samhita Alla <samhita@union.ai>
 Requires-Python: >=3.10
@@ -178,6 +178,7 @@ 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.1.7.dist-info → flyteplugins_codegen-2.1.8.dist-info}/RECORD

@@ -1,17 +1,17 @@
 flyteplugins/codegen/__init__.py,sha256=jumoM0Po0Tx1KqatcdNi_Mk1MYCLcccnyjn6d1zurLQ,366
-flyteplugins/codegen/auto_coder_agent.py,sha256=23KylEpVBVqj2mt0shyuScpu3xwpdSsNFnmV_CPwb2Q,45974
+flyteplugins/codegen/auto_coder_agent.py,sha256=-HkI3nM3T0Ydj8ezDprDI8DLetgQeoixBDGJXnJfmUA,46241
 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=k6H54H-W5DFrDvg6ezCw4oKaUM7mN8g7E38QWGFLFeY,27170
-flyteplugins/codegen/execution/docker.py,sha256=vZtXhf4v5A0J2ZWPNT3aQIB5OzOPDEGfwGP7R7lJapE,7147
+flyteplugins/codegen/execution/agent.py,sha256=eCIsQ09YU3OYRci1dZb8lRqxTPr21DHvW3LLPLfIKc4,27334
+flyteplugins/codegen/execution/docker.py,sha256=LSmg9g7MCiD-yGAMuQT0YCh_npma_dellSh0WIvJHak,7294
 flyteplugins/codegen/generation/__init__.py,sha256=GkLiXfJeVQmLlHf4R08qzgUa5wYG97KseGjRtLCFZhU,1065
 flyteplugins/codegen/generation/llm.py,sha256=FPBt1MlW_OjwtEEcnJhE62sYrbSK4tRB3UEk80cWvgQ,44229
 flyteplugins/codegen/generation/prompts.py,sha256=Te9nK9TldLyHM2KB24PhTiWNqoEXJlW9l8nt09FbWa8,5770
-flyteplugins_codegen-2.1.7.dist-info/METADATA,sha256=Osq7EnaIJ-m02fA8ylvLOyBHJFzlv2jB08jNk6Arjjo,19051
-flyteplugins_codegen-2.1.7.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-flyteplugins_codegen-2.1.7.dist-info/top_level.txt,sha256=cgd779rPu9EsvdtuYgUxNHHgElaQvPn74KhB5XSeMBE,13
-flyteplugins_codegen-2.1.7.dist-info/RECORD,,
+flyteplugins_codegen-2.1.8.dist-info/METADATA,sha256=VUaHhiLQdeQxOOj6dK_e6SdWXYsnxN9NnBhYLbNsZTY,19203
+flyteplugins_codegen-2.1.8.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+flyteplugins_codegen-2.1.8.dist-info/top_level.txt,sha256=cgd779rPu9EsvdtuYgUxNHHgElaQvPn74KhB5XSeMBE,13
+flyteplugins_codegen-2.1.8.dist-info/RECORD,,