sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 23:48:56 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 16

Set python interpret sandbox url via environment variable

Browse source
This commit is contained in:
Debanjum Singh Solanky 2024-10-10 00:28:51 -07:00
parent 2dc5804571
commit 9e7025b330
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/khoj/processor/tools/run_code.py
View file

@ -2,6 +2,7 @@ import asyncio
import datetime
import json
import logging
import os
from typing import Any, Callable, List, Optional
import aiohttp
@ -21,6 +22,9 @@ from khoj.utils.rawconfig import LocationData
logger = logging.getLogger(__name__)
SANDBOX_URL = os.getenv("KHOJ_TERRARIUM_URL", "http://localhost:8080")
async def run_code(
query: str,
conversation_history: dict,
@ -29,7 +33,7 @@ async def run_code(
send_status_func: Optional[Callable] = None,
uploaded_image_url: str = None,
agent: Agent = None,
sandbox_url: str = "http://localhost:8080",
sandbox_url: str = SANDBOX_URL,
):
# Generate Code
if send_status_func:
@ -104,7 +108,7 @@ async def generate_python_code(
return codes
async def execute_sandboxed_python(code: str, sandbox_url: str = "http://localhost:8080") -> dict[str, Any]:
async def execute_sandboxed_python(code: str, sandbox_url: str = SANDBOX_URL) -> dict[str, Any]:
"""
Takes code to run as a string and calls the terrarium API to execute it.
Returns the result of the code execution as a dictionary.