feat: 预支持oracle数据库

Author: jaguarliu

README.md

@@ -1,7 +1,7 @@
 ## rookie_text2data
 
 **Author:** jaguarliuu
-**Version:** 1.0.0
+**Version:** 1.0.1
 **Type:** tool
 
 ### Description

database_schema/factory.py

@@ -2,7 +2,8 @@
 from database_schema.inspectors import (
     MySQLInspector,
     SQLServerInspector,
-    PostgreSQLInspector
+    PostgreSQLInspector,
+    OracleInspector
 )
 
 class InspectorFactory:
@@ -13,7 +14,8 @@ def create_inspector(db_type: str, **kwargs) -> object:
         mapping = {
             'mysql': MySQLInspector,
             'sqlserver': SQLServerInspector,
-            'postgresql': PostgreSQLInspector
+            'postgresql': PostgreSQLInspector,
+            'oracle': OracleInspector
         }
 
         if db_type not in mapping:

database_schema/inspectors/__init__.py

@@ -2,9 +2,11 @@
 from .mysql import MySQLInspector
 from .sqlserver import SQLServerInspector
 from .postgresql import PostgreSQLInspector
+from .oracle import OracleInspector
 
 __all__ = [
     'MySQLInspector',
     'SQLServerInspector',
-    'PostgreSQLInspector'
+    'PostgreSQLInspector',
+    'OracleInspector'
 ]

manifest.yaml

@@ -1,4 +1,4 @@
-version: 1.0.0
+version: 1.0.1
 type: plugin
 author: jaguarliuu
 name: rookie_text2data

prompt_templates/sql_generation/base_prompt.jinja

@@ -24,8 +24,8 @@ T:<表名>(<字段名1>:<类型>, <字段名2>:<类型>, ...)
 ## 自定义提示：
 {{ user_custom_prompt }}
 
-## {{ db_type }}优化原则：
-{% block optimization_rules %}{% endblock %}
+{% block optimization_rules %}{% endblock %}## {{ db_type }}优化原则：
+
 
 ## 验证机制：
 {% block validation_rules %}{% endblock %}

tools/rookie_excute_sql.yaml

@@ -35,6 +35,9 @@ parameters:
       - label:
           en_US: SQL Server
         value: sqlserver
+      - label:
+          en_US: Oracle
+        value: oracle
   - name: host
     type: string
     required: true

tools/rookie_text2data.yaml

@@ -35,6 +35,9 @@ parameters:
       - label:
           en_US: SQL Server
         value: sqlserver
+      - label:
+          en_US: Oracle
+        value: oracle
   - name: limit
     type: number
     required: false
@@ -76,7 +79,7 @@ parameters:
   - name: host
     type: string
     required: true
-    form: form
+    form: llm
     label:
       en_US: Database ip/host
       zh_Hans: 数据库IP/域名
@@ -89,7 +92,7 @@ parameters:
   - name: port
     type: number
     required: true
-    form: form
+    form: llm
     min: 1
     max: 65535
     label:
@@ -104,7 +107,7 @@ parameters:
   - name: db_name
     type: string
     required: true
-    form: form
+    form: llm
     label:
       en_US: Database name
       zh_Hans: 数据库名称
@@ -143,7 +146,7 @@ parameters:
   - name: username
     type: string
     required: true
-    form: form
+    form: llm
     label:
       en_US: Username
       zh_Hans: 用户名
@@ -156,7 +159,7 @@ parameters:
   - name: password
     type: secret-input
     required: true
-    form: form
+    form: llm
     label:
       en_US: Password
       zh_Hans: 密码