Remixai chat #5241
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,43 +1,67 @@ | ||
| import * as packageJson from '../../../../../package.json' | ||
| import { ViewPlugin } from '@remixproject/engine-web' | ||
| import { Plugin } from '@remixproject/engine'; | ||
| import { RemixAITab, ChatApi } from '@remix-ui/remix-ai' | ||
| import React, { useCallback } from 'react'; | ||
| import { ICompletions, IModel, RemoteInferencer, IRemoteModel, IParams, GenerationParams, CodeExplainAgent } from '@remix/remix-ai-core'; | ||
|
|
||
| type chatRequestBufferT<T> = { | ||
| [key in keyof T]: T[key] | ||
| } | ||
|
|
||
| const profile = { | ||
| name: 'remixAI', | ||
| displayName: 'Remix AI', | ||
| methods: ['code_generation', 'code_completion', | ||
| "solidity_answer", "code_explaining", | ||
| "code_insertion", "error_explaining", | ||
| "initialize", 'chatPipe', 'ProcessChatRequestBuffer', 'isChatRequestPending'], | ||
| events: [], | ||
| icon: 'assets/img/remix-logo-blue.png', | ||
| description: 'RemixAI provides AI services to Remix IDE.', | ||
| kind: '', | ||
| location: 'sidePanel', | ||
| documentation: 'https://remix-ide.readthedocs.io/en/latest/remixai.html', | ||
| version: packageJson.version, | ||
| maintainedBy: 'Remix' | ||
| } | ||
|
|
||
| export class RemixAIPlugin extends ViewPlugin { | ||
| isOnDesktop:boolean = false | ||
| aiIsActivated:boolean = false | ||
| readonly remixDesktopPluginName = 'remixAID' | ||
| remoteInferencer:RemoteInferencer = null | ||
| isInferencing: boolean = false | ||
| chatRequestBuffer: chatRequestBufferT<any> = null | ||
| agent: CodeExplainAgent | ||
|
|
||
| constructor(inDesktop:boolean) { | ||
| super(profile) | ||
| this.isOnDesktop = inDesktop | ||
| this.agent = new CodeExplainAgent(this) | ||
| // user machine dont use ressource for remote inferencing | ||
| } | ||
|
|
||
| onActivation(): void { | ||
| if (this.isOnDesktop) { | ||
| console.log('Activating RemixAIPlugin on desktop') | ||
| this.on(this.remixDesktopPluginName, 'activated', () => { | ||
| this.call("remixAI", 'initialize', null, null, null, false); | ||
| }) | ||
| } else { | ||
| console.log('Activating RemixAIPlugin on browser') | ||
| this.initialize() | ||
| } | ||
| this.setRemixAIOnSidePannel(false) | ||
| } | ||
|
|
||
| setRemixAIOnSidePannel(resize:boolean=false){ | ||
| if (resize){ | ||
| this.call('sidePanel', 'pinView', profile) | ||
|
There was a problem hiding this comment.
There was a problem hiding this comment. Waiting for some answers from David to resize the sidepanel There was a problem hiding this comment. we will remove it from there anyway, you don't need to do this @STetsing |
||
|
|
||
| } else { | ||
| this.call('sidePanel', 'pinView', profile) | ||
| } | ||
| } | ||
|
|
||
| async initialize(model1?:IModel, model2?:IModel, remoteModel?:IRemoteModel, useRemote?:boolean){ | ||
|
|
@@ -60,7 +84,6 @@ export class RemixAIPlugin extends Plugin { | |
| } | ||
|
|
||
| } else { | ||
| this.remoteInferencer = new RemoteInferencer(remoteModel?.apiUrl, remoteModel?.completionUrl) | ||
| this.remoteInferencer.event.on('onInference', () => { | ||
| this.isInferencing = true | ||
|
|
@@ -95,61 +118,53 @@ export class RemixAIPlugin extends Plugin { | |
| } | ||
| } | ||
|
|
||
| async solidity_answer(prompt: string, params: IParams=GenerationParams): Promise<any> { | ||
| if (this.isInferencing) { | ||
| this.call('terminal', 'log', { type: 'aitypewriterwarning', value: "RemixAI is already busy!" }) | ||
| return | ||
| } | ||
|
|
||
| const newPrompt = await this.agent.chatCommand(prompt) | ||
| let result | ||
| if (this.isOnDesktop) { | ||
| result = await this.call(this.remixDesktopPluginName, 'solidity_answer', newPrompt) | ||
| } else { | ||
| result = await this.remoteInferencer.solidity_answer(newPrompt) | ||
| } | ||
| if (result && params.terminal_output) this.call('terminal', 'log', { type: 'aitypewriterwarning', value: result }) | ||
| return result | ||
| } | ||
|
|
||
| async code_explaining(prompt: string, context: string, params: IParams=GenerationParams): Promise<any> { | ||
| if (this.isInferencing) { | ||
| this.call('terminal', 'log', { type: 'aitypewriterwarning', value: "RemixAI is already busy!" }) | ||
| return | ||
| } | ||
|
|
||
| let result | ||
| if (this.isOnDesktop) { | ||
| result = await this.call(this.remixDesktopPluginName, 'code_explaining', prompt, context, params) | ||
|
|
||
| } else { | ||
| result = await this.remoteInferencer.code_explaining(prompt, context, params) | ||
| } | ||
| if (result && params.terminal_output) this.call('terminal', 'log', { type: 'aitypewriterwarning', value: result }) | ||
| return result | ||
| } | ||
|
|
||
| async error_explaining(prompt: string, context: string="", params: IParams=GenerationParams): Promise<any> { | ||
| if (this.isInferencing) { | ||
| this.call('terminal', 'log', { type: 'aitypewriterwarning', value: "RemixAI is already busy!" }) | ||
| return | ||
| } | ||
|
|
||
| let result | ||
| if (this.isOnDesktop) { | ||
| result = await this.call(this.remixDesktopPluginName, 'error_explaining', prompt) | ||
| } else { | ||
| result = await this.remoteInferencer.error_explaining(prompt, params) | ||
| } | ||
| if (result && params.terminal_output) this.call('terminal', 'log', { type: 'aitypewriterwarning', value: result }) | ||
| return result | ||
| } | ||
|
|
||
|
|
@@ -161,9 +176,44 @@ export class RemixAIPlugin extends Plugin { | |
| } | ||
| } | ||
|
|
||
|
There was a problem hiding this comment. This goes out from this PR |
||
| chatPipe(fn, prompt: string, context?: string, pipeMessage?: string){ | ||
| if (this.chatRequestBuffer == null){ | ||
| this.chatRequestBuffer = { | ||
| fn_name: fn, | ||
| prompt: prompt, | ||
| context: context | ||
| } | ||
| if (pipeMessage) ChatApi.composer.send(pipeMessage) | ||
| else { | ||
| if (fn === "code_explaining") ChatApi.composer.send("Explain the current code") | ||
| else if (fn === "error_explaining") ChatApi.composer.send("Explain the error") | ||
| else if (fn === "solidity_answer") ChatApi.composer.send("Answer the following question") | ||
| else console.log("chatRequestBuffer is not empty. First process the last request.") | ||
| } | ||
| } | ||
| else { | ||
| console.log("chatRequestBuffer is not empty. First process the last request.") | ||
| } | ||
| } | ||
|
|
||
| async ProcessChatRequestBuffer(params:IParams=GenerationParams){ | ||
| if (this.chatRequestBuffer != null){ | ||
| const result = this[this.chatRequestBuffer.fn_name](this.chatRequestBuffer.prompt, this.chatRequestBuffer.context, params) | ||
| this.chatRequestBuffer = null | ||
| return result | ||
| } | ||
| else { | ||
| console.log("chatRequestBuffer is empty.") | ||
| return "" | ||
| } | ||
| } | ||
| isChatRequestPending(){ | ||
| return this.chatRequestBuffer != null | ||
| } | ||
|
|
||
| render() { | ||
| return ( | ||
| <RemixAITab plugin={this}></RemixAITab> | ||
| ) | ||
| } | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,29 +1,45 @@ | ||
| // interactive code explaining and highlight security vunerabilities | ||
| import * as fs from 'fs'; | ||
|
|
||
| export class CodeExplainAgent { | ||
| private codebase: string[]; // list of code base file | ||
| public currentFile: string; | ||
| plugin | ||
|
|
||
| constructor(props) { | ||
| this.plugin = props | ||
|
|
||
| // git or fs | ||
| const codebase = this.loadCodebase("codebasePath"); | ||
| } | ||
|
|
||
| private loadCodebase(path: string): string[] { | ||
| return [] | ||
| } | ||
|
|
||
| public update(currentFile, lineNumber){ | ||
|
|
||
| } | ||
|
|
||
| async chatCommand(prompt:string){ | ||
| // change this function with indexer or related | ||
| try{ | ||
| if (prompt.includes('Explain briefly the current file')){ | ||
| const file = await this.plugin.call('fileManager', 'getCurrentFile') | ||
| const content = `Explain this code:\n ${await this.plugin.call('fileManager', 'readFile', file)}` | ||
| return content | ||
| } else return prompt | ||
| } catch { | ||
| console.log('There is No file selected') | ||
| return 'There is No file selected' | ||
| } | ||
| } | ||
|
|
||
| public getExplanations(currentLine: string, numSuggestions: number = 3): string[] { | ||
| // process the code base explaining the current file and highlight some details | ||
| const suggestions: string[] = []; | ||
| return suggestions; | ||
| } | ||
| } | ||
|
|
||
| // Handle file changed (significantly) |
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
could you remove all the comments?