openai-simple-chat 0.1.0

大模型对话标准接口。支持模板对话、JSON对话等。

openai-simple-chat

大模型对话标准接口。支持模板对话、JSON对话等。

安装

pip install openai-simple-chat

环境变量配置项

• OPENAI_BASE_URL # 支持openai兼容服务
• OPENAI_API_KEY
• OPENAI_CHAT_MODEL
• OLLAMA_BASE_URL # 支持ollama兼容服务
• OLLAMA_API_KEY
• OLLAMA_CHAT_MODEL
• OPENAI_SIMPLE_CHAT_TEMPLATE_ENGINE # 其它设置
• OPENAI_SIMPLE_CHAT_LOGGER_NAME

使用方法

test_templates/calc.txt

以标准json返回以下计算结果数值【输出格式为：{"result": xx}】：{{expression}}

main1.py

import openai_simple_chat

llm = openai_simple_chat.OpenAIChatService(
    template_engine=openai_simple_chat.get_template_prompt_by_jinjia2,
    template_root="test_templates",
)
response, response_info = llm.jsonchat(
    template="calc.txt",
    expression="1+1",
)
assert response
assert response_info
assert isinstance(response, dict)
assert isinstance(response_info, dict)
assert "result" in response
assert response["result"] == 2

# 注意，如果是stream_chat的话，response可能为空字符串。

main2.py

import openai_simple_chat

llm = openai_simple_chat.OllamaChatService(
    template_engine=openai_simple_chat.get_template_prompt_by_jinjia2,
    template_root="test_templates",
)
response, response_info = llm.jsonchat(template="calc.txt", expression="1+1")
assert response
assert response_info
assert isinstance(response, dict)
assert isinstance(response_info, dict)
assert "result" in response
assert response["result"] == 2

版本记录

v0.1.0

• 版本首发。
• 支持模板对话。
• 支持json对话。
• 兼容openai和ollama服务。
• 兼容django和jinja2模板引擎。
• jsonchat已经对deekseek输出的think过程输出进行处理。