phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-06 10:42:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update client-sdk types

Browse source
This commit is contained in:
Ashwin Bharambe 2025-01-22 22:08:52 -08:00
parent 83b6921302
commit d9f96cee12
3 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
docs/notebooks/Llama_Stack_Building_AI_Applications.ipynb
View file

@ -2088,7 +2088,7 @@
"from llama_stack_client.lib.agents.event_logger import EventLogger\n", "from llama_stack_client.lib.agents.event_logger import EventLogger\n",
"from llama_stack_client.types.agent_create_params import AgentConfig\n", "from llama_stack_client.types.agent_create_params import AgentConfig\n",
"from termcolor import cprint\n", "from termcolor import cprint\n",
"from llama_stack_client.types.tool_runtime import DocumentParam as Document\n", "from llama_stack_client.types import Document\n",
"\n", "\n",
"urls = [\"chat.rst\", \"llama3.rst\", \"datasets.rst\", \"lora_finetune.rst\"]\n", "urls = [\"chat.rst\", \"llama3.rst\", \"datasets.rst\", \"lora_finetune.rst\"]\n",
"documents = [\n", "documents = [\n",

2
llama_stack/distribution/store/registry.py
View file

@ -35,7 +35,7 @@ class DistributionRegistry(Protocol):
REGISTER_PREFIX = "distributions:registry" REGISTER_PREFIX = "distributions:registry"
KEY_VERSION = "v6" KEY_VERSION = "v7"
KEY_FORMAT = f"{REGISTER_PREFIX}:{KEY_VERSION}::" + "{type}:{identifier}" KEY_FORMAT = f"{REGISTER_PREFIX}:{KEY_VERSION}::" + "{type}:{identifier}"

12
tests/client-sdk/tool_runtime/test_rag_tool.py
View file

@ -8,7 +8,7 @@ import random
import pytest import pytest
from llama_stack_client.types.tool_runtime import DocumentParam from llama_stack_client.types import Document
@pytest.fixture(scope="function") @pytest.fixture(scope="function")
@ -38,22 +38,22 @@ def single_entry_vector_db_registry(llama_stack_client, empty_vector_db_registry
@pytest.fixture(scope="session") @pytest.fixture(scope="session")
def sample_documents(): def sample_documents():
return [ return [
DocumentParam( Document(
document_id="test-doc-1", document_id="test-doc-1",
content="Python is a high-level programming language.", content="Python is a high-level programming language.",
metadata={"category": "programming", "difficulty": "beginner"}, metadata={"category": "programming", "difficulty": "beginner"},
), ),
DocumentParam( Document(
document_id="test-doc-2", document_id="test-doc-2",
content="Machine learning is a subset of artificial intelligence.", content="Machine learning is a subset of artificial intelligence.",
metadata={"category": "AI", "difficulty": "advanced"}, metadata={"category": "AI", "difficulty": "advanced"},
), ),
DocumentParam( Document(
document_id="test-doc-3", document_id="test-doc-3",
content="Data structures are fundamental to computer science.", content="Data structures are fundamental to computer science.",
metadata={"category": "computer science", "difficulty": "intermediate"}, metadata={"category": "computer science", "difficulty": "intermediate"},
), ),
DocumentParam( Document(
document_id="test-doc-4", document_id="test-doc-4",
content="Neural networks are inspired by biological neural networks.", content="Neural networks are inspired by biological neural networks.",
metadata={"category": "AI", "difficulty": "advanced"}, metadata={"category": "AI", "difficulty": "advanced"},
@ -148,7 +148,7 @@ def test_vector_db_insert_from_url_and_query(
"llama3.rst", "llama3.rst",
] ]
documents = [ documents = [
DocumentParam( Document(
document_id=f"num-{i}", document_id=f"num-{i}",
content=f"https://raw.githubusercontent.com/pytorch/torchtune/main/docs/source/tutorials/{url}", content=f"https://raw.githubusercontent.com/pytorch/torchtune/main/docs/source/tutorials/{url}",
mime_type="text/plain", mime_type="text/plain",