STetsing
2 months ago
parent
5ee1f2fc9c
commit
ee852709a9
@ -1,283 +0,0 @@ |
||||
import { Plugin } from '@remixproject/engine' |
||||
import axios from 'axios' |
||||
|
||||
export type SuggestOptions = { |
||||
max_new_tokens: number, |
||||
temperature: number, |
||||
do_sample:boolean |
||||
top_k: number, |
||||
top_p:number, |
||||
stream_result:boolean |
||||
} |
||||
|
||||
const _paq = (window._paq = window._paq || []) |
||||
|
||||
const profile = { |
||||
name: 'solcoder', |
||||
displayName: 'solcoder', |
||||
description: 'solcoder', |
||||
methods: ['code_generation', 'code_completion', "solidity_answer", "code_explaining", "code_insertion", "error_explaining"], |
||||
events: [], |
||||
maintainedBy: 'Remix', |
||||
} |
||||
type ChatEntry = [string, string]; |
||||
|
||||
enum BackendOPModel{ |
||||
DeepSeek, |
||||
CodeLLama, |
||||
Mistral |
||||
} |
||||
|
||||
const PromptBuilder = (inst, answr, modelop) => { |
||||
if (modelop === BackendOPModel.CodeLLama) return "" |
||||
if (modelop === BackendOPModel.DeepSeek) return "\n### INSTRUCTION:\n" + inst + "\n### RESPONSE:\n" + answr |
||||
if (modelop === BackendOPModel.Mistral) return "" |
||||
} |
||||
|
||||
export class SolCoder extends Plugin { |
||||
api_url: string |
||||
completion_url: string |
||||
solgpt_chat_history:ChatEntry[] |
||||
max_history = 7 |
||||
model_op = BackendOPModel.DeepSeek |
||||
|
||||
constructor() { |
||||
super(profile) |
||||
this.api_url = "https://solcoder.remixproject.org" |
||||
this.completion_url = "https://completion.remixproject.org" |
||||
this.solgpt_chat_history = [] |
||||
} |
||||
|
||||
pushChatHistory(prompt, result){ |
||||
const chat:ChatEntry = [prompt, result.data[0]] |
||||
this.solgpt_chat_history.push(chat) |
||||
if (this.solgpt_chat_history.length > this.max_history){this.solgpt_chat_history.shift()} |
||||
} |
||||
|
||||
async code_generation(prompt): Promise<any> { |
||||
this.emit("aiInfering") |
||||
this.call('layout', 'maximizeTerminal') |
||||
_paq.push(['trackEvent', 'ai', 'solcoder', 'code_generation']) |
||||
|
||||
let result |
||||
try { |
||||
result = await( |
||||
await fetch(this.api_url, { |
||||
method: 'POST', |
||||
headers: { |
||||
Accept: 'application/json', |
||||
'Content-Type': 'application/json', |
||||
}, |
||||
body: JSON.stringify({ "data":[prompt, "code_completion", "", false,1000,0.9,0.92,50]}), |
||||
}) |
||||
).json() |
||||
if ("error" in result){ |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: result.error }) |
||||
return result |
||||
} |
||||
return result.data |
||||
} catch (e) { |
||||
this.call('terminal', 'log', { type: 'typewritererror', value: `Unable to get a response ${e.message}` }) |
||||
return |
||||
} finally { |
||||
this.emit("aiInferingDone") |
||||
} |
||||
} |
||||
|
||||
async solidity_answer(prompt): Promise<any> { |
||||
this.emit("aiInfering") |
||||
this.call('layout', 'maximizeTerminal') |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: `\n\nWaiting for RemixAI answer...` }) |
||||
_paq.push(['trackEvent', 'ai', 'solcoder', 'answering']) |
||||
|
||||
let result |
||||
try { |
||||
const main_prompt = this._build_solgpt_promt(prompt) |
||||
result = await( |
||||
await fetch(this.api_url, { |
||||
method: 'POST', |
||||
headers: { |
||||
Accept: 'application/json', |
||||
'Content-Type': 'application/json', |
||||
}, |
||||
body: JSON.stringify({ "data":[main_prompt, "solidity_answer", false,1000,0.9,0.8,50]}), |
||||
}) |
||||
).json() |
||||
} catch (e) { |
||||
this.call('terminal', 'log', { type: 'typewritererror', value: `Unable to get a response ${e.message}` }) |
||||
this.solgpt_chat_history = [] |
||||
return |
||||
} finally { |
||||
this.emit("aiInferingDone") |
||||
} |
||||
if (result) { |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: result.data[0] }) |
||||
this.pushChatHistory(prompt, result) |
||||
} else if (result.error) { |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: "Error on request" }) |
||||
} |
||||
|
||||
} |
||||
|
||||
async code_explaining(prompt, context:string=""): Promise<any> { |
||||
this.emit("aiInfering") |
||||
this.call('layout', 'maximizeTerminal') |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: `\n\nWaiting for RemixAI answer...` }) |
||||
_paq.push(['trackEvent', 'ai', 'solcoder', 'explaining']) |
||||
|
||||
let result |
||||
try { |
||||
result = await( |
||||
await fetch(this.api_url, { |
||||
method: 'POST', |
||||
headers: { |
||||
Accept: 'application/json', |
||||
'Content-Type': 'application/json', |
||||
}, |
||||
body: JSON.stringify({ "data":[prompt, "code_explaining", false,2000,0.9,0.8,50, context]}), |
||||
}) |
||||
).json() |
||||
if (result) { |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: result.data[0] }) |
||||
this.pushChatHistory(prompt, result) |
||||
} |
||||
return result.data[0] |
||||
} catch (e) { |
||||
this.call('terminal', 'log', { type: 'typewritererror', value: `Unable to get a response ${e.message}` }) |
||||
return |
||||
} finally { |
||||
this.emit("aiInferingDone") |
||||
} |
||||
} |
||||
|
||||
async code_completion(prompt, options:SuggestOptions=null): Promise<any> { |
||||
this.emit("aiInfering") |
||||
_paq.push(['trackEvent', 'ai', 'solcoder', 'code_completion']) |
||||
|
||||
let result |
||||
try { |
||||
result = await( |
||||
await fetch(this.completion_url, { |
||||
method: 'POST', |
||||
headers: { |
||||
Accept: 'application/json', |
||||
'Content-Type': 'application/json', |
||||
}, |
||||
body: JSON.stringify({ "data": !options? [ |
||||
prompt, // string in 'context_code' Textbox component
|
||||
"code_completion", |
||||
"", // string in 'comment' Textbox component
|
||||
false, // boolean in 'stream_result' Checkbox component
|
||||
30, // number (numeric value between 0 and 2000) in 'max_new_tokens' Slider component
|
||||
0.9, // number (numeric value between 0.01 and 1) in 'temperature' Slider component
|
||||
0.90, // number (numeric value between 0 and 1) in 'top_p' Slider component
|
||||
50, // number (numeric value between 1 and 200) in 'top_k' Slider component
|
||||
] : [ |
||||
prompt, |
||||
"code_completion", |
||||
"", |
||||
options.stream_result, |
||||
options.max_new_tokens, |
||||
options.temperature, |
||||
options.top_p, |
||||
options.top_k |
||||
]}), |
||||
}) |
||||
).json() |
||||
|
||||
if ("error" in result){ |
||||
return result |
||||
} |
||||
return result.data |
||||
|
||||
} catch (e) { |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: `Unable to get a response ${e.message}` }) |
||||
return |
||||
} finally { |
||||
this.emit("aiInferingDone") |
||||
} |
||||
} |
||||
|
||||
async code_insertion(msg_pfx, msg_sfx): Promise<any> { |
||||
this.emit("aiInfering") |
||||
_paq.push(['trackEvent', 'ai', 'solcoder', 'code_insertion']) |
||||
|
||||
let result |
||||
try { |
||||
result = await( |
||||
await fetch(this.completion_url, { |
||||
method: 'POST', |
||||
headers: { |
||||
Accept: 'application/json', |
||||
'Content-Type': 'application/json', |
||||
}, |
||||
body: JSON.stringify({ "data":[ |
||||
msg_pfx, // Text before current cursor line
|
||||
"code_insertion", |
||||
msg_sfx, // Text after current cursor line
|
||||
1024, |
||||
0.5, |
||||
0.92, |
||||
50 |
||||
]}), |
||||
}) |
||||
).json() |
||||
|
||||
if ("error" in result){ |
||||
return result |
||||
} |
||||
return result.data |
||||
|
||||
} catch (e) { |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: `Unable to get a response ${e.message}` }) |
||||
return |
||||
} finally { |
||||
this.emit("aiInferingDone") |
||||
} |
||||
} |
||||
|
||||
async error_explaining(prompt): Promise<any> { |
||||
this.emit("aiInfering") |
||||
this.call('layout', 'maximizeTerminal') |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: `\n\nWaiting for RemixAI answer...` }) |
||||
_paq.push(['trackEvent', 'ai', 'solcoder', 'explaining']) |
||||
|
||||
let result |
||||
try { |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value:"explaining with axios" }) |
||||
result = await axios(this.api_url, { |
||||
method: 'POST', |
||||
headers: { |
||||
Accept: 'application/json', |
||||
'Content-Type': 'application/json', |
||||
}, |
||||
data: JSON.stringify({ "data":[prompt, "error_explaining", false,2000,0.9,0.8,50]}), |
||||
}) |
||||
if (result.statusText === "OK") { |
||||
this.call('terminal', 'log', { type: 'aitypewriterwarning', value: result.data.data[0] }) |
||||
this.pushChatHistory(prompt, result) |
||||
} |
||||
return result.data.data[0] |
||||
} catch (e) { |
||||
this.call('terminal', 'log', { type: 'typewritererror', value: `Unable to get a response ${e.message}` }) |
||||
return |
||||
} finally { |
||||
this.emit("aiInferingDone") |
||||
} |
||||
} |
||||
|
||||
_build_solgpt_promt(user_promt:string){ |
||||
if (this.solgpt_chat_history.length === 0){ |
||||
return user_promt |
||||
} else { |
||||
let new_promt = "" |
||||
for (const [question, answer] of this.solgpt_chat_history) { |
||||
new_promt += PromptBuilder(question.split('sol-gpt')[1], answer, this.model_op) |
||||
} |
||||
// finaly
|
||||
new_promt = "sol-gpt " + new_promt + PromptBuilder(user_promt.split('sol-gpt')[1], "", this.model_op) |
||||
return new_promt |
||||
} |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue