update

.env

@@ -0,0 +1,4 @@
+DEEP_SEEK_API_KEY=
+OPEN_ROUTER_API_KEY=
+TONGYI_API_KEY=
+DEBUG=False

.gitignore

@@ -0,0 +1,12 @@
+# python generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# venv
+.venv
+.idea
+.env.local

.python-version

@@ -0,0 +1 @@
+3.11.9

README.md

@@ -1,8 +1,8 @@
 ---
-title: Starship
-emoji: 📊
-colorFrom: purple
-colorTo: gray
+title: startship
+emoji: 👁
+colorFrom: indigo
+colorTo: indigo
 sdk: gradio
 sdk_version: 4.38.1
 app_file: app.py

app.py

@@ -0,0 +1,182 @@
+import gradio as gr
+from langchain_core.messages import HumanMessage, AIMessage
+from llm import DeepSeekLLM, OpenRouterLLM, TongYiLLM
+from config import settings
+
+
+deep_seek_llm = DeepSeekLLM(api_key=settings.deep_seek_api_key)
+open_router_llm = OpenRouterLLM(api_key=settings.open_router_api_key)
+tongyi_llm = TongYiLLM(api_key=settings.tongyi_api_key)
+
+
+def init_chat():
+    return deep_seek_llm.get_chat_engine()
+
+
+def predict(message, history, chat):
+    if chat is None:
+        chat = init_chat()
+    history_messages = []
+    for human, assistant in history:
+        history_messages.append(HumanMessage(content=human))
+        history_messages.append(AIMessage(content=assistant))
+    history_messages.append(HumanMessage(content=message.text))
+
+    response_message = ''
+    for chunk in chat.stream(history_messages):
+        response_message = response_message + chunk.content
+        yield response_message
+
+
+def update_chat(_provider: str, _chat, _model: str, _temperature: float, _max_tokens: int):
+    print('?????', _provider, _chat, _model, _temperature, _max_tokens)
+    if _provider == 'DeepSeek':
+        _chat = deep_seek_llm.get_chat_engine(model=_model, temperature=_temperature, max_tokens=_max_tokens)
+    if _provider == 'OpenRouter':
+        _chat = open_router_llm.get_chat_engine(model=_model, temperature=_temperature, max_tokens=_max_tokens)
+    if _provider == 'Tongyi':
+        _chat = tongyi_llm.get_chat_engine(model=_model, temperature=_temperature, max_tokens=_max_tokens)
+    return _chat
+
+
+with gr.Blocks() as app:
+    with gr.Tab('聊天'):
+        chat_engine = gr.State(value=None)
+        with gr.Row():
+            with gr.Column(scale=2, min_width=600):
+                chatbot = gr.ChatInterface(
+                    predict,
+                    multimodal=True,
+                    chatbot=gr.Chatbot(elem_id="chatbot", height=600, show_share_button=False),
+                    textbox=gr.MultimodalTextbox(lines=1),
+                    additional_inputs=[chat_engine]
+                )
+            with gr.Column(scale=1, min_width=300):
+                with gr.Accordion('参数设置', open=True):
+                    with gr.Column():
+                        provider = gr.Dropdown(
+                            label='模型厂商',
+                            choices=['DeepSeek', 'OpenRouter', 'Tongyi'],
+                            value='DeepSeek',
+                            info='不同模型厂商参数，效果和价格略有不同，请先设置好对应模型厂商的 API Key。',
+                        )
+
+                    @gr.render(inputs=provider)
+                    def show_model_config_panel(_provider):
+                        if _provider == 'DeepSeek':
+                            with gr.Column():
+                                model = gr.Dropdown(
+                                    label='模型',
+                                    choices=deep_seek_llm.support_models,
+                                    value=deep_seek_llm.default_model
+                                )
+                                temperature = gr.Slider(
+                                    minimum=0.0,
+                                    maximum=1.0,
+                                    step=0.1,
+                                    value=deep_seek_llm.default_temperature,
+                                    label="Temperature",
+                                    key="temperature",
+                                )
+                                max_tokens = gr.Slider(
+                                    minimum=1024,
+                                    maximum=1024 * 20,
+                                    step=128,
+                                    value=deep_seek_llm.default_max_tokens,
+                                    label="Max Tokens",
+                                    key="max_tokens",
+                                )
+                            model.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+                            temperature.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+                            max_tokens.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+                        if _provider == 'OpenRouter':
+                            with gr.Column():
+                                model = gr.Dropdown(
+                                    label='模型',
+                                    choices=open_router_llm.support_models,
+                                    value=open_router_llm.default_model
+                                )
+                                temperature = gr.Slider(
+                                    minimum=0.0,
+                                    maximum=1.0,
+                                    step=0.1,
+                                    value=open_router_llm.default_temperature,
+                                    label="Temperature",
+                                    key="temperature",
+                                )
+                                max_tokens = gr.Slider(
+                                    minimum=1024,
+                                    maximum=1024 * 20,
+                                    step=128,
+                                    value=open_router_llm.default_max_tokens,
+                                    label="Max Tokens",
+                                    key="max_tokens",
+                                )
+                            model.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+                            temperature.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+                            max_tokens.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+                        if _provider == 'Tongyi':
+                            with gr.Column():
+                                model = gr.Dropdown(
+                                    label='模型',
+                                    choices=tongyi_llm.support_models,
+                                    value=tongyi_llm.default_model
+                                )
+                                temperature = gr.Slider(
+                                    minimum=0.0,
+                                    maximum=1.0,
+                                    step=0.1,
+                                    value=tongyi_llm.default_temperature,
+                                    label="Temperature",
+                                    key="temperature",
+                                )
+                                max_tokens = gr.Slider(
+                                    minimum=1000,
+                                    maximum=2000,
+                                    step=100,
+                                    value=tongyi_llm.default_max_tokens,
+                                    label="Max Tokens",
+                                    key="max_tokens",
+                                )
+                            model.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+                            temperature.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+                            max_tokens.change(
+                                fn=update_chat,
+                                inputs=[provider, chat_engine, model, temperature, max_tokens],
+                                outputs=[chat_engine],
+                            )
+
+
+app.launch(debug=settings.debug, show_api=False)

config.py

@@ -0,0 +1,14 @@
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class Settings(BaseSettings):
+    deep_seek_api_key: str
+    open_router_api_key: str
+    tongyi_api_key: str
+    debug: bool
+
+    model_config = SettingsConfigDict(env_file=('.env', '.env.local'), env_file_encoding='utf-8')
+
+
+settings = Settings()
+

llm.py

@@ -0,0 +1,77 @@
+from typing import List
+from abc import ABC
+from langchain_openai import ChatOpenAI
+
+
+class BaseLLM(ABC):
+    _api_key: str
+    _support_models: List[str]
+    _default_model: str
+    _base_url: str
+    _default_temperature: float = 0.5
+    _default_max_tokens: int = 4096
+
+    def __init__(self, *, api_key: str):
+        self._api_key = api_key
+
+    @property
+    def support_models(self) -> List[str]:
+        return self._support_models
+
+    @property
+    def default_model(self) -> str:
+        return self._default_model
+
+    @property
+    def base_url(self) -> str:
+        return self._base_url
+
+    @property
+    def api_key(self) -> str:
+        return self._api_key
+
+    @property
+    def default_temperature(self) -> float:
+        return self._default_temperature
+
+    @property
+    def default_max_tokens(self) -> int:
+        return self._default_max_tokens
+
+    def get_chat_engine(self, *, model: str = None, temperature: float = None, max_tokens: int = None):
+        model = model or self.default_model
+        temperature = temperature or self.default_temperature
+        max_tokens = max_tokens or self.default_max_tokens
+        return ChatOpenAI(
+            model=model,
+            api_key=self.api_key,
+            base_url=self.base_url,
+            temperature=temperature,
+            max_tokens=max_tokens,
+        )
+
+
+class DeepSeekLLM(BaseLLM):
+    _support_models = ['deepseek-chat', 'deepseek-coder']
+    _base_url = 'https://api.deepseek.com/v1'
+    _default_model = 'deepseek-chat'
+
+
+class OpenRouterLLM(BaseLLM):
+    _support_models = [
+        'anthropic/claude-3.5-sonnet', 'openai/gpt-4o',
+        'nvidia/nemotron-4-340b-instruct', 'deepseek/deepseek-coder',
+        'google/gemini-flash-1.5', 'deepseek/deepseek-chat',
+        'liuhaotian/llava-yi-34b', 'qwen/qwen-110b-chat',
+        'qwen/qwen-72b-chat', 'google/gemini-pro-1.5',
+        'cohere/command-r-plus', 'anthropic/claude-3-haiku',
+    ]
+    _base_url = 'https://openrouter.ai/api/v1'
+    _default_model = 'anthropic/claude-3.5-sonnet'
+
+
+class TongYiLLM(BaseLLM):
+    _support_models = ['qwen-turbo', 'qwen-plus', 'qwen-max', 'qwen-long']
+    _default_model = 'qwen-turbo'
+    _base_url = 'https://dashscope.aliyuncs.com/compatible-mode/v1'
+    _default_max_tokens: int = 2000

pyproject.toml

@@ -0,0 +1,33 @@
+[project]
+name = "startship"
+version = "0.1.0"
+description = "Add your description here"
+authors = [
+    { name = "twn39", email = "[email protected]" }
+]
+dependencies = [
+    "gradio>=4.38.1",
+    "langchain>=0.2.7",
+    "pydantic>=2.8.2",
+    "pydantic-settings>=2.3.4",
+    "langchain-openai>=0.1.16",
+    "dashscope>=1.20.1",
+    "setuptools==69.5.1",
+]
+readme = "README.md"
+requires-python = ">= 3.8"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.rye]
+managed = true
+dev-dependencies = []
+
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.hatch.build.targets.wheel]
+packages = ["app.py"]
+

requirements-dev.lock

@@ -0,0 +1,276 @@
+# generated by rye
+# use `rye lock` or `rye sync` to update this lockfile
+#
+# last locked with the following flags:
+#   pre: false
+#   features: []
+#   all-features: false
+#   with-sources: false
+#   generate-hashes: false
+
+-e file:.
+aiofiles==23.2.1
+    # via gradio
+aiohttp==3.9.5
+    # via dashscope
+    # via langchain
+aiosignal==1.3.1
+    # via aiohttp
+altair==5.3.0
+    # via gradio
+annotated-types==0.7.0
+    # via pydantic
+anyio==4.4.0
+    # via httpx
+    # via openai
+    # via starlette
+    # via watchfiles
+attrs==23.2.0
+    # via aiohttp
+    # via jsonschema
+    # via referencing
+certifi==2024.7.4
+    # via httpcore
+    # via httpx
+    # via requests
+charset-normalizer==3.3.2
+    # via requests
+click==8.1.7
+    # via typer
+    # via uvicorn
+contourpy==1.2.1
+    # via matplotlib
+cycler==0.12.1
+    # via matplotlib
+dashscope==1.20.1
+    # via startship
+distro==1.9.0
+    # via openai
+dnspython==2.6.1
+    # via email-validator
+email-validator==2.2.0
+    # via fastapi
+fastapi==0.111.0
+    # via gradio
+fastapi-cli==0.0.4
+    # via fastapi
+ffmpy==0.3.2
+    # via gradio
+filelock==3.15.4
+    # via huggingface-hub
+fonttools==4.53.1
+    # via matplotlib
+frozenlist==1.4.1
+    # via aiohttp
+    # via aiosignal
+fsspec==2024.2.0
+    # via gradio-client
+    # via huggingface-hub
+gradio==4.38.1
+    # via startship
+gradio-client==1.1.0
+    # via gradio
+h11==0.14.0
+    # via httpcore
+    # via uvicorn
+httpcore==1.0.5
+    # via httpx
+httptools==0.6.1
+    # via uvicorn
+httpx==0.27.0
+    # via fastapi
+    # via gradio
+    # via gradio-client
+    # via openai
+huggingface-hub==0.23.4
+    # via gradio
+    # via gradio-client
+idna==3.7
+    # via anyio
+    # via email-validator
+    # via httpx
+    # via requests
+    # via yarl
+importlib-resources==6.4.0
+    # via gradio
+jinja2==3.1.4
+    # via altair
+    # via fastapi
+    # via gradio
+jsonpatch==1.33
+    # via langchain-core
+jsonpointer==3.0.0
+    # via jsonpatch
+jsonschema==4.23.0
+    # via altair
+jsonschema-specifications==2023.12.1
+    # via jsonschema
+kiwisolver==1.4.5
+    # via matplotlib
+langchain==0.2.7
+    # via startship
+langchain-core==0.2.18
+    # via langchain
+    # via langchain-openai
+    # via langchain-text-splitters
+langchain-openai==0.1.16
+    # via startship
+langchain-text-splitters==0.2.2
+    # via langchain
+langsmith==0.1.85
+    # via langchain
+    # via langchain-core
+markdown-it-py==3.0.0
+    # via rich
+markupsafe==2.1.5
+    # via gradio
+    # via jinja2
+matplotlib==3.9.1
+    # via gradio
+mdurl==0.1.2
+    # via markdown-it-py
+multidict==6.0.5
+    # via aiohttp
+    # via yarl
+numpy==1.26.4
+    # via altair
+    # via contourpy
+    # via gradio
+    # via langchain
+    # via matplotlib
+    # via pandas
+openai==1.35.13
+    # via langchain-openai
+orjson==3.10.6
+    # via fastapi
+    # via gradio
+    # via langsmith
+packaging==24.1
+    # via altair
+    # via gradio
+    # via gradio-client
+    # via huggingface-hub
+    # via langchain-core
+    # via matplotlib
+pandas==2.2.2
+    # via altair
+    # via gradio
+pillow==10.4.0
+    # via gradio
+    # via matplotlib
+pydantic==2.8.2
+    # via fastapi
+    # via gradio
+    # via langchain
+    # via langchain-core
+    # via langsmith
+    # via openai
+    # via pydantic-settings
+    # via startship
+pydantic-core==2.20.1
+    # via pydantic
+pydantic-settings==2.3.4
+    # via startship
+pydub==0.25.1
+    # via gradio
+pygments==2.18.0
+    # via rich
+pyparsing==3.1.2
+    # via matplotlib
+python-dateutil==2.9.0.post0
+    # via matplotlib
+    # via pandas
+python-dotenv==1.0.1
+    # via pydantic-settings
+    # via uvicorn
+python-multipart==0.0.9
+    # via fastapi
+    # via gradio
+pytz==2024.1
+    # via pandas
+pyyaml==6.0.1
+    # via gradio
+    # via huggingface-hub
+    # via langchain
+    # via langchain-core
+    # via uvicorn
+referencing==0.35.1
+    # via jsonschema
+    # via jsonschema-specifications
+regex==2024.5.15
+    # via tiktoken
+requests==2.32.3
+    # via dashscope
+    # via huggingface-hub
+    # via langchain
+    # via langsmith
+    # via tiktoken
+rich==13.7.1
+    # via typer
+rpds-py==0.19.0
+    # via jsonschema
+    # via referencing
+ruff==0.5.1
+    # via gradio
+semantic-version==2.10.0
+    # via gradio
+setuptools==69.5.1
+    # via startship
+shellingham==1.5.4
+    # via typer
+six==1.16.0
+    # via python-dateutil
+sniffio==1.3.1
+    # via anyio
+    # via httpx
+    # via openai
+sqlalchemy==2.0.31
+    # via langchain
+starlette==0.37.2
+    # via fastapi
+tenacity==8.5.0
+    # via langchain
+    # via langchain-core
+tiktoken==0.7.0
+    # via langchain-openai
+tomlkit==0.12.0
+    # via gradio
+toolz==0.12.1
+    # via altair
+tqdm==4.66.4
+    # via huggingface-hub
+    # via openai
+typer==0.12.3
+    # via fastapi-cli
+    # via gradio
+typing-extensions==4.12.2
+    # via fastapi
+    # via gradio
+    # via gradio-client
+    # via huggingface-hub
+    # via openai
+    # via pydantic
+    # via pydantic-core
+    # via sqlalchemy
+    # via typer
+tzdata==2024.1
+    # via pandas
+ujson==5.10.0
+    # via fastapi
+urllib3==2.2.2
+    # via gradio
+    # via requests
+uvicorn==0.30.1
+    # via fastapi
+    # via gradio
+uvloop==0.19.0
+    # via uvicorn
+watchfiles==0.22.0
+    # via uvicorn
+websocket-client==1.8.0
+    # via dashscope
+websockets==11.0.3
+    # via gradio-client
+    # via uvicorn
+yarl==1.9.4
+    # via aiohttp

requirements.lock

@@ -0,0 +1,276 @@
+# generated by rye
+# use `rye lock` or `rye sync` to update this lockfile
+#
+# last locked with the following flags:
+#   pre: false
+#   features: []
+#   all-features: false
+#   with-sources: false
+#   generate-hashes: false
+
+-e file:.
+aiofiles==23.2.1
+    # via gradio
+aiohttp==3.9.5
+    # via dashscope
+    # via langchain
+aiosignal==1.3.1
+    # via aiohttp
+altair==5.3.0
+    # via gradio
+annotated-types==0.7.0
+    # via pydantic
+anyio==4.4.0
+    # via httpx
+    # via openai
+    # via starlette
+    # via watchfiles
+attrs==23.2.0
+    # via aiohttp
+    # via jsonschema
+    # via referencing
+certifi==2024.7.4
+    # via httpcore
+    # via httpx
+    # via requests
+charset-normalizer==3.3.2
+    # via requests
+click==8.1.7
+    # via typer
+    # via uvicorn
+contourpy==1.2.1
+    # via matplotlib
+cycler==0.12.1
+    # via matplotlib
+dashscope==1.20.1
+    # via startship
+distro==1.9.0
+    # via openai
+dnspython==2.6.1
+    # via email-validator
+email-validator==2.2.0
+    # via fastapi
+fastapi==0.111.0
+    # via gradio
+fastapi-cli==0.0.4
+    # via fastapi
+ffmpy==0.3.2
+    # via gradio
+filelock==3.15.4
+    # via huggingface-hub
+fonttools==4.53.1
+    # via matplotlib
+frozenlist==1.4.1
+    # via aiohttp
+    # via aiosignal
+fsspec==2024.2.0
+    # via gradio-client
+    # via huggingface-hub
+gradio==4.38.1
+    # via startship
+gradio-client==1.1.0
+    # via gradio
+h11==0.14.0
+    # via httpcore
+    # via uvicorn
+httpcore==1.0.5
+    # via httpx
+httptools==0.6.1
+    # via uvicorn
+httpx==0.27.0
+    # via fastapi
+    # via gradio
+    # via gradio-client
+    # via openai
+huggingface-hub==0.23.4
+    # via gradio
+    # via gradio-client
+idna==3.7
+    # via anyio
+    # via email-validator
+    # via httpx
+    # via requests
+    # via yarl
+importlib-resources==6.4.0
+    # via gradio
+jinja2==3.1.4
+    # via altair
+    # via fastapi
+    # via gradio
+jsonpatch==1.33
+    # via langchain-core
+jsonpointer==3.0.0
+    # via jsonpatch
+jsonschema==4.23.0
+    # via altair
+jsonschema-specifications==2023.12.1
+    # via jsonschema
+kiwisolver==1.4.5
+    # via matplotlib
+langchain==0.2.7
+    # via startship
+langchain-core==0.2.18
+    # via langchain
+    # via langchain-openai
+    # via langchain-text-splitters
+langchain-openai==0.1.16
+    # via startship
+langchain-text-splitters==0.2.2
+    # via langchain
+langsmith==0.1.85
+    # via langchain
+    # via langchain-core
+markdown-it-py==3.0.0
+    # via rich
+markupsafe==2.1.5
+    # via gradio
+    # via jinja2
+matplotlib==3.9.1
+    # via gradio
+mdurl==0.1.2
+    # via markdown-it-py
+multidict==6.0.5
+    # via aiohttp
+    # via yarl
+numpy==1.26.4
+    # via altair
+    # via contourpy
+    # via gradio
+    # via langchain
+    # via matplotlib
+    # via pandas
+openai==1.35.13
+    # via langchain-openai
+orjson==3.10.6
+    # via fastapi
+    # via gradio
+    # via langsmith
+packaging==24.1
+    # via altair
+    # via gradio
+    # via gradio-client
+    # via huggingface-hub
+    # via langchain-core
+    # via matplotlib
+pandas==2.2.2
+    # via altair
+    # via gradio
+pillow==10.4.0
+    # via gradio
+    # via matplotlib
+pydantic==2.8.2
+    # via fastapi
+    # via gradio
+    # via langchain
+    # via langchain-core
+    # via langsmith
+    # via openai
+    # via pydantic-settings
+    # via startship
+pydantic-core==2.20.1
+    # via pydantic
+pydantic-settings==2.3.4
+    # via startship
+pydub==0.25.1
+    # via gradio
+pygments==2.18.0
+    # via rich
+pyparsing==3.1.2
+    # via matplotlib
+python-dateutil==2.9.0.post0
+    # via matplotlib
+    # via pandas
+python-dotenv==1.0.1
+    # via pydantic-settings
+    # via uvicorn
+python-multipart==0.0.9
+    # via fastapi
+    # via gradio
+pytz==2024.1
+    # via pandas
+pyyaml==6.0.1
+    # via gradio
+    # via huggingface-hub
+    # via langchain
+    # via langchain-core
+    # via uvicorn
+referencing==0.35.1
+    # via jsonschema
+    # via jsonschema-specifications
+regex==2024.5.15
+    # via tiktoken
+requests==2.32.3
+    # via dashscope
+    # via huggingface-hub
+    # via langchain
+    # via langsmith
+    # via tiktoken
+rich==13.7.1
+    # via typer
+rpds-py==0.19.0
+    # via jsonschema
+    # via referencing
+ruff==0.5.1
+    # via gradio
+semantic-version==2.10.0
+    # via gradio
+setuptools==69.5.1
+    # via startship
+shellingham==1.5.4
+    # via typer
+six==1.16.0
+    # via python-dateutil
+sniffio==1.3.1
+    # via anyio
+    # via httpx
+    # via openai
+sqlalchemy==2.0.31
+    # via langchain
+starlette==0.37.2
+    # via fastapi
+tenacity==8.5.0
+    # via langchain
+    # via langchain-core
+tiktoken==0.7.0
+    # via langchain-openai
+tomlkit==0.12.0
+    # via gradio
+toolz==0.12.1
+    # via altair
+tqdm==4.66.4
+    # via huggingface-hub
+    # via openai
+typer==0.12.3
+    # via fastapi-cli
+    # via gradio
+typing-extensions==4.12.2
+    # via fastapi
+    # via gradio
+    # via gradio-client
+    # via huggingface-hub
+    # via openai
+    # via pydantic
+    # via pydantic-core
+    # via sqlalchemy
+    # via typer
+tzdata==2024.1
+    # via pandas
+ujson==5.10.0
+    # via fastapi
+urllib3==2.2.2
+    # via gradio
+    # via requests
+uvicorn==0.30.1
+    # via fastapi
+    # via gradio
+uvloop==0.19.0
+    # via uvicorn
+watchfiles==0.22.0
+    # via uvicorn
+websocket-client==1.8.0
+    # via dashscope
+websockets==11.0.3
+    # via gradio-client
+    # via uvicorn
+yarl==1.9.4
+    # via aiohttp

requirements.txt

@@ -0,0 +1,504 @@
+# generated by rye
+# use `rye lock` or `rye sync` to update this lockfile
+#
+# last locked with the following flags:
+#   pre: false
+#   features: []
+#   all-features: false
+#   with-sources: false
+#   generate-hashes: false
+#   universal: false
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+absl-py==2.1.0
+    # via keras
+    # via tensorboard
+    # via tensorflow-intel
+accelerate==0.32.1
+    # via aitoolkits-webui
+addict==2.4.0
+    # via modelscope
+aiofiles==23.2.1
+    # via gradio
+aiohttp==3.9.5
+    # via dashscope
+    # via datasets
+    # via fsspec
+    # via langchain
+aiosignal==1.3.1
+    # via aiohttp
+aliyun-python-sdk-core==2.15.1
+    # via aliyun-python-sdk-kms
+    # via oss2
+aliyun-python-sdk-kms==2.16.3
+    # via oss2
+altair==5.3.0
+    # via gradio
+annotated-types==0.7.0
+    # via pydantic
+anyio==4.4.0
+    # via httpx
+    # via openai
+    # via starlette
+    # via watchfiles
+astunparse==1.6.3
+    # via tensorflow-intel
+async-timeout==4.0.3
+    # via aiohttp
+    # via langchain
+attrs==23.2.0
+    # via aiohttp
+    # via jsonschema
+    # via modelscope
+    # via referencing
+certifi==2024.7.4
+    # via httpcore
+    # via httpx
+    # via requests
+cffi==1.16.0
+    # via cryptography
+charset-normalizer==3.3.2
+    # via requests
+click==8.1.7
+    # via typer
+    # via uvicorn
+colorama==0.4.6
+    # via click
+    # via tqdm
+    # via uvicorn
+contourpy==1.2.1
+    # via matplotlib
+crcmod==1.7
+    # via oss2
+cryptography==42.0.8
+    # via aliyun-python-sdk-core
+cycler==0.12.1
+    # via matplotlib
+dashscope==1.20.1
+    # via aitoolkits-webui
+datasets==2.18.0
+    # via modelscope
+diffusers==0.29.2
+    # via aitoolkits-webui
+dill==0.3.8
+    # via datasets
+    # via multiprocess
+distro==1.9.0
+    # via openai
+dnspython==2.6.1
+    # via email-validator
+einops==0.8.0
+    # via modelscope
+email-validator==2.2.0
+    # via fastapi
+exceptiongroup==1.2.2
+    # via anyio
+fastapi==0.111.0
+    # via gradio
+fastapi-cli==0.0.4
+    # via fastapi
+ffmpy==0.3.2
+    # via gradio
+filelock==3.15.4
+    # via datasets
+    # via diffusers
+    # via huggingface-hub
+    # via torch
+    # via transformers
+flatbuffers==24.3.25
+    # via tensorflow-intel
+fonttools==4.53.1
+    # via matplotlib
+frozenlist==1.4.1
+    # via aiohttp
+    # via aiosignal
+fsspec==2024.2.0
+    # via datasets
+    # via gradio-client
+    # via huggingface-hub
+    # via torch
+gast==0.6.0
+    # via tensorflow-intel
+google-pasta==0.2.0
+    # via tensorflow-intel
+gradio==4.38.1
+    # via aitoolkits-webui
+gradio-client==1.1.0
+    # via gradio
+greenlet==3.0.3
+    # via sqlalchemy
+grpcio==1.65.1
+    # via tensorboard
+    # via tensorflow-intel
+h11==0.14.0
+    # via httpcore
+    # via uvicorn
+h5py==3.11.0
+    # via keras
+    # via tensorflow-intel
+httpcore==1.0.5
+    # via httpx
+httptools==0.6.1
+    # via uvicorn
+httpx==0.27.0
+    # via fastapi
+    # via gradio
+    # via gradio-client
+    # via openai
+huggingface-hub==0.23.4
+    # via accelerate
+    # via datasets
+    # via diffusers
+    # via gradio
+    # via gradio-client
+    # via tokenizers
+    # via transformers
+idna==3.7
+    # via anyio
+    # via email-validator
+    # via httpx
+    # via requests
+    # via yarl
+importlib-metadata==8.0.0
+    # via diffusers
+importlib-resources==6.4.0
+    # via gradio
+intel-openmp==2021.4.0
+    # via mkl
+jinja2==3.1.4
+    # via altair
+    # via fastapi
+    # via gradio
+    # via torch
+jmespath==0.10.0
+    # via aliyun-python-sdk-core
+jsonpatch==1.33
+    # via langchain-core
+jsonpointer==3.0.0
+    # via jsonpatch
+jsonschema==4.23.0
+    # via altair
+jsonschema-specifications==2023.12.1
+    # via jsonschema
+keras==3.4.1
+    # via tensorflow-intel
+kiwisolver==1.4.5
+    # via matplotlib
+kornia==0.7.3
+    # via aitoolkits-webui
+kornia-rs==0.1.5
+    # via kornia
+langchain==0.2.7
+    # via aitoolkits-webui
+langchain-core==0.2.18
+    # via langchain
+    # via langchain-openai
+    # via langchain-text-splitters
+langchain-openai==0.1.16
+    # via aitoolkits-webui
+langchain-text-splitters==0.2.2
+    # via langchain
+langsmith==0.1.85
+    # via langchain
+    # via langchain-core
+libclang==18.1.1
+    # via tensorflow-intel
+markdown==3.6
+    # via tensorboard
+markdown-it-py==3.0.0
+    # via rich
+markupsafe==2.1.5
+    # via gradio
+    # via jinja2
+    # via werkzeug
+matplotlib==3.9.1
+    # via gradio
+mdurl==0.1.2
+    # via markdown-it-py
+mkl==2021.4.0
+    # via torch
+ml-dtypes==0.4.0
+    # via keras
+    # via tensorflow-intel
+modelscope==1.16.1
+    # via aitoolkits-webui
+    # via modelscope
+mpmath==1.3.0
+    # via sympy
+multidict==6.0.5
+    # via aiohttp
+    # via yarl
+multiprocess==0.70.16
+    # via datasets
+namex==0.0.8
+    # via keras
+networkx==3.3
+    # via torch
+numpy==1.26.4
+    # via accelerate
+    # via altair
+    # via contourpy
+    # via datasets
+    # via diffusers
+    # via gradio
+    # via h5py
+    # via keras
+    # via langchain
+    # via matplotlib
+    # via ml-dtypes
+    # via opencv-python
+    # via opt-einsum
+    # via pandas
+    # via pyarrow
+    # via scipy
+    # via tensorboard
+    # via tensorflow-intel
+    # via torchvision
+    # via transformers
+    # via xformers
+openai==1.35.13
+    # via langchain-openai
+opencv-python==4.10.0.84
+    # via aitoolkits-webui
+opt-einsum==3.3.0
+    # via tensorflow-intel
+optree==0.12.1
+    # via keras
+orjson==3.10.6
+    # via fastapi
+    # via gradio
+    # via langsmith
+oss2==2.18.6
+    # via aitoolkits-webui
+    # via modelscope
+packaging==24.1
+    # via accelerate
+    # via altair
+    # via datasets
+    # via gradio
+    # via gradio-client
+    # via huggingface-hub
+    # via keras
+    # via kornia
+    # via langchain-core
+    # via matplotlib
+    # via tensorflow-intel
+    # via transformers
+pandas==2.2.2
+    # via altair
+    # via datasets
+    # via gradio
+pillow==10.4.0
+    # via diffusers
+    # via gradio
+    # via matplotlib
+    # via torchvision
+protobuf==4.25.3
+    # via tensorboard
+    # via tensorflow-intel
+psutil==6.0.0
+    # via accelerate
+pyarrow==17.0.0
+    # via datasets
+pyarrow-hotfix==0.6
+    # via datasets
+pycparser==2.22
+    # via cffi
+pycryptodome==3.20.0
+    # via oss2
+pydantic==2.8.2
+    # via aitoolkits-webui
+    # via fastapi
+    # via gradio
+    # via langchain
+    # via langchain-core
+    # via langsmith
+    # via openai
+    # via pydantic-settings
+pydantic-core==2.20.1
+    # via pydantic
+pydantic-settings==2.3.4
+    # via aitoolkits-webui
+pydub==0.25.1
+    # via gradio
+pygments==2.18.0
+    # via rich
+pyparsing==3.1.2
+    # via matplotlib
+python-dateutil==2.9.0.post0
+    # via matplotlib
+    # via modelscope
+    # via pandas
+python-dotenv==1.0.1
+    # via pydantic-settings
+    # via uvicorn
+python-multipart==0.0.9
+    # via fastapi
+    # via gradio
+pytz==2024.1
+    # via pandas
+pyyaml==6.0.1
+    # via accelerate
+    # via datasets
+    # via gradio
+    # via huggingface-hub
+    # via langchain
+    # via langchain-core
+    # via transformers
+    # via uvicorn
+referencing==0.35.1
+    # via jsonschema
+    # via jsonschema-specifications
+regex==2024.5.15
+    # via diffusers
+    # via tiktoken
+    # via transformers
+requests==2.32.3
+    # via dashscope
+    # via datasets
+    # via diffusers
+    # via huggingface-hub
+    # via langchain
+    # via langsmith
+    # via modelscope
+    # via oss2
+    # via tensorflow-intel
+    # via tiktoken
+    # via transformers
+rich==13.7.1
+    # via keras
+    # via typer
+rpds-py==0.19.0
+    # via jsonschema
+    # via referencing
+ruff==0.5.1
+    # via gradio
+safetensors==0.4.3
+    # via accelerate
+    # via diffusers
+    # via transformers
+scipy==1.14.0
+    # via modelscope
+semantic-version==2.10.0
+    # via gradio
+shellingham==1.5.4
+    # via typer
+simplejson==3.19.2
+    # via modelscope
+six==1.16.0
+    # via astunparse
+    # via google-pasta
+    # via oss2
+    # via python-dateutil
+    # via tensorboard
+    # via tensorflow-intel
+sniffio==1.3.1
+    # via anyio
+    # via httpx
+    # via openai
+sortedcontainers==2.4.0
+    # via modelscope
+sqlalchemy==2.0.31
+    # via langchain
+starlette==0.37.2
+    # via fastapi
+sympy==1.13.0
+    # via torch
+tbb==2021.13.0
+    # via mkl
+tenacity==8.5.0
+    # via langchain
+    # via langchain-core
+tensorboard==2.17.0
+    # via tensorflow-intel
+tensorboard-data-server==0.7.2
+    # via tensorboard
+tensorflow==2.17.0
+    # via aitoolkits-webui
+tensorflow-io-gcs-filesystem==0.31.0
+    # via tensorflow-intel
+termcolor==2.4.0
+    # via tensorflow-intel
+tiktoken==0.7.0
+    # via langchain-openai
+tokenizers==0.19.1
+    # via transformers
+tomlkit==0.12.0
+    # via gradio
+toolz==0.12.1
+    # via altair
+torch==2.3.1+cu121
+    # via accelerate
+    # via aitoolkits-webui
+    # via kornia
+    # via torchaudio
+    # via torchvision
+    # via xformers
+torchaudio==2.3.1+cu121
+    # via aitoolkits-webui
+torchvision==0.18.1+cu121
+    # via aitoolkits-webui
+tqdm==4.66.4
+    # via datasets
+    # via huggingface-hub
+    # via modelscope
+    # via openai
+    # via transformers
+transformers==4.42.4
+    # via aitoolkits-webui
+typer==0.12.3
+    # via fastapi-cli
+    # via gradio
+typing-extensions==4.12.2
+    # via altair
+    # via anyio
+    # via fastapi
+    # via gradio
+    # via gradio-client
+    # via huggingface-hub
+    # via openai
+    # via optree
+    # via pydantic
+    # via pydantic-core
+    # via sqlalchemy
+    # via tensorflow-intel
+    # via torch
+    # via typer
+    # via uvicorn
+tzdata==2024.1
+    # via pandas
+ujson==5.10.0
+    # via fastapi
+urllib3==2.2.2
+    # via gradio
+    # via modelscope
+    # via requests
+uvicorn==0.30.1
+    # via fastapi
+    # via gradio
+watchfiles==0.22.0
+    # via uvicorn
+websocket-client==1.8.0
+    # via dashscope
+websockets==11.0.3
+    # via gradio-client
+    # via uvicorn
+werkzeug==3.0.3
+    # via tensorboard
+wheel==0.43.0
+    # via astunparse
+wrapt==1.16.0
+    # via tensorflow-intel
+xformers==0.0.27
+    # via aitoolkits-webui
+xxhash==3.4.1
+    # via datasets
+yarl==1.9.4
+    # via aiohttp
+zipp==3.19.2
+    # via importlib-metadata
+setuptools==69.5.1
+    # via aitoolkits-webui
+    # via modelscope
+    # via tensorboard
+    # via tensorflow-intel