github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 00:48:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Add support for custom runtimes to Riza tools (#30664)

Browse Source 
**Description:**
Adds support for Riza custom runtimes to the two Riza code interpreter
tools, allowing users to run LLM-generated code that depends on
libraries outside stdlib.
**Issue:** N/A
**Dependencies:** None
**Twitter handle:** @rizaio
This commit is contained in:
Andrew Benton 2025-04-04 08:03:14 -07:00 committed by GitHub
parent aa37893c00
commit 4e7a9a7014
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
libs/community/langchain_community/tools/riza/command.py
View File

@ -66,8 +66,11 @@ class ExecPython(BaseTool): # type: ignore[override, override]
handle_tool_error: bool = True handle_tool_error: bool = True
client: Any = None client: Any = None
runtime_revision_id: Optional[str] = None
def __init__(self, **kwargs: Any) -> None: def __init__(
self, runtime_revision_id: Optional[str] = None, **kwargs: Any
) -> None:
try: try:
from rizaio import Riza from rizaio import Riza
except ImportError as e: except ImportError as e:
@ -77,11 +80,14 @@ class ExecPython(BaseTool): # type: ignore[override, override]
) from e ) from e
super().__init__(**kwargs) super().__init__(**kwargs)
self.client = Riza() self.client = Riza()
self.runtime_revision_id = runtime_revision_id
def _run( def _run(
self, code: str, run_manager: Optional[CallbackManagerForToolRun] = None self, code: str, run_manager: Optional[CallbackManagerForToolRun] = None
) -> str: ) -> str:
output = self.client.command.exec(language="PYTHON", code=code) output = self.client.command.exec(
runtime_revision_id=self.runtime_revision_id, language="python", code=code
)
if output.exit_code > 0: if output.exit_code > 0:
raise ToolException( raise ToolException(
f"Riza code execution returned a non-zero exit code. " f"Riza code execution returned a non-zero exit code. "
@ -107,8 +113,11 @@ class ExecJavaScript(BaseTool): # type: ignore[override, override]
handle_tool_error: bool = True handle_tool_error: bool = True
client: Any = None client: Any = None
runtime_revision_id: Optional[str] = None
def __init__(self, **kwargs: Any) -> None: def __init__(
self, runtime_revision_id: Optional[str] = None, **kwargs: Any
) -> None:
try: try:
from rizaio import Riza from rizaio import Riza
except ImportError as e: except ImportError as e:
@ -118,11 +127,16 @@ class ExecJavaScript(BaseTool): # type: ignore[override, override]
) from e ) from e
super().__init__(**kwargs) super().__init__(**kwargs)
self.client = Riza() self.client = Riza()
self.runtime_revision_id = runtime_revision_id
def _run( def _run(
self, code: str, run_manager: Optional[CallbackManagerForToolRun] = None self, code: str, run_manager: Optional[CallbackManagerForToolRun] = None
) -> str: ) -> str:
output = self.client.command.exec(language="JAVASCRIPT", code=code) output = self.client.command.exec(
runtime_revision_id=self.runtime_revision_id,
language="javascript",
code=code,
)
if output.exit_code > 0: if output.exit_code > 0:
raise ToolException( raise ToolException(
f"Riza code execution returned a non-zero exit code. " f"Riza code execution returned a non-zero exit code. "