phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Remove dependence on os.environ["USER"]

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-05 15:37:12 -07:00
parent 6c69e09c6a
commit f6b5e394ab
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_toolchain/agentic_system/meta_reference/agentic_system.py
View file

@ -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

4
llama_toolchain/tools/builtin.py
View file

@ -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)