Skip to content

GitLab

Commit 0f7a3e5f authored by Atream's avatar Atream
Browse files

fix-client

parent cc94a02a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ktransformers/tests/test_client.py
View file @ 0f7a3e5f
...@@ -93,12 +93,12 @@ async def main(prompt_id, model, stream, max_tokens, temperature, top_p): ...@@ -93,12 +93,12 @@ async def main(prompt_id, model, stream, max_tokens, temperature, top_p):
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Event Stream Request Tester") parser = argparse.ArgumentParser(description="Event Stream Request Tester")
parser.add_argument("--question_id", type=int, default=0) parser.add_argument("--question_id", type=int, default=0)
parser.add_argument("--model", type=str, required=True) parser.add_argument("--model", type=str, default="DeepSeek-V3")
parser.add_argument("--stream", type=bool, default=True) parser.add_argument("--stream", type=bool, default=True)
parser.add_argument("--max_tokens", type=int, default=500) parser.add_argument("--max_tokens", type=int, default=500)
parser.add_argument("--temperature", type=float, default=0.8) parser.add_argument("--temperature", type=float, default=0.8)
parser.add_argument("--top_p", type=float, default=1) parser.add_argument("--top_p", type=float, default=1)
parser.add_argument("--api_url", type=str, default="http://localhost:10006/v1/chat/completions", help="API URL") parser.add_argument("--api_url", type=str, default="http://localhost:10002/v1/chat/completions", help="API URL")
args = parser.parse_args() args = parser.parse_args()
SERVER_URL = args.api_url SERVER_URL = args.api_url
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment