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

update inference prompt msg

Browse source
This commit is contained in:
Xi Yan 2024-09-19 12:01:46 -07:00
parent 9eb01dd664
commit a30b919ae1
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/apis/inference/client.py
View file

@ -86,7 +86,9 @@ class InferenceClient(Inference):
async def run_main(host: str, port: int, stream: bool): async def run_main(host: str, port: int, stream: bool):
client = InferenceClient(f"http://{host}:{port}") client = InferenceClient(f"http://{host}:{port}")
message = UserMessage(content="hello world, troll me in two-paragraphs about 42") message = UserMessage(
content="hello world, write me a 2 sentence poem about the moon"
)
cprint(f"User>{message.content}", "green") cprint(f"User>{message.content}", "green")
iterator = client.chat_completion( iterator = client.chat_completion(
ChatCompletionRequest( ChatCompletionRequest(