cnllm 0.1.0__tar.gz

cnllm-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 CNLLM Contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

cnllm-0.1.0/PKG-INFO

@@ -0,0 +1,257 @@
+Metadata-Version: 2.4
+Name: cnllm
+Version: 0.1.0
+Summary: 统一的中文大模型适配库，将中国大模型 API 输出转换为 OpenAI 格式，无缝接入openai、langchain等任何openai结构适配的python库
+Author-email: kanchengw <wangkancheng1122@163.com>
+License: MIT
+Project-URL: Homepage, https://github.com/kanchengw/cnllm
+Project-URL: Documentation, https://github.com/kanchengw/cnllm#readme
+Project-URL: Repository, https://github.com/kanchengw/cnllm
+Project-URL: Issues, https://github.com/kanchengw/cnllm/issues
+Keywords: ai,llm,openai,langchain,minimax,chinese-llm,adapter
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests>=2.25.0
+Requires-Dist: python-dotenv>=0.19.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: flake8>=6.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Dynamic: license-file
+
+# CNLLM - Chinese LLM Adapter
+
+[English](README_en.md) | 中文
+
+![PyPI Version](https://img.shields.io/pypi/v/cnllm)
+![Python Versions](https://img.shields.io/pypi/pyversions/cnllm)
+![License](https://img.shields.io/github/license/kanchengw/cnllm)
+![GitHub Stars](https://img.shields.io/github/stars/kanchengw/cnllm?style=social)
+![GitHub Forks](https://img.shields.io/github/forks/kanchengw/cnllm?style=social)
+
+---
+
+统一的中文大模型适配库，将各种国产大模型（如 MiniMax、字节豆包、Kimi 等）的 API 输出转换为统一的 OpenAI 格式，零成本接入 LangChain、AutoGen 等主流 AI 框架。
+
+## 特性
+
+- **OpenAI 兼容** - 所有输出完全对齐 OpenAI API 标准格式
+- **LangChain 原生支持** - 可直接使用 LangChain 的消息类型和工具函数
+- **统一接口** - 一套代码，无缝切换不同大模型
+- **流式输出** - 支持流式响应（规划中）
+- **重试机制** - 内置超时和自动重试
+- **详细日志** - 清晰的错误信息和调试支持
+
+## 支持的模型
+
+### 已验证
+- [x] MiniMax-M2.7
+- [x] MiniMax-M2.5
+
+### 开发中
+- [ ] 字节豆包 (Doubao)
+- [ ] Kimi (Moonshot)
+- [ ] 阶跃星辰 (StepFun)
+- [ ] 百度文心一言 (ERNIE)
+- [ ] 阿里通义千问 (Qwen)
+- [ ] 智谱 GLM (ChatGLM)
+
+## 安装
+
+```bash
+pip install cnllm
+```
+
+或从源码安装：
+
+```bash
+git clone https://github.com/yourusername/cnllm.git
+cd cnllm
+pip install -e .
+```
+
+## 快速开始
+
+### 基础使用
+
+```python
+from cnllm import CNLLM, MINIMAX_API_KEY
+
+# 初始化客户端
+client = CNLLM(
+    model="minimax-m2.7",  # 或 "minimax-m2.5"
+    api_key=MINIMAX_API_KEY
+)
+
+# 发送消息
+resp = client.chat.create(
+    messages=[
+        {"role": "user", "content": "用一句话介绍自己"}
+    ]
+)
+
+# 获取回复
+print(resp["choices"][0]["message"]["content"])
+```
+
+### 环境变量配置
+
+创建 `.env` 文件：
+
+```env
+MINIMAX_API_KEY=your_api_key_here
+```
+
+### 在 LangChain 中使用
+
+```python
+from langchain_core.messages import HumanMessage, AIMessage
+from cnllm import CNLLM, MINIMAX_API_KEY
+
+client = CNLLM(model="minimax-m2.7", api_key=MINIMAX_API_KEY)
+
+# CNLLM 的输出可以直接被 LangChain 使用
+resp = client.chat.create(
+    messages=[{"role": "user", "content": "你好"}]
+)
+
+# 转换为 LangChain 消息
+ai_msg = AIMessage(content=resp["choices"][0]["message"]["content"])
+print(ai_msg.content)
+```
+
+## API 参考
+
+### CNLLM 客户端
+
+```python
+from cnllm import CNLLM
+
+client = CNLLM(
+    model="minimax-m2.7",      # 模型名称
+    api_key="your_api_key",    # API 密钥
+    timeout=30,                # 请求超时（秒）
+    max_retries=3,             # 最大重试次数
+    retry_delay=1.0             # 重试延迟（秒）
+)
+```
+
+### chat.create()
+
+```python
+resp = client.chat.create(
+    messages=[
+        {"role": "system", "content": "你是一个有帮助的助手"},
+        {"role": "user", "content": "你好"}
+    ],
+    temperature=0.7,           # 温度参数
+    stream=False,              # 是否流式输出
+    model="minimax-m2.7"      # 可覆盖默认模型
+)
+```
+
+### 返回格式（OpenAI 标准）
+
+```python
+{
+    "id": "chatcmpl-xxx",
+    "object": "chat.completion",
+    "created": 1234567890,
+    "model": "minimax-m2.7",
+    "choices": [
+        {
+            "index": 0,
+            "message": {
+                "role": "assistant",
+                "content": "你好！有什么可以帮助你的吗？"
+            },
+            "finish_reason": "stop"
+        }
+    ],
+    "usage": {
+        "prompt_tokens": 20,
+        "completion_tokens": 15,
+        "total_tokens": 35
+    }
+}
+```
+
+## 项目结构
+
+```
+cnllm/
+├── adapters/              # 适配器层
+│   └── minimax/          # MiniMax 适配器
+│       └── chat.py
+├── core/                  # 核心组件
+│   ├── base.py          # HTTP 客户端
+│   ├── config.py        # 配置管理
+│   ├── exceptions.py    # 异常定义
+│   └── types.py         # 类型定义
+├── utils/                # 工具类
+│   └── cleaner.py       # 输出清理
+├── client.py             # 统一客户端入口
+└── __init__.py
+```
+
+## 错误处理
+
+```python
+from cnllm import CNLLM
+from cnllm.core.exceptions import ModelAPIError, ParseError
+
+try:
+    client = CNLLM(model="minimax-m2.7", api_key="invalid_key")
+    resp = client.chat.create(messages=[{"role": "user", "content": "你好"}])
+except ModelAPIError as e:
+    print(f"API 错误: {e}")
+except ParseError as e:
+    print(f"解析错误: {e}")
+except ValueError as e:
+    print(f"参数错误: {e}")
+```
+
+## 开发
+
+### 运行测试
+
+```bash
+# 安装开发依赖
+pip install -e .[dev]
+
+# 运行所有测试
+python test_CNLLM.py
+```
+
+### 添加新的模型适配器
+
+1. 在 `adapters/` 下创建新的适配器目录
+2. 实现 `create_completion()` 方法
+3. 实现 `_to_openai_format()` 转换方法
+4. 在 `client.py` 中注册适配器
+
+## 贡献
+
+欢迎提交 Issue 和 Pull Request！
+
+## 许可证
+
+MIT License - 详见 [LICENSE](LICENSE) 文件
+
+## 联系方式
+
+- GitHub Issues: [https://github.com/yourusername/cnllm/issues](https://github.com/yourusername/cnllm/issues)
+- Email: your.email@example.com

cnllm-0.1.0/README.md

@@ -0,0 +1,223 @@
+# CNLLM - Chinese LLM Adapter
+
+[English](README_en.md) | 中文
+
+![PyPI Version](https://img.shields.io/pypi/v/cnllm)
+![Python Versions](https://img.shields.io/pypi/pyversions/cnllm)
+![License](https://img.shields.io/github/license/kanchengw/cnllm)
+![GitHub Stars](https://img.shields.io/github/stars/kanchengw/cnllm?style=social)
+![GitHub Forks](https://img.shields.io/github/forks/kanchengw/cnllm?style=social)
+
+---
+
+统一的中文大模型适配库，将各种国产大模型（如 MiniMax、字节豆包、Kimi 等）的 API 输出转换为统一的 OpenAI 格式，零成本接入 LangChain、AutoGen 等主流 AI 框架。
+
+## 特性
+
+- **OpenAI 兼容** - 所有输出完全对齐 OpenAI API 标准格式
+- **LangChain 原生支持** - 可直接使用 LangChain 的消息类型和工具函数
+- **统一接口** - 一套代码，无缝切换不同大模型
+- **流式输出** - 支持流式响应（规划中）
+- **重试机制** - 内置超时和自动重试
+- **详细日志** - 清晰的错误信息和调试支持
+
+## 支持的模型
+
+### 已验证
+- [x] MiniMax-M2.7
+- [x] MiniMax-M2.5
+
+### 开发中
+- [ ] 字节豆包 (Doubao)
+- [ ] Kimi (Moonshot)
+- [ ] 阶跃星辰 (StepFun)
+- [ ] 百度文心一言 (ERNIE)
+- [ ] 阿里通义千问 (Qwen)
+- [ ] 智谱 GLM (ChatGLM)
+
+## 安装
+
+```bash
+pip install cnllm
+```
+
+或从源码安装：
+
+```bash
+git clone https://github.com/yourusername/cnllm.git
+cd cnllm
+pip install -e .
+```
+
+## 快速开始
+
+### 基础使用
+
+```python
+from cnllm import CNLLM, MINIMAX_API_KEY
+
+# 初始化客户端
+client = CNLLM(
+    model="minimax-m2.7",  # 或 "minimax-m2.5"
+    api_key=MINIMAX_API_KEY
+)
+
+# 发送消息
+resp = client.chat.create(
+    messages=[
+        {"role": "user", "content": "用一句话介绍自己"}
+    ]
+)
+
+# 获取回复
+print(resp["choices"][0]["message"]["content"])
+```
+
+### 环境变量配置
+
+创建 `.env` 文件：
+
+```env
+MINIMAX_API_KEY=your_api_key_here
+```
+
+### 在 LangChain 中使用
+
+```python
+from langchain_core.messages import HumanMessage, AIMessage
+from cnllm import CNLLM, MINIMAX_API_KEY
+
+client = CNLLM(model="minimax-m2.7", api_key=MINIMAX_API_KEY)
+
+# CNLLM 的输出可以直接被 LangChain 使用
+resp = client.chat.create(
+    messages=[{"role": "user", "content": "你好"}]
+)
+
+# 转换为 LangChain 消息
+ai_msg = AIMessage(content=resp["choices"][0]["message"]["content"])
+print(ai_msg.content)
+```
+
+## API 参考
+
+### CNLLM 客户端
+
+```python
+from cnllm import CNLLM
+
+client = CNLLM(
+    model="minimax-m2.7",      # 模型名称
+    api_key="your_api_key",    # API 密钥
+    timeout=30,                # 请求超时（秒）
+    max_retries=3,             # 最大重试次数
+    retry_delay=1.0             # 重试延迟（秒）
+)
+```
+
+### chat.create()
+
+```python
+resp = client.chat.create(
+    messages=[
+        {"role": "system", "content": "你是一个有帮助的助手"},
+        {"role": "user", "content": "你好"}
+    ],
+    temperature=0.7,           # 温度参数
+    stream=False,              # 是否流式输出
+    model="minimax-m2.7"      # 可覆盖默认模型
+)
+```
+
+### 返回格式（OpenAI 标准）
+
+```python
+{
+    "id": "chatcmpl-xxx",
+    "object": "chat.completion",
+    "created": 1234567890,
+    "model": "minimax-m2.7",
+    "choices": [
+        {
+            "index": 0,
+            "message": {
+                "role": "assistant",
+                "content": "你好！有什么可以帮助你的吗？"
+            },
+            "finish_reason": "stop"
+        }
+    ],
+    "usage": {
+        "prompt_tokens": 20,
+        "completion_tokens": 15,
+        "total_tokens": 35
+    }
+}
+```
+
+## 项目结构
+
+```
+cnllm/
+├── adapters/              # 适配器层
+│   └── minimax/          # MiniMax 适配器
+│       └── chat.py
+├── core/                  # 核心组件
+│   ├── base.py          # HTTP 客户端
+│   ├── config.py        # 配置管理
+│   ├── exceptions.py    # 异常定义
+│   └── types.py         # 类型定义
+├── utils/                # 工具类
+│   └── cleaner.py       # 输出清理
+├── client.py             # 统一客户端入口
+└── __init__.py
+```
+
+## 错误处理
+
+```python
+from cnllm import CNLLM
+from cnllm.core.exceptions import ModelAPIError, ParseError
+
+try:
+    client = CNLLM(model="minimax-m2.7", api_key="invalid_key")
+    resp = client.chat.create(messages=[{"role": "user", "content": "你好"}])
+except ModelAPIError as e:
+    print(f"API 错误: {e}")
+except ParseError as e:
+    print(f"解析错误: {e}")
+except ValueError as e:
+    print(f"参数错误: {e}")
+```
+
+## 开发
+
+### 运行测试
+
+```bash
+# 安装开发依赖
+pip install -e .[dev]
+
+# 运行所有测试
+python test_CNLLM.py
+```
+
+### 添加新的模型适配器
+
+1. 在 `adapters/` 下创建新的适配器目录
+2. 实现 `create_completion()` 方法
+3. 实现 `_to_openai_format()` 转换方法
+4. 在 `client.py` 中注册适配器
+
+## 贡献
+
+欢迎提交 Issue 和 Pull Request！
+
+## 许可证
+
+MIT License - 详见 [LICENSE](LICENSE) 文件
+
+## 联系方式
+
+- GitHub Issues: [https://github.com/yourusername/cnllm/issues](https://github.com/yourusername/cnllm/issues)
+- Email: your.email@example.com

cnllm-0.1.0/cnllm/__init__.py

@@ -0,0 +1,5 @@
+from .client import CNLLM
+from .core.config import MINIMAX_API_KEY
+
+__version__ = "0.1.0"
+__all__ = ["CNLLM", "MINIMAX_API_KEY"]

cnllm-0.1.0/cnllm/adapters/minimax/chat.py

@@ -0,0 +1,116 @@
+import re
+import uuid
+import time
+from typing import Dict, Any, List, Optional
+from ...core.base import BaseHttpClient
+from ...core.exceptions import ParseError, ModelAPIError
+from ...utils.cleaner import OutputCleaner
+
+
+class MiniMaxAdapter:
+    SUPPORTED_MODELS = ["minimax-m2.7", "minimax-m2.5"]
+    DEFAULT_MODEL = "minimax-m2.7"
+    
+    def __init__(
+        self,
+        api_key: str,
+        model: str = None,
+        timeout: int = 30,
+        max_retries: int = 3,
+        retry_delay: float = 1.0
+    ):
+        self.client = BaseHttpClient(
+            api_key=api_key,
+            base_url="https://api.minimaxi.com",
+            timeout=timeout,
+            max_retries=max_retries,
+            retry_delay=retry_delay
+        )
+        self.cleaner = OutputCleaner()
+        self.model = model or self.DEFAULT_MODEL
+        
+        if self.model not in self.SUPPORTED_MODELS:
+            raise ValueError(
+                f"不支持的模型: {model}\n"
+                f"支持的模型: {', '.join(self.SUPPORTED_MODELS)}"
+            )
+
+    def create_completion(
+        self,
+        messages: List[Dict[str, str]],
+        temperature: float = 0.1,
+        stream: bool = False,
+        model: str = None
+    ) -> Dict[str, Any]:
+        use_model = model or self.model
+        
+        if use_model not in self.SUPPORTED_MODELS:
+            raise ValueError(f"不支持的模型: {use_model}")
+        
+        payload = {
+            "model": self._to_minimax_model_name(use_model),
+            "messages": messages,
+            "temperature": temperature,
+            "stream": stream
+        }
+
+        try:
+            raw_resp = self.client.post("/v1/text/chatcompletion_v2", payload)
+        except RuntimeError as e:
+            raise ModelAPIError(f"MiniMax API 请求失败: {e}")
+        
+        return self._to_openai_format(raw_resp, use_model)
+
+    def _to_minimax_model_name(self, model: str) -> str:
+        """将统一模型名转换为MiniMax API模型名"""
+        mapping = {
+            "minimax-m2.7": "MiniMax-M2.7",
+            "minimax-m2.5": "MiniMax-M2.5"
+        }
+        return mapping.get(model, model)
+
+    def _to_openai_format(self, raw: Dict[str, Any], model: str) -> Dict[str, Any]:
+        """
+        MiniMax M2.5/M2.7 原生响应 → 转换为 OpenAI 标准结构
+        """
+        base_resp = raw.get("base_resp", {})
+        if base_resp.get("status_code") and base_resp["status_code"] != 0:
+            raise ModelAPIError(
+                f"MiniMax API 错误: {base_resp.get('status_msg', '未知错误')}\n"
+                f"状态码: {base_resp.get('status_code')}"
+            )
+        
+        try:
+            raw_content = raw["choices"][0]["message"]["content"]
+            finish_reason = raw["choices"][0].get("finish_reason", "stop")
+        except (KeyError, IndexError) as e:
+            raise ParseError(f"解析响应失败: 缺少必要字段 {e}\n原始响应: {raw}")
+        
+        cleaned = self.cleaner.clean(raw_content)
+        
+        usage = raw.get("usage", {})
+        prompt_tokens = usage.get("prompt_tokens", 0)
+        completion_tokens = usage.get("completion_tokens", 0)
+        total_tokens = usage.get("total_tokens", 0)
+        
+        return {
+            "id": f"chatcmpl-{uuid.uuid4().hex[:24]}",
+            "object": "chat.completion",
+            "created": int(time.time()),
+            "model": model,
+            "choices": [
+                {
+                    "index": 0,
+                    "message": {
+                        "role": "assistant",
+                        "content": cleaned
+                    },
+                    "finish_reason": finish_reason
+                }
+            ],
+            "usage": {
+                "prompt_tokens": prompt_tokens,
+                "completion_tokens": completion_tokens,
+                "total_tokens": total_tokens
+            }
+        }

cnllm-0.1.0/cnllm/client.py

@@ -0,0 +1,73 @@
+from typing import Optional, Dict, Any
+from dataclasses import dataclass
+
+from .adapters.minimax.chat import MiniMaxAdapter
+
+
+class CNLLM:
+    SUPPORTED_MODELS = ["minimax", "minimax-m2.5", "minimax-m2.7"]
+    
+    def __init__(
+        self,
+        model: str,
+        api_key: str,
+        timeout: int = 30,
+        max_retries: int = 3,
+        retry_delay: float = 1.0
+    ):
+        self.model = self._normalize_model(model)
+        self.api_key = api_key
+        self.timeout = timeout
+        self.max_retries = max_retries
+        self.retry_delay = retry_delay
+        self.adapter = self._get_adapter()
+        self.chat = self.ChatNamespace(self)
+
+    def _normalize_model(self, model: str) -> str:
+        model = model.lower()
+        if model == "minimax":
+            model = "minimax-m2.7"
+        return model
+
+    def _get_adapter(self):
+        if self.model.startswith("minimax"):
+            return MiniMaxAdapter(
+                api_key=self.api_key,
+                model=self.model,
+                timeout=self.timeout,
+                max_retries=self.max_retries,
+                retry_delay=self.retry_delay
+            )
+        raise ValueError(f"暂不支持模型: {self.model}\n支持的模型: {', '.join(self.SUPPORTED_MODELS)}")
+
+    def create_chat_completion(
+        self,
+        messages: list[Dict[str, str]],
+        temperature: float = 0.1,
+        stream: bool = False,
+        model: str = None
+    ) -> Dict[str, Any]:
+        return self.adapter.create_completion(
+            messages=messages,
+            temperature=temperature,
+            stream=stream,
+            model=model
+        )
+
+    class ChatNamespace:
+        def __init__(self, parent):
+            self.parent = parent
+
+        def create(
+            self,
+            messages: list[Dict[str, str]],
+            temperature: float = 0.1,
+            stream: bool = False,
+            model: str = None
+        ):
+            return self.parent.create_chat_completion(
+                messages=messages,
+                temperature=temperature,
+                stream=stream,
+                model=model
+            )

cnllm-0.1.0/cnllm/core/base.py

@@ -0,0 +1,88 @@
+import time
+import requests
+from typing import Dict, Any
+from .exceptions import ModelAPIError
+
+
+class BaseHttpClient:
+    def __init__(
+        self,
+        api_key: str,
+        base_url: str,
+        timeout: int = 30,
+        max_retries: int = 3,
+        retry_delay: float = 1.0
+    ):
+        self.api_key = ''.join(api_key.strip().split())
+        self.base_url = base_url.strip()
+        self.timeout = timeout
+        self.max_retries = max_retries
+        self.retry_delay = retry_delay
+
+    def post(self, path: str, payload: Dict[str, Any]) -> Dict[str, Any]:
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json; charset=utf-8"
+        }
+        
+        last_error = None
+        for attempt in range(self.max_retries):
+            try:
+                response = requests.post(
+                    url=f"{self.base_url}{path}",
+                    headers=headers,
+                    json=payload,
+                    timeout=self.timeout
+                )
+                
+                if response.status_code == 429:
+                    wait_time = self.retry_delay * (2 ** attempt)
+                    raise ModelAPIError(
+                        f"请求被限流 (429)。\n"
+                        f"等待 {wait_time:.1f} 秒后重试... (尝试 {attempt + 1}/{self.max_retries})"
+                    )
+                
+                response.raise_for_status()
+                return response.json()
+                
+            except requests.exceptions.Timeout:
+                last_error = f"请求超时 (timeout={self.timeout}s)"
+                if attempt < self.max_retries - 1:
+                    time.sleep(self.retry_delay)
+                    continue
+                    
+            except requests.exceptions.ConnectionError as e:
+                last_error = f"连接失败: {str(e)}"
+                if attempt < self.max_retries - 1:
+                    time.sleep(self.retry_delay)
+                    continue
+                    
+            except requests.exceptions.HTTPError as e:
+                status_code = e.response.status_code
+                if status_code == 429:
+                    wait_time = self.retry_delay * (2 ** attempt)
+                    if attempt < self.max_retries - 1:
+                        time.sleep(wait_time)
+                        continue
+                    raise ModelAPIError(
+                        f"请求被限流 (429)。已重试 {self.max_retries} 次。\n"
+                        f"请稍后重试或联系 API 提供商。"
+                    )
+                elif status_code >= 500:
+                    last_error = f"服务器错误 ({status_code})"
+                    if attempt < self.max_retries - 1:
+                        time.sleep(self.retry_delay * (2 ** attempt))
+                        continue
+                    raise ModelAPIError(
+                        f"API 服务器错误 ({status_code})。已重试 {self.max_retries} 次。\n"
+                        f"请稍后重试。"
+                    )
+                else:
+                    raise ModelAPIError(
+                        f"API 请求失败 (HTTP {status_code}): {str(e)}\n"
+                        f"请检查 API Key 和请求参数。"
+                    )
+            except Exception as e:
+                raise ModelAPIError(f"API 请求失败: {str(e)}")
+        
+        raise ModelAPIError(f"重试 {self.max_retries} 次后仍然失败。最后错误: {last_error}")

cnllm-0.1.0/cnllm/core/config.py

@@ -0,0 +1,14 @@
+import os
+from dotenv import load_dotenv
+
+# 自动加载 .env 文件
+load_dotenv()
+
+def get_env(key: str, default: str = None) -> str:
+    value = os.getenv(key, default)
+    if not value:
+        raise ValueError(f"环境变量 {key} 未设置，请检查 .env 文件")
+    return value.strip()
+
+# MiniMax
+MINIMAX_API_KEY = get_env("MINIMAX_API_KEY")

cnllm-0.1.0/cnllm/core/exceptions.py

@@ -0,0 +1,5 @@
+class ModelAPIError(Exception):
+    pass
+
+class ParseError(Exception):
+    pass

cnllm-0.1.0/cnllm/core/types.py

@@ -0,0 +1,13 @@
+from dataclasses import dataclass
+from typing import Optional
+
+@dataclass
+class Message:
+    role: str
+    content: str
+
+@dataclass
+class ChatResponse:
+    content: str
+    role: str = "assistant"
+    finish_reason: Optional[str] = "stop"

cnllm-0.1.0/cnllm/example.py

@@ -0,0 +1,13 @@
+from cnllm import MiniMaxChat, Message
+
+llm = MiniMaxChat(
+    api_key="你的 MINIMAX_API_KEY",
+    model="MiniMax-M1"
+)
+
+messages = [
+    Message(role="user", content="提取实验标题和总结")
+]
+
+resp = llm.invoke(messages)
+print(resp.content)

cnllm-0.1.0/cnllm/utils/cleaner.py

@@ -0,0 +1,12 @@
+import re
+
+
+class OutputCleaner:
+    @staticmethod
+    def clean(text: str) -> str:
+        if not text:
+            return text
+        
+        text = re.sub(r'<think>[\s\S]*?</think>', '', text)
+        text = re.sub(r'\n{3,}', '\n\n', text)
+        return text.strip()

cnllm-0.1.0/cnllm.egg-info/PKG-INFO

@@ -0,0 +1,257 @@
+Metadata-Version: 2.4
+Name: cnllm
+Version: 0.1.0
+Summary: 统一的中文大模型适配库，将中国大模型 API 输出转换为 OpenAI 格式，无缝接入openai、langchain等任何openai结构适配的python库
+Author-email: kanchengw <wangkancheng1122@163.com>
+License: MIT
+Project-URL: Homepage, https://github.com/kanchengw/cnllm
+Project-URL: Documentation, https://github.com/kanchengw/cnllm#readme
+Project-URL: Repository, https://github.com/kanchengw/cnllm
+Project-URL: Issues, https://github.com/kanchengw/cnllm/issues
+Keywords: ai,llm,openai,langchain,minimax,chinese-llm,adapter
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests>=2.25.0
+Requires-Dist: python-dotenv>=0.19.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: flake8>=6.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Dynamic: license-file
+
+# CNLLM - Chinese LLM Adapter
+
+[English](README_en.md) | 中文
+
+![PyPI Version](https://img.shields.io/pypi/v/cnllm)
+![Python Versions](https://img.shields.io/pypi/pyversions/cnllm)
+![License](https://img.shields.io/github/license/kanchengw/cnllm)
+![GitHub Stars](https://img.shields.io/github/stars/kanchengw/cnllm?style=social)
+![GitHub Forks](https://img.shields.io/github/forks/kanchengw/cnllm?style=social)
+
+---
+
+统一的中文大模型适配库，将各种国产大模型（如 MiniMax、字节豆包、Kimi 等）的 API 输出转换为统一的 OpenAI 格式，零成本接入 LangChain、AutoGen 等主流 AI 框架。
+
+## 特性
+
+- **OpenAI 兼容** - 所有输出完全对齐 OpenAI API 标准格式
+- **LangChain 原生支持** - 可直接使用 LangChain 的消息类型和工具函数
+- **统一接口** - 一套代码，无缝切换不同大模型
+- **流式输出** - 支持流式响应（规划中）
+- **重试机制** - 内置超时和自动重试
+- **详细日志** - 清晰的错误信息和调试支持
+
+## 支持的模型
+
+### 已验证
+- [x] MiniMax-M2.7
+- [x] MiniMax-M2.5
+
+### 开发中
+- [ ] 字节豆包 (Doubao)
+- [ ] Kimi (Moonshot)
+- [ ] 阶跃星辰 (StepFun)
+- [ ] 百度文心一言 (ERNIE)
+- [ ] 阿里通义千问 (Qwen)
+- [ ] 智谱 GLM (ChatGLM)
+
+## 安装
+
+```bash
+pip install cnllm
+```
+
+或从源码安装：
+
+```bash
+git clone https://github.com/yourusername/cnllm.git
+cd cnllm
+pip install -e .
+```
+
+## 快速开始
+
+### 基础使用
+
+```python
+from cnllm import CNLLM, MINIMAX_API_KEY
+
+# 初始化客户端
+client = CNLLM(
+    model="minimax-m2.7",  # 或 "minimax-m2.5"
+    api_key=MINIMAX_API_KEY
+)
+
+# 发送消息
+resp = client.chat.create(
+    messages=[
+        {"role": "user", "content": "用一句话介绍自己"}
+    ]
+)
+
+# 获取回复
+print(resp["choices"][0]["message"]["content"])
+```
+
+### 环境变量配置
+
+创建 `.env` 文件：
+
+```env
+MINIMAX_API_KEY=your_api_key_here
+```
+
+### 在 LangChain 中使用
+
+```python
+from langchain_core.messages import HumanMessage, AIMessage
+from cnllm import CNLLM, MINIMAX_API_KEY
+
+client = CNLLM(model="minimax-m2.7", api_key=MINIMAX_API_KEY)
+
+# CNLLM 的输出可以直接被 LangChain 使用
+resp = client.chat.create(
+    messages=[{"role": "user", "content": "你好"}]
+)
+
+# 转换为 LangChain 消息
+ai_msg = AIMessage(content=resp["choices"][0]["message"]["content"])
+print(ai_msg.content)
+```
+
+## API 参考
+
+### CNLLM 客户端
+
+```python
+from cnllm import CNLLM
+
+client = CNLLM(
+    model="minimax-m2.7",      # 模型名称
+    api_key="your_api_key",    # API 密钥
+    timeout=30,                # 请求超时（秒）
+    max_retries=3,             # 最大重试次数
+    retry_delay=1.0             # 重试延迟（秒）
+)
+```
+
+### chat.create()
+
+```python
+resp = client.chat.create(
+    messages=[
+        {"role": "system", "content": "你是一个有帮助的助手"},
+        {"role": "user", "content": "你好"}
+    ],
+    temperature=0.7,           # 温度参数
+    stream=False,              # 是否流式输出
+    model="minimax-m2.7"      # 可覆盖默认模型
+)
+```
+
+### 返回格式（OpenAI 标准）
+
+```python
+{
+    "id": "chatcmpl-xxx",
+    "object": "chat.completion",
+    "created": 1234567890,
+    "model": "minimax-m2.7",
+    "choices": [
+        {
+            "index": 0,
+            "message": {
+                "role": "assistant",
+                "content": "你好！有什么可以帮助你的吗？"
+            },
+            "finish_reason": "stop"
+        }
+    ],
+    "usage": {
+        "prompt_tokens": 20,
+        "completion_tokens": 15,
+        "total_tokens": 35
+    }
+}
+```
+
+## 项目结构
+
+```
+cnllm/
+├── adapters/              # 适配器层
+│   └── minimax/          # MiniMax 适配器
+│       └── chat.py
+├── core/                  # 核心组件
+│   ├── base.py          # HTTP 客户端
+│   ├── config.py        # 配置管理
+│   ├── exceptions.py    # 异常定义
+│   └── types.py         # 类型定义
+├── utils/                # 工具类
+│   └── cleaner.py       # 输出清理
+├── client.py             # 统一客户端入口
+└── __init__.py
+```
+
+## 错误处理
+
+```python
+from cnllm import CNLLM
+from cnllm.core.exceptions import ModelAPIError, ParseError
+
+try:
+    client = CNLLM(model="minimax-m2.7", api_key="invalid_key")
+    resp = client.chat.create(messages=[{"role": "user", "content": "你好"}])
+except ModelAPIError as e:
+    print(f"API 错误: {e}")
+except ParseError as e:
+    print(f"解析错误: {e}")
+except ValueError as e:
+    print(f"参数错误: {e}")
+```
+
+## 开发
+
+### 运行测试
+
+```bash
+# 安装开发依赖
+pip install -e .[dev]
+
+# 运行所有测试
+python test_CNLLM.py
+```
+
+### 添加新的模型适配器
+
+1. 在 `adapters/` 下创建新的适配器目录
+2. 实现 `create_completion()` 方法
+3. 实现 `_to_openai_format()` 转换方法
+4. 在 `client.py` 中注册适配器
+
+## 贡献
+
+欢迎提交 Issue 和 Pull Request！
+
+## 许可证
+
+MIT License - 详见 [LICENSE](LICENSE) 文件
+
+## 联系方式
+
+- GitHub Issues: [https://github.com/yourusername/cnllm/issues](https://github.com/yourusername/cnllm/issues)
+- Email: your.email@example.com

cnllm-0.1.0/cnllm.egg-info/SOURCES.txt

@@ -0,0 +1,21 @@
+LICENSE
+README.md
+pyproject.toml
+cnllm/__init__.py
+cnllm/client.py
+cnllm/example.py
+cnllm.egg-info/PKG-INFO
+cnllm.egg-info/SOURCES.txt
+cnllm.egg-info/dependency_links.txt
+cnllm.egg-info/requires.txt
+cnllm.egg-info/top_level.txt
+cnllm/adapters/__init__.py
+cnllm/adapters/minimax/__init__.py
+cnllm/adapters/minimax/chat.py
+cnllm/core/__init__.py
+cnllm/core/base.py
+cnllm/core/config.py
+cnllm/core/exceptions.py
+cnllm/core/types.py
+cnllm/utils/__init__.py
+cnllm/utils/cleaner.py

cnllm-0.1.0/cnllm.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

cnllm-0.1.0/cnllm.egg-info/requires.txt

@@ -0,0 +1,9 @@
+requests>=2.25.0
+python-dotenv>=0.19.0
+
+[dev]
+pytest>=7.0.0
+pytest-cov>=4.0.0
+black>=23.0.0
+flake8>=6.0.0
+mypy>=1.0.0

cnllm-0.1.0/cnllm.egg-info/top_level.txt

@@ -0,0 +1 @@
+cnllm

cnllm-0.1.0/pyproject.toml

@@ -0,0 +1,70 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "cnllm"
+version = "0.1.0"
+description = "统一的中文大模型适配库，将中国大模型 API 输出转换为 OpenAI 格式，无缝接入openai、langchain等任何openai结构适配的python库"
+readme = "README.md"
+license = {text = "MIT"}
+authors = [
+    {name = "kanchengw", email = "wangkancheng1122@163.com"}
+]
+keywords = ["ai", "llm", "openai", "langchain", "minimax", "chinese-llm", "adapter"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+requires-python = ">=3.8"
+dependencies = [
+    "requests>=2.25.0",
+    "python-dotenv>=0.19.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0.0",
+    "pytest-cov>=4.0.0",
+    "black>=23.0.0",
+    "flake8>=6.0.0",
+    "mypy>=1.0.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/kanchengw/cnllm"
+Documentation = "https://github.com/kanchengw/cnllm#readme"
+Repository = "https://github.com/kanchengw/cnllm"
+Issues = "https://github.com/kanchengw/cnllm/issues"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["cnllm*"]
+
+[tool.black]
+line-length = 100
+target-version = ['py38', 'py39', 'py310', 'py311', 'py312']
+
+[tool.isort]
+profile = "black"
+line_length = 100
+
+[tool.mypy]
+python_version = "3.8"
+warn_return_any = true
+warn_unused_configs = true
+disallow_untyped_defs = false
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+python_functions = ["test_*"]
+addopts = "-v --tb=short"

cnllm-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+