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

test client:

Browse source
This commit is contained in:
Xi Yan 2024-10-23 13:55:55 -07:00
parent bb43369521
commit 51d5ad67c4
1 changed files with 1 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

18
llama_stack/apis/datasets/client.py
View file

@ -5,9 +5,7 @@
# the root directory of this source tree.
import asyncio
import base64
import json
import mimetypes
import os
from pathlib import Path
from typing import Optional
@ -19,21 +17,7 @@ from termcolor import cprint
from .datasets import * # noqa: F403
from llama_stack.apis.datasets import * # noqa: F403
from llama_stack.apis.common.type_system import * # noqa: F403
def data_url_from_file(file_path: str) -> str:
if not os.path.exists(file_path):
raise FileNotFoundError(f"File not found: {file_path}")
with open(file_path, "rb") as file:
file_content = file.read()
base64_content = base64.b64encode(file_content).decode("utf-8")
mime_type, _ = mimetypes.guess_type(file_path)
data_url = f"data:{mime_type};base64,{base64_content}"
return data_url
from llama_stack.providers.tests.datasetio.test_datasetio import data_url_from_file
class DatasetsClient(Datasets):