Plat 6358 add stellar ai assistant (#516)

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

Author: 3 people

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

@@ -0,0 +1,22 @@
+import type { AiFunctionPropertyDefinition } from '../types/function-definition.ts';
+import type { StellarCommonContractOptions } from '../types/languages.ts';
+
+export const stellarCommonFunctionDescription = {
+  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.',
+  },
+
+  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<StellarCommonContractOptions>['properties'];

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

@@ -0,0 +1,28 @@
+import type { AiFunctionDefinition } from '../types/function-definition.ts';
+import { addFunctionPropertiesFrom } from './shared.ts';
+import { stellarCommonFunctionDescription } from './stellar-shared.ts';
+
+export const stellarFungibleAIFunctionDefinition = {
+  name: 'Fungible',
+  description: 'Fungible Token Standard, compatible with SEP-41, similar to ERC-20',
+  parameters: {
+    type: 'object',
+    properties: {
+      ...addFunctionPropertiesFrom(stellarCommonFunctionDescription, [
+        'name',
+        'symbol',
+        'burnable',
+        'pausable',
+        'mintable',
+        'access',
+        'info',
+      ]),
+      premint: {
+        type: 'string',
+        description: 'The number of tokens to premint for the deployer.',
+      },
+    },
+    required: ['name', 'symbol'],
+    additionalProperties: false,
+  },
+} as const satisfies AiFunctionDefinition<'stellar', 'Fungible'>;

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

@@ -9,6 +9,10 @@ export type { RoyaltyInfoOptions as CairoRoyaltyInfoOptions } from '../../../../
 import type { KindedOptions as CairoAlphaKindedOptions } from '../../../../core/cairo_alpha/dist';
 export type { CommonContractOptions as CairoAlphaCommonContractOptions } from '../../../../core/cairo_alpha/dist/common-options';
 export type { RoyaltyInfoOptions as CairoAlphaRoyaltyInfoOptions } from '../../../../core/cairo_alpha/dist/set-royalty-info';
+//Stellar
+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 };
 
 // Add supported language here
 export type LanguagesContractsOptions = {
@@ -18,11 +22,15 @@ export type LanguagesContractsOptions = {
   };
   cairo: CairoKindedOptions;
   cairoAlpha: CairoAlphaKindedOptions;
+  stellar: Omit<StellarKindedOptions, 'Fungible'> & {
+    Fungible: StellarKindedOptions['Fungible'] & StellarCommonContractOptions;
+  };
 };
 
 export type AllLanguagesContractsOptions = LanguagesContractsOptions['solidity'] &
   LanguagesContractsOptions['cairo'] &
-  LanguagesContractsOptions['cairoAlpha'];
+  LanguagesContractsOptions['cairoAlpha'] &
+  LanguagesContractsOptions['stellar'];
 //
 
 export type SupportedLanguage = keyof LanguagesContractsOptions;

packages/ui/api/ai.ts

@@ -2,6 +2,7 @@ import { OpenAIStream } from 'ai';
 import * as solidityFunctions from './ai-assistant/function-definitions/solidity.ts';
 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 { saveChatInRedisIfDoesNotExist } from './services/redis.ts';
 import { getOpenAiInstance } from './services/open-ai.ts';
 import { getEnvironmentVariableOr } from './utils/env.ts';
@@ -19,6 +20,7 @@ const getFunctionsContext = <TLanguage extends SupportedLanguage = SupportedLang
     solidity: solidityFunctions,
     cairo: cairoFunctions,
     cairoAlpha: cairoAlphaFunctions,
+    stellar: stellarFunctions,
   };
 
   return Object.values(functionPerLanguages[language] ?? {});

packages/ui/src/stellar/App.svelte

@@ -25,6 +25,10 @@
   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 WizStellar = createWiz<'stellar'>();
 
   const dispatch = createEventDispatcher();
 
@@ -105,9 +109,22 @@
       await postConfig(opts, 'download-file', language);
     }
   };
+
+  const applyFunctionCall = ({ detail: aiFunctionCall }: CustomEvent<AiFunctionCall<'stellar'>>) => {
+    tab = sanitizeKind(aiFunctionCall.name);
+    allOpts = mergeAiAssistanceOptions(allOpts, aiFunctionCall);
+  };
 </script>
 
 <div class="container flex flex-col gap-4 p-4 rounded-3xl">
+  <WizStellar
+    language="stellar"
+    bind:currentOpts={opts}
+    bind:currentCode={code}
+    on:function-call-response={applyFunctionCall}
+    sampleMessages={['Make a fungible token with supply of 10', 'What does mintable do?']}
+  ></WizStellar>
+
   <div class="header flex flex-row justify-between">
     <div class="tab overflow-hidden">
       <OverflowMenu>

packages/ui/src/stellar/FungibleControls.svelte

@@ -7,7 +7,7 @@
   import InfoSection from './InfoSection.svelte';
   import { error } from '../common/error-tooltip';
 
-  export const opts: Required<KindedOptions['Fungible']> = {
+  export let opts: Required<KindedOptions['Fungible']> = {
     kind: 'Fungible',
     ...fungible.defaults,
     premint: '', // default to empty premint in UI instead of 0