Skip to content

Commit

Permalink
update inference prompt msg
Browse files Browse the repository at this point in the history
  • Loading branch information
yanxi0830 committed Sep 19, 2024
1 parent 9eb01dd commit a30b919
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion llama_stack/apis/inference/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,9 @@ async def chat_completion(self, request: ChatCompletionRequest) -> AsyncGenerato
async def run_main(host: str, port: int, stream: bool):
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")
iterator = client.chat_completion(
ChatCompletionRequest(
Expand Down

0 comments on commit a30b919

Please sign in to comment.