From 37bd5db401ce891a56449ec4798a9744ad69db2c Mon Sep 17 00:00:00 2001 From: Liu Date: Fri, 27 Dec 2024 07:32:20 +0000 Subject: [PATCH 1/2] docs: update the LLM / Embedding models for Gitee AI (#549) --- backend/app/rag/embed_model_option.py | 2 +- frontend/app/src/pages/docs/embedding-model.mdx | 13 +++++++++++++ frontend/app/src/pages/docs/llm.mdx | 1 + 3 files changed, 15 insertions(+), 1 deletion(-) diff --git a/backend/app/rag/embed_model_option.py b/backend/app/rag/embed_model_option.py index 9ab712ff..72659ae6 100644 --- a/backend/app/rag/embed_model_option.py +++ b/backend/app/rag/embed_model_option.py @@ -91,7 +91,7 @@ class EmbeddingModelOption(BaseModel): provider_description="Gitee AI is a third-party model provider that offers ready-to-use cutting-edge model APIs for AI developers.", provider_url="https://ai.gitee.com", default_embedding_model="bge-large-zh-v1.5", - embedding_model_description=f"Find more information about Gitee AI Embeddings at https://ai.gitee.com/docs/openapi/v1#tag/serverless/POST/embeddings", + embedding_model_description=f"Find more information about Gitee AI Embeddings at https://ai.gitee.com/docs/openapi/v1#tag/%E7%89%B9%E5%BE%81%E6%8A%BD%E5%8F%96/POST/embeddings", credentials_display_name="Gitee AI API Key", credentials_description="The API key of Gitee AI, you can find it in https://ai.gitee.com/dashboard/settings/tokens", credentials_type="str", diff --git a/frontend/app/src/pages/docs/embedding-model.mdx b/frontend/app/src/pages/docs/embedding-model.mdx index 4b84cd6e..e483e76f 100644 --- a/frontend/app/src/pages/docs/embedding-model.mdx +++ b/frontend/app/src/pages/docs/embedding-model.mdx @@ -130,6 +130,19 @@ To use Ollama, you'll need to configure the API base URL in the **Advanced Setti For more information, see the [Ollama embedding models documentation](https://ollama.com/blog/embedding-models). +### Gitee AI + +Gitee AI is a third-party model provider that offers ready-to-use cutting-edge model APIs for AI developers. + +**Supported Models**: + +| Embedding Model | Vector Dimensions | Max Tokens | +| `bge-m3` | 1024 | 8192 | +| `bge-large-zh-v1.5` | 1024 | 512 | +| `bge-small-zh-v1.5` | 512 | 512 | + +For more information, see the [Gitee AI embedding models documentation](https://ai.gitee.com/docs/openapi/v1#tag/%E7%89%B9%E5%BE%81%E6%8A%BD%E5%8F%96/POST/embeddings). + ### Local Embedding Server Autoflow's local embedding server is a self-hosted embedding service built upon [sentence-transformers](https://www.sentence-transformers.org/) and deployed on your own infrastructure. diff --git a/frontend/app/src/pages/docs/llm.mdx b/frontend/app/src/pages/docs/llm.mdx index 3e1a31b5..e0ad9d6b 100644 --- a/frontend/app/src/pages/docs/llm.mdx +++ b/frontend/app/src/pages/docs/llm.mdx @@ -30,6 +30,7 @@ Currently Autoflow supports the following LLM providers: - [Google Gemini](https://gemini.google.com/) - [Anthropic Vertex AI](https://cloud.google.com/vertex-ai/generative-ai/docs/partner-models/use-claude) - [Amazon Bedrock](https://aws.amazon.com/bedrock/) +- [Gitee AI](https://ai.gitee.com/serverless-api) - And all OpenAI-Like models: - [OpenRouter](https://openrouter.ai/) - Default config: From aa0eb2d8c6c61a6a612c64e805d91775afb2c03b Mon Sep 17 00:00:00 2001 From: Mini256 Date: Sun, 29 Dec 2024 20:02:07 +0800 Subject: [PATCH 2/2] format --- frontend/app/src/pages/docs/embedding-model.mdx | 1 + 1 file changed, 1 insertion(+) diff --git a/frontend/app/src/pages/docs/embedding-model.mdx b/frontend/app/src/pages/docs/embedding-model.mdx index e483e76f..42a2de52 100644 --- a/frontend/app/src/pages/docs/embedding-model.mdx +++ b/frontend/app/src/pages/docs/embedding-model.mdx @@ -137,6 +137,7 @@ Gitee AI is a third-party model provider that offers ready-to-use cutting-edge m **Supported Models**: | Embedding Model | Vector Dimensions | Max Tokens | +| ------------------- | ----------------- | ---------- | | `bge-m3` | 1024 | 8192 | | `bge-large-zh-v1.5` | 1024 | 512 | | `bge-small-zh-v1.5` | 512 | 512 |