hgagent / lagent /llms /__init__.py
Chisleep
update01
f8d0193
from .anthropic_llm import AsyncClaudeAPI, ClaudeAPI
from .base_api import AsyncBaseAPILLM, BaseAPILLM
from .base_llm import AsyncBaseLLM, BaseLLM
from .huggingface import HFTransformer, HFTransformerCasualLM, HFTransformerChat
from .lmdeploy_wrapper import (
AsyncLMDeployClient,
AsyncLMDeployPipeline,
AsyncLMDeployServer,
LMDeployClient,
LMDeployPipeline,
LMDeployServer,
)
from .meta_template import INTERNLM2_META
from .openai import GPTAPI, AsyncGPTAPI
from .sensenova import SensenovaAPI
from .vllm_wrapper import AsyncVllmModel, VllmModel
__all__ = [
'AsyncBaseLLM',
'BaseLLM',
'AsyncBaseAPILLM',
'BaseAPILLM',
'AsyncGPTAPI',
'GPTAPI',
'LMDeployClient',
'AsyncLMDeployClient',
'LMDeployPipeline',
'AsyncLMDeployPipeline',
'LMDeployServer',
'AsyncLMDeployServer',
'HFTransformer',
'HFTransformerCasualLM',
'INTERNLM2_META',
'HFTransformerChat',
'VllmModel',
'AsyncVllmModel',
'SensenovaAPI',
'AsyncClaudeAPI',
'ClaudeAPI',
]