From 4b85b8ae51ae0f43bcc9b49c4511b591f239b0ae Mon Sep 17 00:00:00 2001 From: Ihar Hrachyshka Date: Tue, 4 Feb 2025 18:32:38 -0500 Subject: [PATCH] [docs] Make RAG example self-contained Before the patch, the example could not be executed verbatim without copy-pasting client function from the inference example. I think it's better to have examples self-contained, especially in a getting started guide. Signed-off-by: Ihar Hrachyshka --- docs/source/getting_started/index.md | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/docs/source/getting_started/index.md b/docs/source/getting_started/index.md index 00301d6e7..66e7dddbc 100644 --- a/docs/source/getting_started/index.md +++ b/docs/source/getting_started/index.md @@ -150,6 +150,23 @@ from llama_stack_client.lib.agents.event_logger import EventLogger from llama_stack_client.types.agent_create_params import AgentConfig from llama_stack_client.types import Document + +def create_http_client(): + from llama_stack_client import LlamaStackClient + + return LlamaStackClient( + base_url=f"http://localhost:{os.environ['LLAMA_STACK_PORT']}" + ) + + +def create_library_client(template="ollama"): + from llama_stack import LlamaStackAsLibraryClient + + client = LlamaStackAsLibraryClient(template) + client.initialize() + return client + + client = ( create_library_client() ) # or create_http_client() depending on the environment you picked