From f6b5e394ab9d61a0928ac477708b7bcdde499868 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Thu, 5 Sep 2024 15:37:12 -0700 Subject: [PATCH] Remove dependence on os.environ["USER"] --- .../agentic_system/meta_reference/agentic_system.py | 4 ++-- llama_toolchain/tools/builtin.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/llama_toolchain/agentic_system/meta_reference/agentic_system.py b/llama_toolchain/agentic_system/meta_reference/agentic_system.py index 4fa2aa584..09fbfdde5 100644 --- a/llama_toolchain/agentic_system/meta_reference/agentic_system.py +++ b/llama_toolchain/agentic_system/meta_reference/agentic_system.py @@ -6,7 +6,7 @@ import logging -import os +import tempfile import uuid from typing import AsyncGenerator @@ -71,7 +71,7 @@ class MetaReferenceAgenticSystemImpl(AgenticSystem): tool = CodeInterpreterTool() elif isinstance(tool_defn, PhotogenToolDefinition): tool = PhotogenTool( - dump_dir="/tmp/photogen_dump_" + os.environ["USER"], + dump_dir=tempfile.mkdtemp(), ) else: continue diff --git a/llama_toolchain/tools/builtin.py b/llama_toolchain/tools/builtin.py index f2ddeefa7..3a53e2e26 100644 --- a/llama_toolchain/tools/builtin.py +++ b/llama_toolchain/tools/builtin.py @@ -5,8 +5,8 @@ # the root directory of this source tree. import json -import os import re +import tempfile from abc import abstractmethod from typing import List, Optional @@ -284,7 +284,7 @@ class WolframAlphaTool(SingleMessageBuiltinTool): class CodeInterpreterTool(BaseTool): def __init__(self) -> None: ctx = CodeExecutionContext( - matplotlib_dump_dir=f"/tmp/{os.environ['USER']}_matplotlib_dump", + matplotlib_dump_dir=tempfile.mkdtemp(), ) self.code_executor = CodeExecutor(ctx)