Plat 6359 add stylus ai assistant (#515)

Co-authored-by: Eric Lau <[email protected]>
Co-authored-by: Copilot <[email protected]>

Author: 3 people

packages/ui/api/ai-assistant/function-definitions/stylus-shared.ts

@@ -0,0 +1,22 @@
+import type { AiFunctionPropertyDefinition } from '../types/function-definition.ts';
+import type { StylusCommonContractOptions } from '../types/languages.ts';
+
+export const stylusCommonFunctionDescription = {
+  access: {
+    type: 'boolean',
+    enum: [false],
+    description: 'The type of access control to provision, currently not supported',
+    // 'The type of access control to provision. Ownable is a simple mechanism with a single account authorized for all privileged actions. Roles is a flexible mechanism with a separate role for each privileged action. A role can have many authorized accounts.',
+  },
+
+  info: {
+    type: 'object',
+    description: 'Metadata about the contract and author',
+    properties: {
+      license: {
+        type: 'string',
+        description: 'The license used by the contract, default is "MIT"',
+      },
+    },
+  },
+} as const satisfies AiFunctionPropertyDefinition<StylusCommonContractOptions>['properties'];

packages/ui/api/ai-assistant/function-definitions/stylus.ts

@@ -0,0 +1,61 @@
+import type { AiFunctionDefinition } from '../types/function-definition.ts';
+import { addFunctionPropertiesFrom } from './shared.ts';
+import { stylusCommonFunctionDescription } from './stylus-shared.ts';
+
+export const stylusERC20AIFunctionDefinition = {
+  name: 'ERC20',
+  description: 'Make a fungible token per the ERC-20 standard',
+  parameters: {
+    type: 'object',
+    properties: {
+      ...addFunctionPropertiesFrom(stylusCommonFunctionDescription, ['name', 'burnable', 'access', 'info']),
+      permit: {
+        type: 'boolean',
+        description:
+          'Whether without paying gas, token holders will be able to allow third parties to transfer from their account.',
+      },
+      flashmint: {
+        type: 'boolean',
+        description:
+          "Whether to include built-in flash loans to allow lending tokens without requiring collateral as long as they're returned in the same transaction.",
+      },
+    },
+    required: ['name'],
+    additionalProperties: false,
+  },
+} as const satisfies AiFunctionDefinition<'stylus', 'ERC20'>;
+
+export const stylusERC721AIFunctionDefinition = {
+  name: 'ERC721',
+  description: 'Make a non-fungible token per the ERC-721 standard',
+  parameters: {
+    type: 'object',
+    properties: {
+      ...addFunctionPropertiesFrom(stylusCommonFunctionDescription, ['name', 'burnable', 'access', 'info']),
+      enumerable: {
+        type: 'boolean',
+        description:
+          'Whether to allow on-chain enumeration of all tokens or those owned by an account. Increases gas cost of transfers.',
+      },
+    },
+    required: ['name'],
+    additionalProperties: false,
+  },
+} as const satisfies AiFunctionDefinition<'stylus', 'ERC721'>;
+
+export const stylusERC1155AIFunctionDefinition = {
+  name: 'ERC1155',
+  description: 'Make a non-fungible token per the ERC-1155 standard',
+  parameters: {
+    type: 'object',
+    properties: {
+      ...addFunctionPropertiesFrom(stylusCommonFunctionDescription, ['name', 'burnable', 'access', 'info']),
+      supply: {
+        type: 'boolean',
+        description: 'Whether to keep track of total supply of tokens',
+      },
+    },
+    required: ['name'],
+    additionalProperties: false,
+  },
+} as const satisfies AiFunctionDefinition<'stylus', 'ERC1155'>;

packages/ui/api/ai-assistant/types/languages.ts

@@ -13,6 +13,10 @@ export type { RoyaltyInfoOptions as CairoAlphaRoyaltyInfoOptions } from '../../.
 import type { KindedOptions as StellarKindedOptions } from '../../../../core/stellar/dist';
 import type { CommonContractOptions as StellarCommonContractOptionsBase } from '../../../../core/stellar/dist/common-options';
 export type StellarCommonContractOptions = Omit<StellarCommonContractOptionsBase, 'access'> & { access?: false };
+// Stylus
+import type { KindedOptions as StylusKindedOptions } from '../../../../core/stylus/dist';
+import type { CommonContractOptions as StylusCommonContractOptionsBase } from '../../../../core/stylus/dist/common-options';
+export type StylusCommonContractOptions = Omit<StylusCommonContractOptionsBase, 'access'> & { access?: false };
 
 // Add supported language here
 export type LanguagesContractsOptions = {
@@ -25,12 +29,18 @@ export type LanguagesContractsOptions = {
   stellar: Omit<StellarKindedOptions, 'Fungible'> & {
     Fungible: StellarKindedOptions['Fungible'] & StellarCommonContractOptions;
   };
+  stylus: Omit<StylusKindedOptions, 'ERC20' | 'ERC721' | 'ERC1155'> & {
+    ERC20: StylusKindedOptions['ERC20'] & StylusCommonContractOptions;
+    ERC721: StylusKindedOptions['ERC721'] & StylusCommonContractOptions;
+    ERC1155: StylusKindedOptions['ERC1155'] & StylusCommonContractOptions;
+  };
 };
 
 export type AllLanguagesContractsOptions = LanguagesContractsOptions['solidity'] &
   LanguagesContractsOptions['cairo'] &
   LanguagesContractsOptions['cairoAlpha'] &
-  LanguagesContractsOptions['stellar'];
+  LanguagesContractsOptions['stellar'] &
+  LanguagesContractsOptions['stylus'];
 //
 
 export type SupportedLanguage = keyof LanguagesContractsOptions;

packages/ui/api/ai.ts

@@ -3,6 +3,7 @@ import * as solidityFunctions from './ai-assistant/function-definitions/solidity
 import * as cairoFunctions from './ai-assistant/function-definitions/cairo.ts';
 import * as cairoAlphaFunctions from './ai-assistant/function-definitions/cairo-alpha.ts';
 import * as stellarFunctions from './ai-assistant/function-definitions/stellar.ts';
+import * as stylusFunctions from './ai-assistant/function-definitions/stylus.ts';
 import { saveChatInRedisIfDoesNotExist } from './services/redis.ts';
 import { getOpenAiInstance } from './services/open-ai.ts';
 import { getEnvironmentVariableOr } from './utils/env.ts';
@@ -21,6 +22,7 @@ const getFunctionsContext = <TLanguage extends SupportedLanguage = SupportedLang
     cairo: cairoFunctions,
     cairoAlpha: cairoAlphaFunctions,
     stellar: stellarFunctions,
+    stylus: stylusFunctions,
   };
 
   return Object.values(functionPerLanguages[language] ?? {});

packages/ui/src/stylus/App.svelte

@@ -27,9 +27,13 @@
   import { saveAs } from 'file-saver';
   import { injectHyperlinks } from './inject-hyperlinks';
   import type { InitialOptions } from '../common/initial-options';
+  import { createWiz, mergeAiAssistanceOptions } from '../common/Wiz.svelte';
+  import type { AiFunctionCall } from '../../api/ai-assistant/types/assistant';
 
   const dispatch = createEventDispatcher();
 
+  const WizStylus = createWiz<'stylus'>();
+
   let showCode = true;
   async function allowRendering() {
     showCode = false;
@@ -113,9 +117,21 @@
       await postConfig(opts, 'download-file', language);
     }
   };
+
+  const applyFunctionCall = ({ detail: aiFunctionCall }: CustomEvent<AiFunctionCall<'stylus'>>) => {
+    tab = sanitizeKind(aiFunctionCall.name);
+    allOpts = mergeAiAssistanceOptions(allOpts, aiFunctionCall);
+  };
 </script>
 
 <div class="container flex flex-col gap-4 p-4 rounded-3xl">
+  <WizStylus
+    language="stylus"
+    bind:currentOpts={opts}
+    bind:currentCode={code}
+    on:function-call-response={applyFunctionCall}
+    sampleMessages={['Make a token with flash loan capability', 'What is a flash loan?']}
+  />
   <div class="header flex flex-row justify-between">
     <div class="tab overflow-hidden">
       <OverflowMenu>

packages/ui/src/stylus/ERC1155Controls.svelte

@@ -8,7 +8,7 @@
   import InfoSection from './InfoSection.svelte';
   import { error } from '../common/error-tooltip';
 
-  export const opts: Required<KindedOptions['ERC1155']> = {
+  export let opts: Required<KindedOptions['ERC1155']> = {
     kind: 'ERC1155',
     ...erc1155.defaults,
     info: { ...infoDefaults }, // create new object since Info is nested

packages/ui/src/stylus/ERC20Controls.svelte

@@ -8,7 +8,7 @@
   import InfoSection from './InfoSection.svelte';
   import { error } from '../common/error-tooltip';
 
-  export const opts: Required<KindedOptions['ERC20']> = {
+  export let opts: Required<KindedOptions['ERC20']> = {
     kind: 'ERC20',
     ...erc20.defaults,
     info: { ...infoDefaults }, // create new object since Info is nested

packages/ui/src/stylus/ERC721Controls.svelte

@@ -8,7 +8,7 @@
   import InfoSection from './InfoSection.svelte';
   import { error } from '../common/error-tooltip';
 
-  export const opts: Required<KindedOptions['ERC721']> = {
+  export let opts: Required<KindedOptions['ERC721']> = {
     kind: 'ERC721',
     ...erc721.defaults,
     info: { ...infoDefaults }, // create new object since Info is nested