├── .github └── workflows │ └── tests.yml ├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── config └── example.yaml ├── generation_config ├── internlm2 │ └── generation_config.json └── qwen │ └── generation_config.json ├── pyproject.toml ├── requirements.txt ├── setup.py ├── src └── imitater │ ├── __init__.py │ ├── agent │ ├── __init__.py │ ├── aligned.py │ ├── function.py │ ├── react.py │ ├── registry.py │ └── types.py │ ├── model │ ├── __init__.py │ ├── chat_model.py │ └── embed_model.py │ ├── service │ ├── __init__.py │ ├── app.py │ ├── chat.py │ ├── common.py │ ├── embed.py │ └── protocol.py │ └── utils │ ├── __init__.py │ ├── generic.py │ └── modelscope.py ├── templates ├── baichuan.jinja ├── default.jinja ├── internlm2.jinja ├── qwen.jinja └── yi.jinja └── tests └── test_openai.py /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/README.md -------------------------------------------------------------------------------- /config/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/config/example.yaml -------------------------------------------------------------------------------- /generation_config/internlm2/generation_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/generation_config/internlm2/generation_config.json -------------------------------------------------------------------------------- /generation_config/qwen/generation_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/generation_config/qwen/generation_config.json -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/setup.py -------------------------------------------------------------------------------- /src/imitater/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.2.5.dev0" 2 | -------------------------------------------------------------------------------- /src/imitater/agent/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/agent/__init__.py -------------------------------------------------------------------------------- /src/imitater/agent/aligned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/agent/aligned.py -------------------------------------------------------------------------------- /src/imitater/agent/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/agent/function.py -------------------------------------------------------------------------------- /src/imitater/agent/react.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/agent/react.py -------------------------------------------------------------------------------- /src/imitater/agent/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/agent/registry.py -------------------------------------------------------------------------------- /src/imitater/agent/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/agent/types.py -------------------------------------------------------------------------------- /src/imitater/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/model/__init__.py -------------------------------------------------------------------------------- /src/imitater/model/chat_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/model/chat_model.py -------------------------------------------------------------------------------- /src/imitater/model/embed_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/model/embed_model.py -------------------------------------------------------------------------------- /src/imitater/service/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/imitater/service/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/service/app.py -------------------------------------------------------------------------------- /src/imitater/service/chat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/service/chat.py -------------------------------------------------------------------------------- /src/imitater/service/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/service/common.py -------------------------------------------------------------------------------- /src/imitater/service/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/service/embed.py -------------------------------------------------------------------------------- /src/imitater/service/protocol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/service/protocol.py -------------------------------------------------------------------------------- /src/imitater/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/imitater/utils/generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/utils/generic.py -------------------------------------------------------------------------------- /src/imitater/utils/modelscope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/src/imitater/utils/modelscope.py -------------------------------------------------------------------------------- /templates/baichuan.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/templates/baichuan.jinja -------------------------------------------------------------------------------- /templates/default.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/templates/default.jinja -------------------------------------------------------------------------------- /templates/internlm2.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/templates/internlm2.jinja -------------------------------------------------------------------------------- /templates/qwen.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/templates/qwen.jinja -------------------------------------------------------------------------------- /templates/yi.jinja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/templates/yi.jinja -------------------------------------------------------------------------------- /tests/test_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/the-seeds/imitater/HEAD/tests/test_openai.py --------------------------------------------------------------------------------