使用openai库调用大模型,记录一下之前的测试代码;

#!/usr/bin/env python3
# coding = utf-8
"""
# Project: workspace_py
# File: test_openai.py
# Author: XWF
# Time: 2025/12/24 17:24
"""
import base64
from pathlib import Path

from openai import OpenAI

client = OpenAI(api_key='test', base_url='http://192.168.1.32:11434/v1')
print(client.models.list())
response_stream = client.chat.completions.create(
    model='qwen3-vl:32b',
    messages=[{'role': 'user', 'content': '你是谁'}],
    stream=True
)
# print(response.choices[0].message.content)
for response in response_stream:
    print(response.choices[0].delta.content, end='')
print()


# 识别图片
img = base64.b64encode(Path('./abc.jpg').read_bytes()).decode()

# img = cv2.imread('a.jpg')
# success, img = cv2.imencode('.jpg', img)
# print('imencode success?', success)
# img = base64.b64encode(img.tobytes()).decode('utf-8')

img_url = f'data:image/jpeg;base64,{img}'

client = OpenAI(api_key='test', base_url='http://192.168.1.32:11434/v1')
print(client.models.list())
response = client.chat.completions.create(
    model='qwen3-vl:32b',
    messages=[{'role': 'system', 'content': '你是一个图片处理员'},
              {'role': 'user', 'content': [
                {'type': 'text', 'text': '图片中有什么'},
                # {'type': 'text', 'text': '图片中的人物是贝优妮塔吗?'},
                {'type': 'image_url', 'image_url': {'url': img_url}}
                ]
               }
              ],
    stream=False
)
print(response.choices[0].message.content)

执行结果:

更多推荐