@@ -151,22 +151,42 @@ client = OpenAI(
151151model_type = client.models.list ().data[0].id
152152print(f' model_type: {model_type}' )
153153
154- query = ' 浙江 -> 杭州\n安徽 -> 合肥\n四川 ->'
155- kwargs = {' model' : model_type, ' messages' : query, ' seed' : 42, ' temperature' : 0.1, ' max_tokens' : 32}
156-
157- resp = client.chat.completions.create(** kwargs)
158- response = resp.choices[0].text
154+ query = ' 浙江的省会在哪里?'
155+ messages = [{
156+ ' role' : ' user' ,
157+ ' content' : query
158+ }]
159+ resp = client.chat.completions.create(
160+ model=model_type,
161+ messages=messages,
162+ seed=42)
163+ response = resp.choices[0].message.content
159164print(f' query: {query}' )
160165print(f' response: {response}' )
161166
162167# 流式
163- stream_resp = client.completions.create(stream=True, ** kwargs)
164- response = resp.choices[0].text
168+ messages.append({' role' : ' assistant' , ' content' : response})
169+ query = ' 这有什么好吃的?'
170+ messages.append({' role' : ' user' , ' content' : query})
171+ stream_resp = client.chat.completions.create(
172+ model=model_type,
173+ messages=messages,
174+ stream=True,
175+ seed=42)
176+
165177print(f' query: {query}' )
166178print(' response: ' , end=' ' )
167179for chunk in stream_resp:
168- print(chunk.choices[0].text , end=' ' , flush=True)
180+ print(chunk.choices[0].delta.content , end=' ' , flush=True)
169181print ()
182+
183+ " " " Out[0]
184+ model_type: qwen-7b-chat
185+ query: 浙江的省会在哪里?
186+ response: 浙江省的省会是杭州市。
187+ query: 这有什么好吃的?
188+ response: 杭州有许多美食,例如西湖醋鱼、东坡肉、龙井虾仁、叫化童子鸡等。此外,杭州还有许多特色小吃,如西湖藕粉、杭州小笼包、杭州油条等。
189+ " " "
170190` ` `
171191
172192# llama.cpp
0 commit comments