github-clones
/
override
mirror of https://github.com/linux-do/override.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add some scripts for replacing and restoring max tokens (#21)

This commit is contained in:
wzdnzd 2024-05-19 16:00:21 +08:00 committed by GitHub
parent 247a8748dc
commit ed40f68e99
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 79 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
scripts/replace_max_tokens.vbs Normal file
View File

@ -0,0 +1,49 @@
' VBScript to change max tokens to 2048
MsgBox "It may take a few seconds to execute this script." & vbCrLf & vbCrLf & "Click 'OK' button and wait for the prompt of 'Done.' to pop up!"
Const ForReading = 1
Const ForWriting = 2
' Subpath of the file to be replaced
subpath = "dist\extension.js"
pattern = "\.maxPromptCompletionTokens\(([a-zA-Z0-9_]+),([0-9]+)\)"
replacement = ".maxPromptCompletionTokens($1,2048)"
' Iterate over all github copilot directories
Set objFSO = CreateObject("Scripting.FileSystemObject")
Set objShell = CreateObject("WScript.Shell")
Set colExtensions = objFSO.GetFolder(objShell.ExpandEnvironmentStrings("%USERPROFILE%") & "\.vscode\extensions").SubFolders
For Each objExtension In colExtensions
extension_path = objExtension.Path & "\" & subpath
If objFSO.FileExists(extension_path) Then
backupfile = extension_path & ".bak"
' Delete if backup file exists
If objFSO.FileExists(backupfile) Then
objFSO.DeleteFile backupfile, True
End If
' Backup
objFSO.CopyFile extension_path, backupfile
' Do search and replace with pattern
Set objFile = objFSO.OpenTextFile(extension_path, ForReading)
strContent = objFile.ReadAll
objFile.Close
Set objRegEx = New RegExp
objRegEx.Global = True
objRegEx.IgnoreCase = True
objRegEx.Pattern = pattern
strContent = objRegEx.Replace(strContent, replacement)
Set objFile = objFSO.OpenTextFile(extension_path, ForWriting)
objFile.Write strContent
objFile.Close
End If
Next
MsgBox "Max tokens modification completed"

30
scripts/restore_max_tokens.vbs Normal file
View File

@ -0,0 +1,30 @@
' VBScript to recovery max tokens
MsgBox "It may take a few seconds to execute this script." & vbCrLf & vbCrLf & "Click 'OK' button and wait for the prompt of 'Done.' to pop up!"
Const ForReading = 1
Const ForWriting = 2
' Subpath of the file to be recovery
subpath = "dist\extension.js"
' Iterate over all github copilot directories
Set objFSO = CreateObject("Scripting.FileSystemObject")
Set objShell = CreateObject("WScript.Shell")
Set colExtensions = objFSO.GetFolder(objShell.ExpandEnvironmentStrings("%USERPROFILE%") & "\.vscode\extensions").SubFolders
For Each objExtension In colExtensions
extension_path = objExtension.Path & "\" & subpath
backupfile = extension_path & ".bak"
If objFSO.FileExists(backupfile) Then
' Delete if exist extension file
If objFSO.FileExists(extension_path) Then
objFSO.DeleteFile extension_path, True
End If
' Replace
objFSO.MoveFile backupfile, extension_path
End If
Next
MsgBox "Restore max tokens to default successed"