更新版本至 1.1.12，优化 LLM 构建逻辑并引入必要的模块

Author: zhangshulin

docapi/llm/llm_builder.py

@@ -1,8 +1,4 @@
 import os
-from docapi.llm.openai_llm import OpenAILLM
-from docapi.llm.azure_openai_llm import AzureOpenAILLM
-from docapi.llm.baidu_llm import BaiduLLM
-from docapi.llm.zhipu_llm import ZhipuLLM
 
 
 def build_llm(model):
@@ -11,6 +7,8 @@ def build_llm(model):
     model_name = model_name.strip()
 
     if provider in ['deepseek', 'baichuan', 'doubao', 'moonshot', 'openai', 'xai', 'aliyun', 'open-source']:
+        from docapi.llm.openai_llm import OpenAILLM
+
         if provider == 'deepseek':
             api_key = os.getenv('DEEPSEEK_API_KEY')
             base_url = 'https://api.deepseek.com'
@@ -42,17 +40,23 @@ def build_llm(model):
         return OpenAILLM(api_key=api_key, base_url=base_url, model=model_name)
 
     elif provider == 'azure-openai':
+        from docapi.llm.azure_openai_llm import AzureOpenAILLM
+
         api_key = os.getenv('AZURE_OPENAI_API_KEY')
         endpoint = os.getenv('AZURE_OPENAI_ENDPOINT')
         api_version = os.getenv('OPENAI_API_VERSION')
         return AzureOpenAILLM(api_key=api_key, endpoint=endpoint, api_version=api_version, model=model_name)
 
     elif provider == 'baidu':
+        from docapi.llm.baidu_llm import BaiduLLM
+
         access_key = os.getenv('QIANFAN_ACCESS_KEY')
         secret_key = os.getenv('QIANFAN_SECRET_KEY')
         return BaiduLLM(access_key=access_key, secret_key=secret_key, model=model_name)
 
     elif provider == 'zhipu':
+        from docapi.llm.zhipu_llm import ZhipuLLM
+
         api_key = os.getenv('ZHIPUAI_API_KEY')
         return ZhipuLLM(api_key=api_key, model=model_name)
 

docapi/main.py

@@ -9,7 +9,7 @@
 from docapi.docapi import DocAPI
 
 
-VERSION = '1.1.11'
+VERSION = '1.1.12'
 
 
 class Main:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docapi"
-version = "1.1.11"
+version = "1.1.12"
 description = "DocAPI is a Python package that automatically generates API documentation using LLMs. It currently supports Flask and Django frameworks."
 authors = ["zhangshulin <zhangslwork@yeah.net>"]
 license = "Apache-2.0"