(Prototype) Multi-Agent AI UI Designer

lib/apitoolgen/request_consolidator.dart

@@ -0,0 +1,97 @@
+class APIDashRequestDescription {
+  final String endpoint;
+  final String method;
+  final Map<String, dynamic>? queryParams;
+  final List<Map>? formData;
+  final Map<String, dynamic>? headers;
+  final String? bodyTXT;
+  final Map? bodyJSON;
+  final String? responseType;
+  final dynamic response;
+
+  APIDashRequestDescription({
+    required this.endpoint,
+    required this.method,
+    this.queryParams,
+    this.formData,
+    this.headers,
+    this.bodyTXT,
+    this.bodyJSON,
+    this.responseType,
+    this.response,
+  });
+
+  String get generateREQDATA {
+    //Note Down the Query parameters
+    String queryParamStr = '';
+    if (queryParams != null) {
+      for (final x in queryParams!.keys) {
+        queryParamStr +=
+            '\t$x: ${queryParams![x]} <${queryParams![x].runtimeType}>\n';
+      }
+      queryParamStr = 'QUERY_PARAMETERS: {\n$queryParamStr}';
+    }
+
+    //Note Down the Headers
+    String headersStr = '';
+    if (headers != null) {
+      for (final x in headers!.keys) {
+        headersStr += '\t$x: ${headers![x]} <${headers![x].runtimeType}>\n';
+      }
+      headersStr = 'HEADERS: {\n$headersStr}';
+    }
+
+    String bodyDetails = '';
+    if (bodyTXT != null) {
+      bodyDetails = 'BODY_TYPE: TEXT\nBODY_TEXT:$bodyTXT';
+    } else if (bodyJSON != null) {
+      //Note Down the JSONData
+      String jsonBodyStr = '';
+      if (bodyJSON != null) {
+        getTyp(input, [i = 0]) {
+          String indent = "\t";
+          for (int j = 0; j < i; j++) indent += "\t";
+          if (input.runtimeType.toString().toLowerCase().contains('map')) {
+            String typd = '{';
+            for (final z in input.keys) {
+              typd += "$indent$z: TYPE: ${getTyp(input[z], i + 1)}\n";
+            }
+            return "$indent$typd}";
+          }
+          return input.runtimeType.toString();
+        }
+
+        for (final x in bodyJSON!.keys) {
+          jsonBodyStr += '\t$x: TYPE: <${getTyp(bodyJSON![x])}>\n';
+        }
+        jsonBodyStr = 'BODY_JSON: {\n$jsonBodyStr}';
+      }
+      bodyDetails = 'BODY_TYPE: JSON\n$jsonBodyStr';
+    } else if (formData != null) {
+      //Note Down the FormData
+      String formDataStr = '';
+      if (formData != null) {
+        for (final x in formData!) {
+          formDataStr += '\t$x\n';
+        }
+        formDataStr = 'BODY_FORM_DATA: {\n$formDataStr}';
+      }
+      bodyDetails = 'BODY_TYPE: FORM-DATA\n$formDataStr';
+    }
+
+    String responseDetails = '';
+    if (responseType != null && response != null) {
+      responseDetails =
+          '-----RESPONSE_DETAILS-----\nRESPONSE_TYPE: $responseType\nRESPONSE_BODY: $response';
+    }
+
+    return """REST API (HTTP)
+METHOD: $method
+ENDPOINT: $endpoint
+HEADERS: ${headersStr.isEmpty ? '{}' : headersStr}
+$queryParamStr
+$bodyDetails
+$responseDetails
+""";
+  }
+}

lib/apitoolgen/tool_templates.dart

@@ -0,0 +1,112 @@
+const GENERAL_ARG_PROPERTY_FORMAT_PY = """:ARG_NAME: {
+  "type": ":ARG_TYPE:",
+  "description: ":ARG_DESC:"
+}""";
+
+const GENERAL_PYTHON_TOOL_FORMAT = """
+:FUNC:
+
+api_tool = {
+    "function": func,
+    "definition": {
+        "name": ":TOOL_NAME:",
+        "description": ":TOOL_DESCRIPTION:",
+        "parameters": {
+            "type": "object",
+            "properties": :TOOL_PARAMS:,
+            "required": [:REQUIRED_PARAM_NAMES:],
+            "additionalProperties": False
+        }
+    }
+}
+
+__all__ = ["api_tool"]
+""";
+
+const GENERAL_JAVASCRIPT_TOOL_FORMAT = """
+:FUNC:
+
+const apiTool = {
+  function: func,
+  definition: {
+    type: 'function',
+    function: {
+      name: ':TOOL_NAME:',
+      description: ':TOOL_DESCRIPTION:',
+      parameters: {
+        type: 'object',
+        properties: :TOOL_PARAMS:,
+        required: [:REQUIRED_PARAM_NAMES:]
+        additionalProperties: false
+      }
+    }
+  }
+};
+
+export { apiTool };
+""";
+
+const LANGCHAIN_PYTHON_TOOL_FORMAT = """
+from langchain.tools import StructuredTool
+
+:INPUT_SCHEMA:
+
+:FUNC:
+
+api_tool = StructuredTool.from_function(
+    func=func,
+    name=":TOOL_NAME:",
+    description=":TOOL_DESCRIPTION:",
+    args_schema=INPUT_SCHEMA,
+)
+__all__ = ["api_tool"]
+""";
+
+const LANGCHAIN_JAVASCRIPT_TOOL_FORMAT = """
+import { DynamicStructuredTool } from 'langchain/tools';
+import { z } from 'zod';
+
+:INPUT_SCHEMA:
+
+:FUNC:
+
+const apiTool = new DynamicStructuredTool({
+  func: func,
+  name: ':TOOL_NAME:',
+  description: ':TOOL_DESCRIPTION:',
+  schema: INPUT_SCHEMA
+});
+
+export { apiTool };
+""";
+
+const MICROSOFT_AUTOGEN_TOOL_FORMAT = """
+:FUNC:
+
+api_tool = {
+    "function": func,
+    "name": ":TOOL_NAME:",
+    "description": ":TOOL_DESCRIPTION:"
+}
+
+__all__ = ["api_tool"]
+""";
+
+class APIToolGenTemplateSelector {
+  static String getTemplate(String language, String agent) {
+    if (language == 'PYTHON') {
+      if (agent == 'MICROSOFT_AUTOGEN') {
+        return MICROSOFT_AUTOGEN_TOOL_FORMAT;
+      } else if (agent == 'LANGCHAIN') {
+        return LANGCHAIN_PYTHON_TOOL_FORMAT;
+      }
+      return GENERAL_PYTHON_TOOL_FORMAT;
+    } else if (language == 'JAVASCRIPT') {
+      if (agent == 'LANGCHAIN') {
+        return LANGCHAIN_JAVASCRIPT_TOOL_FORMAT;
+      }
+      return GENERAL_JAVASCRIPT_TOOL_FORMAT;
+    }
+    return 'NO_TEMPLATE';
+  }
+}

lib/consts.dart

@@ -141,6 +141,18 @@ enum CodegenLanguage {
   final String ext;
 }
 
+enum LLMProvider {
+  ollama('Ollama', 'llama3'),
+  chatgpt('ChatGPT', '4o'),
+  gemini('Gemini', '2.0-flash'),
+  claude('Claude', '3.7'),
+  azureOpenAI('AzureOpenAI', 'custom');
+
+  const LLMProvider(this.label, this.model);
+  final String label;
+  final String model;
+}
+
 enum ImportFormat {
   curl("cURL"),
   postman("Postman Collection v2.1"),
@@ -485,3 +497,4 @@ const kMsgClearHistory =
 const kMsgClearHistorySuccess = 'History cleared successfully';
 const kMsgClearHistoryError = 'Error clearing history';
 const kMsgShareError = "Unable to share";
+const kLabelGenerateUI = "Generate UI";

lib/main.dart

@@ -6,8 +6,10 @@ import 'providers/providers.dart';
 import 'services/services.dart';
 import 'consts.dart';
 import 'app.dart';
+import 'package:stac/stac.dart' as stac;
 
 void main() async {
+  await stac.Stac.initialize();
   WidgetsFlutterBinding.ensureInitialized();
   var settingsModel = await getSettingsFromSharedPrefs();
   var onboardingStatus = await getOnboardingStatusFromSharedPrefs();

lib/models/settings_model.dart

@@ -17,6 +17,8 @@ class SettingsModel {
     this.historyRetentionPeriod = HistoryRetentionPeriod.oneWeek,
     this.workspaceFolderPath,
     this.isSSLDisabled = false,
+    this.defaultLLMProvider = LLMProvider.ollama,
+    this.defaultLLMProviderCredentials = '',
   });
 
   final bool isDark;
@@ -31,6 +33,8 @@ class SettingsModel {
   final HistoryRetentionPeriod historyRetentionPeriod;
   final String? workspaceFolderPath;
   final bool isSSLDisabled;
+  final LLMProvider defaultLLMProvider;
+  final String defaultLLMProviderCredentials;
 
   SettingsModel copyWith({
     bool? isDark,
@@ -45,6 +49,8 @@ class SettingsModel {
     HistoryRetentionPeriod? historyRetentionPeriod,
     String? workspaceFolderPath,
     bool? isSSLDisabled,
+    LLMProvider? defaultLLMProvider,
+    String? defaultLLMProviderCredentials,
   }) {
     return SettingsModel(
       isDark: isDark ?? this.isDark,
@@ -61,6 +67,9 @@ class SettingsModel {
           historyRetentionPeriod ?? this.historyRetentionPeriod,
       workspaceFolderPath: workspaceFolderPath ?? this.workspaceFolderPath,
       isSSLDisabled: isSSLDisabled ?? this.isSSLDisabled,
+      defaultLLMProvider: defaultLLMProvider ?? this.defaultLLMProvider,
+      defaultLLMProviderCredentials:
+          defaultLLMProviderCredentials ?? this.defaultLLMProviderCredentials,
     );
   }
 
@@ -80,6 +89,8 @@ class SettingsModel {
       historyRetentionPeriod: historyRetentionPeriod,
       workspaceFolderPath: workspaceFolderPath,
       isSSLDisabled: isSSLDisabled,
+      defaultLLMProvider: defaultLLMProvider,
+      defaultLLMProviderCredentials: defaultLLMProviderCredentials,
     );
   }
 
@@ -119,6 +130,28 @@ class SettingsModel {
         // pass
       }
     }
+
+    final defaultLLMProviderStr = data["defaultLLMProvider"] as String?;
+    LLMProvider? defaultLLMProvider;
+    if (defaultLLMProviderStr != null) {
+      try {
+        defaultLLMProvider = LLMProvider.values.byName(defaultLLMProviderStr);
+      } catch (e) {
+        // pass
+      }
+    }
+
+    final defaultLLMProviderCredentialsStr =
+        data['defaultLLMProviderCredentials'] as String?;
+    String? defaultLLMProviderCredentials;
+    if (defaultLLMProviderCredentialsStr != null) {
+      try {
+        defaultLLMProviderCredentials = defaultLLMProviderCredentialsStr;
+      } catch (e) {
+        // pass
+      }
+    }
+
     final saveResponses = data["saveResponses"] as bool?;
     final promptBeforeClosing = data["promptBeforeClosing"] as bool?;
     final activeEnvironmentId = data["activeEnvironmentId"] as String?;
@@ -151,6 +184,8 @@ class SettingsModel {
           historyRetentionPeriod ?? HistoryRetentionPeriod.oneWeek,
       workspaceFolderPath: workspaceFolderPath,
       isSSLDisabled: isSSLDisabled,
+      defaultLLMProvider: defaultLLMProvider,
+      defaultLLMProviderCredentials: defaultLLMProviderCredentials,
     );
   }
 
@@ -170,6 +205,8 @@ class SettingsModel {
       "historyRetentionPeriod": historyRetentionPeriod.name,
       "workspaceFolderPath": workspaceFolderPath,
       "isSSLDisabled": isSSLDisabled,
+      "defaultLLMProvider": defaultLLMProvider.name,
+      "defaultLLMProviderCredentials": defaultLLMProviderCredentials,
     };
   }
 
@@ -194,7 +231,9 @@ class SettingsModel {
         other.activeEnvironmentId == activeEnvironmentId &&
         other.historyRetentionPeriod == historyRetentionPeriod &&
         other.workspaceFolderPath == workspaceFolderPath &&
-        other.isSSLDisabled == isSSLDisabled;
+        other.isSSLDisabled == isSSLDisabled &&
+        other.defaultLLMProvider == defaultLLMProvider &&
+        other.defaultLLMProviderCredentials == defaultLLMProviderCredentials;
   }
 
   @override
@@ -213,6 +252,8 @@ class SettingsModel {
       historyRetentionPeriod,
       workspaceFolderPath,
       isSSLDisabled,
+      defaultLLMProvider,
+      defaultLLMProviderCredentials,
     );
   }
 }

lib/providers/settings_providers.dart

@@ -8,6 +8,12 @@ import '../consts.dart';
 final codegenLanguageStateProvider = StateProvider<CodegenLanguage>((ref) =>
     ref.watch(settingsProvider.select((value) => value.defaultCodeGenLang)));
 
+final llmProviderStateProvider = StateProvider<LLMProvider>((ref) =>
+    ref.watch(settingsProvider.select((value) => value.defaultLLMProvider)));
+
+final llmProviderCredentialsProvider = StateProvider<String>((ref) => ref.watch(
+    settingsProvider.select((value) => value.defaultLLMProviderCredentials)));
+
 final activeEnvironmentIdStateProvider = StateProvider<String?>((ref) =>
     ref.watch(settingsProvider.select((value) => value.activeEnvironmentId)));
 
@@ -33,6 +39,8 @@ class ThemeStateNotifier extends StateNotifier<SettingsModel> {
     HistoryRetentionPeriod? historyRetentionPeriod,
     String? workspaceFolderPath,
     bool? isSSLDisabled,
+    LLMProvider? defaultLLMProvider,
+    String? defaultLLMProviderCredentials,
   }) async {
     state = state.copyWith(
       isDark: isDark,
@@ -47,6 +55,8 @@ class ThemeStateNotifier extends StateNotifier<SettingsModel> {
       historyRetentionPeriod: historyRetentionPeriod,
       workspaceFolderPath: workspaceFolderPath,
       isSSLDisabled: isSSLDisabled,
+      defaultLLMProvider: defaultLLMProvider,
+      defaultLLMProviderCredentials: defaultLLMProviderCredentials,
     );
     await setSettingsToSharedPrefs(state);
   }