From 3c9ce684a9b8d4cf24d5405823f6fe2366ef0e55 Mon Sep 17 00:00:00 2001 From: Hatter Jiang Date: Sun, 19 Apr 2026 10:16:47 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=84=20Update=20chat=20completion=20log?= =?UTF-8?q?ic=20and=20improve=20debugging=20output=20in=20main.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- projects/local-api-test/main.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/projects/local-api-test/main.py b/projects/local-api-test/main.py index abdac15..39eed48 100644 --- a/projects/local-api-test/main.py +++ b/projects/local-api-test/main.py @@ -1,20 +1,29 @@ import os +from pprint import pprint from openai import OpenAI from openai._types import Timeout +from openai.types.chat import ChatCompletionUserMessageParam, ChatCompletionSystemMessageParam def main(): client = OpenAI( api_key=os.environ.get("OPENAI_API_KEY", "sk-dummy"), base_url=os.environ.get("OPENAI_API_BASE", "http://localhost:11434/v1"), - timeout=Timeout(timeout=30), + timeout=Timeout(30, connect=5), ) response = client.chat.completions.create( - model=os.environ.get("OPENAI_MODEL", "llama3"), - messages=[{"role": "user", "content": "Hello!"}], + model=os.environ.get("OPENAI_MODEL", "qwen-math-turbo"), + messages=[ + ChatCompletionSystemMessageParam(role="system", content="You are a useful assistant."), + # ChatCompletionUserMessageParam(**{"role": "user", "content": "Hello!"}), + ChatCompletionUserMessageParam(role="user", content="Hello!"), + ], ) + print() + pprint(vars(response)) + print() print(response.choices[0].message.content)