Asistente de base de datos Ollama MCP
Una interfaz de chat interactiva que combina las capacidades LLM de Ollama con el acceso a bases de datos PostgreSQL mediante el Protocolo de Contexto de Modelo (MCP). Pregunte sobre sus datos en lenguaje natural y obtenga respuestas basadas en IA, respaldadas por consultas SQL reales.
Características
- Interfaz de lenguaje natural para su base de datos PostgreSQL
- Generación automática de consultas SQL
- Respuestas que tienen en cuenta el esquema
- Interfaz de chat interactiva
- Acceso seguro y de solo lectura a la base de datos
Prerrequisitos
- Node.js 16 o superior
- Una base de datos PostgreSQL en ejecución
- Ollama instalado y ejecutándose localmente
- El modelo qwen2.5-coder:7b-instruct incluido en Ollama
Configuración
- Clonar el repositorio:
git clone [your-repo-url]
cd [your-repo-name]
- Instalar dependencias:
- Extraiga el modelo Ollama requerido:
ollama pull qwen2.5-coder:7b-instruct
- Cree un archivo
.env en la raíz del proyecto:
DATABASE_URL=postgresql://user:password@localhost:5432/dbname
OLLAMA_MODEL=qwen2.5-coder:7b-instruct # Optional - this is the default
Uso
- Iniciar la interfaz de chat:
- Haga preguntas sobre sus datos en lenguaje natural:
Connected to database. You can now ask questions about your data.
Type "exit" to quit.
What would you like to know about your data? Which products generated the most revenue last month?
Analyzing...
[AI will generate and execute a SQL query, then explain the results]
- Escriba ‘exit’ para salir de la aplicación.
Cómo funciona
- La aplicación se conecta a su base de datos PostgreSQL a través del servidor PostgreSQL MCP
- Carga y almacena en caché el esquema de su base de datos.
- Cuando haces una pregunta:
- El esquema y la pregunta se envían a Ollama
- Ollama genera una consulta SQL apropiada
- La consulta se ejecuta a través de MCP
- Los resultados se envían a Ollama para su interpretación.
- Recibirás una respuesta en lenguaje natural
Variables de entorno
| Variable | Descripción | Por defecto |
|---|
| URL DE LA BASE DE DATOS | Cadena de conexión de PostgreSQL | Requerido |
| OLLAMA_MODELO | Modelo Ollama a utilizar | qwen2.5-coder:7b-instruct |
Seguridad
- Todo acceso a la base de datos es de sólo lectura.
- Las consultas SQL están restringidas a sentencias SELECT
- Las credenciales de la base de datos se mantienen seguras en su archivo .env
Desarrollo
Construido con:
- Mecanografiado
- Protocolo de Contexto Modelo (MCP)
- Ollama
- PostgreSQL
Solución de problemas
Problemas comunes
- "Error al conectar con la base de datos"
- Comprueba tu DATABASE_URL en .env
- Verificar que PostgreSQL se esté ejecutando
- Comprobar la conectividad de la red
- "Error al conectar con Ollama"
- Asegúrese de que Ollama se esté ejecutando (
ollama serve ) - Verificar que el modelo esté instalado (
ollama list )
- "Error al ejecutar la consulta"
- Comprobar los permisos de la base de datos
- Verificar los nombres de tablas/columnas en el esquema
Licencia
Instituto Tecnológico de Massachusetts (MIT)
Contribuyendo
- Bifurcar el repositorio
- Crea tu rama de funciones
- Confirme sus cambios
- Empujar hacia la rama
- Abrir una solicitud de extracción