phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'tgi-integration' of github.com:hanouticelina/llama-stack into tgi-integration

Browse source
This commit is contained in:
Celina Hanouti 2024-09-06 17:58:22 +02:00
commit 5ab4fd31f7
4 changed files with 6 additions and 5 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 logging
import os import tempfile
import uuid import uuid
from typing import AsyncGenerator from typing import AsyncGenerator
@ -71,7 +71,7 @@ class MetaReferenceAgenticSystemImpl(AgenticSystem):
tool = CodeInterpreterTool() tool = CodeInterpreterTool()
elif isinstance(tool_defn, PhotogenToolDefinition): elif isinstance(tool_defn, PhotogenToolDefinition):
tool = PhotogenTool( tool = PhotogenTool(
dump_dir="/tmp/photogen_dump_" + os.environ["USER"], dump_dir=tempfile.mkdtemp(),
) )
else: else:
continue continue

1
llama_toolchain/core/build_container.sh
View file

@ -48,6 +48,7 @@ RUN apt-get update && apt-get install -y \
curl wget telnet \ curl wget telnet \
procps psmisc lsof \ procps psmisc lsof \
traceroute \ traceroute \
bubblewrap \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
EOF EOF

4
llama_toolchain/tools/builtin.py
View file

@ -5,8 +5,8 @@
# the root directory of this source tree. # the root directory of this source tree.
import json import json
import os
import re import re
import tempfile
from abc import abstractmethod from abc import abstractmethod
from typing import List, Optional from typing import List, Optional
@ -284,7 +284,7 @@ class WolframAlphaTool(SingleMessageBuiltinTool):
class CodeInterpreterTool(BaseTool): class CodeInterpreterTool(BaseTool):
def __init__(self) -> None: def __init__(self) -> None:
ctx = CodeExecutionContext( ctx = CodeExecutionContext(
matplotlib_dump_dir=f"/tmp/{os.environ['USER']}_matplotlib_dump", matplotlib_dump_dir=tempfile.mkdtemp(),
) )
self.code_executor = CodeExecutor(ctx) self.code_executor = CodeExecutor(ctx)

2
setup.py
View file

@ -16,7 +16,7 @@ def read_requirements():
setup( setup(
name="llama_toolchain", name="llama_toolchain",
version="0.0.12", version="0.0.13",
author="Meta Llama", author="Meta Llama",
author_email="llama-oss@meta.com", author_email="llama-oss@meta.com",
description="Llama toolchain", description="Llama toolchain",