init: 初始化项目

Author: bigcyy

.gitignore

@@ -0,0 +1,102 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+pip-wheel-metadata/
+*.egg-info/
+*.egg
+dist/
+build/
+
+# Virtual environments
+# Common virtual environment folders
+.venv/
+.venv.bak/
+ENV/
+env/
+env.bak/
+venv/
+venv.bak/
+
+# Flask specific
+instance/
+.webassets-cache
+
+# Pytest
+.pytest_cache/
+.coverage
+.tox/
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# Environments
+.env
+.env.*
+*.env
+
+# PyCharm
+.idea/
+
+# VS Code
+.vscode/
+
+# Sublime Text
+*.sublime-project
+*.sublime-workspace
+
+# MacOS
+.DS_Store
+
+# Windows
+Thumbs.db
+ehthumbs.db
+Desktop.ini
+
+# Other IDEs and editors
+*.swp
+*~
+
+# Logs and databases
+*.log
+*.sqlite3
+
+# Test output
+htmlcov/
+
+# MyPy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Docker
+*.dockerignore
+docker-compose.yml
+Dockerfile
+__pycache__/
+
+# pylint
+pylint-report.txt
+pylint-global.rc
+
+# Coverage reports
+htmlcov/
+.coverage
+.coverage.*
+.cache
+.nox/

README.md

@@ -0,0 +1,18 @@
+## Chat2SQLGraph
+
+**Chat2SQLGraph** 是一个基于对话的智能数据分析和可视化工具。通过自然语言对话，用户可以生成 SQL 查询、执行数据检索，并自动将查询结果转化为可视化图表。该项目利用大语言模型（LLM）实现自然语言到 SQL 的翻译，并将 SQL 查询结果直观地展现在图表中，极大地方便了用户的数据分析过程。
+
+### 主要功能：
+
+- **自然语言转 SQL 查询**：通过与系统的对话，生成复杂的 SQL 查询语句。
+- **智能数据检索**：执行生成的 SQL 查询，获取数据库中的数据。
+- **自动数据可视化**：将 SQL 查询结果转化为图表，如柱状图、饼图、折线图等，帮助用户更好地理解数据。
+- **易于集成和扩展**：支持不同的数据库系统和图表库，适用于各种数据分析场景。
+
+### 应用场景：
+
+- 数据分析师快速生成和执行 SQL 查询并可视化结果。
+- 产品经理和业务分析师通过自然语言探索数据洞察。
+- 数据科学团队协作分析和共享可视化结果。
+
+**Chat2SQLGraph** ,欢迎贡献和反馈！

backend/chat/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

backend/chat/admin.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ChatConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "chat"

backend/chat/apps.py

@@ -0,0 +1,65 @@
+# Generated by Django 5.1.1 on 2024-09-26 12:01
+
+import django.db.models.deletion
+import uuid
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ("setting", "0005_alter_datasource_url"),
+        ("user", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="ChatInfo",
+            fields=[
+                (
+                    "id",
+                    models.UUIDField(
+                        default=uuid.uuid1,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="会话 id",
+                    ),
+                ),
+                ("user_demand", models.TextField(verbose_name="用户需求")),
+                ("chat_content", models.JSONField(verbose_name="会话内容")),
+                (
+                    "created_at",
+                    models.DateTimeField(auto_now_add=True, verbose_name="创建时间"),
+                ),
+                (
+                    "updated_at",
+                    models.DateTimeField(auto_now=True, verbose_name="更新时间"),
+                ),
+                (
+                    "datasource_id",
+                    models.ForeignKey(
+                        db_column="datasource_id",
+                        on_delete=django.db.models.deletion.CASCADE,
+                        to="setting.datasource",
+                        verbose_name="数据源 id",
+                    ),
+                ),
+                (
+                    "user_id",
+                    models.ForeignKey(
+                        db_column="user_id",
+                        on_delete=django.db.models.deletion.CASCADE,
+                        to="user.user",
+                        verbose_name="用户 id",
+                    ),
+                ),
+            ],
+            options={
+                "verbose_name": "会话信息",
+                "verbose_name_plural": "会话信息",
+                "db_table": "chat_info",
+            },
+        ),
+    ]

backend/chat/migrations/0001_initial.py

@@ -0,0 +1,23 @@
+# Generated by Django 5.1.1 on 2024-09-26 12:13
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("chat", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="chatinfo",
+            name="chat_content",
+            field=models.JSONField(blank=True, null=True, verbose_name="会话内容"),
+        ),
+        migrations.AlterField(
+            model_name="chatinfo",
+            name="user_demand",
+            field=models.TextField(blank=True, null=True, verbose_name="用户需求"),
+        ),
+    ]

backend/chat/migrations/0002_alter_chatinfo_chat_content_and_more.py

@@ -0,0 +1,21 @@
+from django.db import models
+from setting.models import Datasource
+import uuid
+from user.models import User
+
+# Create your models here.
+
+class ChatInfo(models.Model):
+    id = models.UUIDField(verbose_name="会话 id",max_length=32,primary_key=True,default=uuid.uuid1)
+    datasource_id = models.ForeignKey(Datasource,verbose_name="数据源 id",on_delete=models.CASCADE,db_column="datasource_id")
+    user_id = models.ForeignKey(User,verbose_name="用户 id",on_delete=models.CASCADE,db_column="user_id")
+    user_demand = models.TextField(blank = True, null = True, verbose_name = "用户需求")
+    chat_content = models.JSONField (verbose_name="会话内容",blank = True, null = True)
+
+    created_at = models.DateTimeField(verbose_name="创建时间",auto_now_add=True)
+    updated_at = models.DateTimeField(verbose_name="更新时间",auto_now=True)
+
+    class Meta:
+        db_table = "chat_info"
+        verbose_name = "会话信息"
+        verbose_name_plural = "会话信息"