Mac 上安转文字转 SQL 利器 WrenAI

发布于：2024-07-11 ⋅ 阅读:(154) ⋅ 点赞:(0)

WrenAI 是一个开源的 Text-SQL 的工具，通过导入数据库结构，通过提问的方式生成 SQL。本文将讲述如何在 MacOS 上安装 WrenAI。要运行WrenAI，首先需要安装 Docker 桌面版。

下载 WrenAI

https://github.com/Canner/WrenAI/releases/tag/0.7.0

在这里插入图片描述

初始化配置文件

wget -O .env.ai.example https://raw.githubusercontent.com/canner/WrenAI/main/docker/.env.ai.example && \
mkdir -p ~/.wrenai && cp .env.ai.example ~/.wrenai/.env.ai

默认系统使用的是 OpenAI 模型，改为使用 Ollama 模型

## LLM
 LLM_PROVIDER=ollama_llm # openai_llm, azure_openai_llm, ollama_llm
 GENERATION_MODEL=qwen2:7b
 GENERATION_MODEL_KWARGS={"temperature": 0}

 # openai or openai-api-compatible
 LLM_OPENAI_API_KEY=sk-xxxx
 LLM_OPENAI_API_BASE=https://api.openai.com/v1

 # azure_openai
 LLM_AZURE_OPENAI_API_KEY=
 LLM_AZURE_OPENAI_API_BASE=
 LLM_AZURE_OPENAI_VERSION=

 # ollama
 LLM_OLLAMA_URL=http://host.docker.internal:11434


 ## EMBEDDER
 EMBEDDER_PROVIDER=ollama_embedder # openai_embedder, azure_openai_embedder, ollama_embedder
 # supported embedding models providers by qdrant: https://qdrant.tech/documentation/embeddings/
 EMBEDDING_MODEL=nomic-embed-text
 EMBEDDING_MODEL_DIMENSION=768

 # openai or openai-api-compatible
 EMBEDDER_OPENAI_API_KEY=sk-xxxx
 EMBEDDER_OPENAI_API_BASE=https://api.openai.com/v1

 # azure_openai
 EMBEDDER_AZURE_OPENAI_API_KEY=
 EMBEDDER_AZURE_OPENAI_API_BASE=
 EMBEDDER_AZURE_OPENAI_VERSION=

 # ollama
 EMBEDDER_OLLAMA_URL=http://host.docker.internal:11434


 ## DOCUMENT_STORE
 DOCUMENT_STORE_PROVIDER=qdrant

 QDRANT_HOST=qdrant