feat: Added AI extension support and SQL generation APIs

console/atest-ui/src/views/store.ts

@@ -38,6 +38,7 @@
 const ExtensionKindMongoDB = "atest-store-mongodb"
 const ExtensionKindElasticsearch = "atest-store-elasticsearch"
 const ExtensionKindOpengeMini = "atest-store-opengemini"
+const ExtensionKindAI = "atest-ext-ai"
 
 export const ExtensionKind = {
     ExtensionKindGit,
@@ -49,7 +50,8 @@
     ExtensionKindRedis,
     ExtensionKindMongoDB,
     ExtensionKindElasticsearch,
-    ExtensionKindOpengeMini
+    ExtensionKindOpengeMini,
+    ExtensionKindAI
 }
 
 const MySQL = "mysql";
@@ -61,13 +63,151 @@
 export const GetDriverName = (store: Store): string => {
     switch (store.kind.name) {
     case 'atest-store-orm':
-        return store.properties.find((p: Pair) => p.key === 'driver')?.value || MySQL;
+        const properties = store.properties as Pair[];
+        for (let i = 0; i < properties.length; i++) {
+            if (properties[i].key === 'driver') {
+                return properties[i].value || MySQL;
+            }
+        }
+        return MySQL;
     case 'atest-store-cassandra':
         return Cassandra;
     }
     return ""
 }
 
+const storeExtensions = [
+    {
+        name: ExtensionKindGit,
+        params: [{
+            key: 'insecure'
+        }, {
+            key: 'timeout'
+        }, {
+            key: 'targetpath'
+        }, {
+            key: 'branch'
+        }, {
+            key: 'email',
+            description: 'See also: git config --local user.email [email protected]'
+        }, {
+            key: 'name',
+            description: 'See also: git config --local user.name xxx'
+        }],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-git'
+    },
+    {
+        name: ExtensionKindS3,
+        params: [{
+            key: 'accesskeyid'
+        }, {
+            key: 'secretaccesskey'
+        }, {
+            key: 'sessiontoken'
+        }, {
+            key: 'region'
+        }, {
+            key: 'disablessl'
+        }, {
+            key: 'forcepathstyle'
+        }, {
+            key: 'bucket'
+        }],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-s3'
+    },
+    {
+        name: ExtensionKindORM,
+        params: [{
+            key: 'driver',
+            defaultValue: 'mysql',
+            enum: [MySQL, Postgres, SQLite, TDengine],
+            description: 'Supported: mysql, postgres, sqlite, tdengine'
+        }, {
+            key: 'database',
+            defaultValue: 'atest'
+        }, {
+            key: 'historyLimit',
+            defaultValue: '',
+            // type: 'number',
+            description: 'Set the limit of the history record count'
+        }],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-orm'
+    },
+    {
+        name: ExtensionKindIotDB,
+        params: [],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-iotdb'
+    },
+    {
+        name: ExtensionKindCassandra,
+        params: [{
+            key: 'keyspace',
+            defaultValue: ''
+        }],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-cassandra'
+    },
+    {
+        name: ExtensionKindEtcd,
+        params: [],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-etcd'
+    },
+    {
+        name: ExtensionKindRedis,
+        params: [],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-redis'
+    },
+    {
+        name: ExtensionKindMongoDB,
+        params: [{
+            key: 'collection'
+        }, {
+            key: 'database',
+            defaultValue: 'atest'
+        }],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-mongodb'
+    },
+    {
+        name: ExtensionKindElasticsearch,
+        params: [],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-elasticsearch'
+    },
+    {
+        name: ExtensionKindOpengeMini,
+        params: [],
+        link: 'https://github.com/LinuxSuRen/atest-ext-store-opengemini'
+    },
+    {
+        name: ExtensionKindAI,
+        params: [{
+            key: 'model_provider',
+            defaultValue: 'ollama',
+            enum: ['ollama', 'openai'],
+            description: 'AI model provider used in GenerateContentRequest.parameters for content generation'
+        }, {
+            key: 'model_name',
+            defaultValue: 'llama3.2',
+            description: 'AI model name passed to GenerateContentRequest.parameters["model_name"] for specific model selection'
+        }, {
+            key: 'api_key',
+            defaultValue: '',
+            description: 'API key for external providers, used in GenerateContentRequest.parameters["api_key"] for authentication'
+        }, {
+            key: 'base_url',
+            defaultValue: 'http://localhost:11434',
+            description: 'Base URL for AI service, used in GenerateContentRequest.parameters["base_url"] to specify service endpoint'
+        }, {
+            key: 'temperature',
+            defaultValue: '0.7',
+            description: 'Controls randomness in AI responses (0.0-1.0), passed via GenerateContentRequest.parameters["temperature"]'
+        }, {
+            key: 'max_tokens',
+            defaultValue: '2048',
+            description: 'Maximum tokens in AI response, used in GenerateContentRequest.parameters["max_tokens"] to limit output length'
+        }],
+        link: 'https://github.com/LinuxSuRen/atest-ext-ai'
+    }
+]
+
 export const Driver = {
     MySQL, Postgres, SQLite, TDengine, Cassandra
 }

extensions/README.md

@@ -15,6 +15,157 @@
 | Agent | [collector](https://github.com/LinuxSuRen/atest-ext-collector)           |  |
 | Secret | [Vault](https://github.com/LinuxSuRen/api-testing-vault-extension)       | |
 | Data | [Swagger](https://github.com/LinuxSuRen/atest-ext-data-swagger) | |
+| AI | [ai-extension](https://github.com/LinuxSuRen/atest-ext-ai) | 50051 |
+
+## AI Extension Interface
+
+This extension provides AI-powered content generation capabilities through a unified gRPC interface.
+
+### Unified Interface Design
+
+#### `GenerateContent`
+The single, unified interface for all AI content generation tasks. This interface can handle various content types including SQL generation, test case writing, mock service creation, and more through a single endpoint.
+
+**Request Parameters:**
+- `prompt`: Natural language description of what you want to generate
+- `contentType`: Type of content to generate (e.g., "sql", "testcase", "mock")
+- `context`: Additional context information as key-value pairs
+- `parameters`: Additional parameters specific to the content type
+
+**Response:**
+- `success.content`: Generated content
+- `success.explanation`: Explanation of the generated content
+- `success.confidenceScore`: Confidence score (0.0 to 1.0)
+- `success.metadata`: Additional metadata about the generation
+- `error`: Error information if generation fails
+
+### Content Types and Task Identification
+
+The `contentType` parameter serves as the task identifier, allowing the system to:
+
+1. **Apply appropriate prompting strategies** for different content types
+2. **Use specialized processing logic** for each task type
+3. **Provide task-specific validation and post-processing**
+4. **Generate relevant metadata** for each content type
+
+#### Supported Content Types:
+
+1. **SQL Generation** (`contentType: "sql"`)
+   - Generates SQL queries from natural language
+   - Supports multiple database types via parameters
+   - Context can include schema information
+   - Applies SQL-specific validation and dialect adaptation
+
+2. **Test Case Generation** (`contentType: "testcase"`)
+   - Generates comprehensive test cases
+   - Context can include code specifications
+   - Uses test-specific prompting strategies
+
+3. **Mock Service Generation** (`contentType: "mock"`)
+   - Generates mock services and API responses
+   - Context can include API specifications
+   - Applies service-specific formatting
+
+4. **Generic Content** (any other `contentType`)
+   - Handles any other content generation requests
+   - Flexible prompt-based generation
+   - Extensible for future content types
+
+### Architecture Benefits
+
+**Single Interface Advantages:**
+- **Consistency**: All content generation follows the same request/response pattern
+- **Extensibility**: New content types can be added without API changes
+- **Simplicity**: Clients only need to integrate with one interface
+- **Flexibility**: Rich context and parameter support for all content types
+
+**Task Differentiation:**
+While using a single interface, the system internally routes requests to specialized handlers based on `contentType`, ensuring:
+- Appropriate AI model prompting for each task
+- Task-specific validation and processing
+- Relevant metadata and confidence scoring
+- Optimized performance for different content types
+
+### Configuration Parameters
+
+The following parameters can be passed via `GenerateContentRequest.parameters`:
+
+- `model_provider`: AI provider ("ollama", "openai") - used in GenerateContentRequest.parameters
+- `model_name`: Specific model name (e.g., "llama3.2", "gpt-4") - passed via parameters["model_name"]
+- `api_key`: Authentication key for external providers - used in parameters["api_key"]
+- `base_url`: Service endpoint URL - specified in parameters["base_url"]
+- `temperature`: Response randomness control (0.0-1.0) - set via parameters["temperature"]
+- `max_tokens`: Maximum response length - controlled by parameters["max_tokens"]
+
+### Usage Examples
+
+```protobuf
+// Generate SQL query
+GenerateContentRequest {
+  prompt: "Show me all users from California"
+  contentType: "sql"
+  context: {
+    "schema": "CREATE TABLE users (id INT, name VARCHAR(100), state VARCHAR(50))"
+  }
+  parameters: {
+    "model_provider": "ollama"
+    "model_name": "llama3.2"
+    "database_type": "mysql"
+  }
+}
+
+// Generate test case
+GenerateContentRequest {
+  prompt: "Create a test case for user registration API"
+  contentType: "testcase"
+  context: {
+    "api_spec": "POST /api/users {name, email, password}"
+  }
+}
+
+// Generate mock service
+GenerateContentRequest {
+  prompt: "Create a mock REST API for user management"
+  contentType: "mock"
+  context: {
+    "endpoints": "GET /users, POST /users, PUT /users/{id}, DELETE /users/{id}"
+    "response_format": "JSON"
+  }
+}
+```
+
+### Migration Notes
+
+This version removes the legacy `GenerateSQLFromNaturalLanguage` interface in favor of the unified `GenerateContent` approach. All SQL generation should now use:
+
+```protobuf
+GenerateContentRequest {
+  prompt: "your natural language query"
+  contentType: "sql"
+  parameters: {"database_type": "mysql|postgresql|sqlite"}
+  context: {"schemas": "table definitions"}
+}
+```
+
+### Frequently Asked Questions
+
+**Q1: Do I need different interfaces for different tasks (SQL generation, test case writing, mock service creation)?**
+
+A: No, you only need the single `GenerateContent` interface. The `contentType` parameter tells the system what kind of content you want to generate, and the system internally routes your request to the appropriate specialized handler. This design provides:
+- **Task identification**: The system knows exactly what you're trying to accomplish
+- **Specialized processing**: Each content type gets optimized prompting and validation
+- **Consistent API**: You always use the same interface regardless of the task
+- **Future extensibility**: New content types can be added without changing the API
+
+**Q2: Will removing the SQL-specific interface affect other files?**
+
+A: The removal has been carefully implemented to ensure minimal impact:
+- **Protobuf files**: All generated code (Go, Python, gRPC-gateway, OpenAPI) has been updated
+- **Implementation files**: The Python gRPC server has been updated to use the new unified interface
+- **Backward compatibility**: While the old interface is removed, the same functionality is available through `GenerateContent` with `contentType: "sql"`
+- **No breaking changes**: The core functionality remains the same, just accessed through a cleaner, unified interface
+
+The migration ensures that all SQL generation capabilities are preserved while providing a more maintainable and extensible architecture.
 
 ## Contribute a new extension