|
|
|
@ -1,18 +1,20 @@ |
|
|
|
|
import {Plugin} from '@remixproject/engine' |
|
|
|
|
import {SuggestionService, SuggestOptions} from './suggestion-service' |
|
|
|
|
import axios, {AxiosResponse} from 'axios' |
|
|
|
|
const _paq = (window._paq = window._paq || []) //eslint-disable-line
|
|
|
|
|
|
|
|
|
|
const profile = { |
|
|
|
|
name: 'copilot-suggestion', |
|
|
|
|
displayName: 'copilot-suggestion', |
|
|
|
|
description: 'Get Solidity suggestions in editor', |
|
|
|
|
methods: ['suggest', 'init', 'uninstall', 'status', 'isActivate'], |
|
|
|
|
methods: ['suggest', 'init', 'uninstall', 'status', 'isActivate', 'useRemoteService', 'discardRemoteService'], |
|
|
|
|
version: '0.1.0-alpha', |
|
|
|
|
maintainedBy: "Remix" |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
export class CopilotSuggestion extends Plugin { |
|
|
|
|
service: SuggestionService |
|
|
|
|
remoteService: string |
|
|
|
|
context: string |
|
|
|
|
ready: boolean |
|
|
|
|
constructor() { |
|
|
|
@ -29,6 +31,14 @@ export class CopilotSuggestion extends Plugin { |
|
|
|
|
})
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
useRemoteService(service: string) { |
|
|
|
|
this.remoteService = service |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
discardRemoteService() { |
|
|
|
|
this.remoteService = null |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
status () { |
|
|
|
|
return this.ready |
|
|
|
|
} |
|
|
|
@ -53,7 +63,14 @@ export class CopilotSuggestion extends Plugin { |
|
|
|
|
temperature: temperature || 0, |
|
|
|
|
max_new_tokens: max_new_tokens || 0 |
|
|
|
|
} |
|
|
|
|
return this.service.suggest(this.context ? this.context + '\n\n' + content : content, options) |
|
|
|
|
|
|
|
|
|
if (this.remoteService) { |
|
|
|
|
const {data} = await axios.post(this.remoteService, {context: content, max_new_words: options.max_new_tokens, temperature: options.temperature}) |
|
|
|
|
const parsedData = JSON.parse(data).trimStart() |
|
|
|
|
return {output: [{generated_text: parsedData}]} |
|
|
|
|
} else { |
|
|
|
|
return this.service.suggest(this.context ? this.context + '\n\n' + content : content, options) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
async loadModeContent() { |
|
|
|
|