diff --git a/pilot-env/bin/Activate.ps1 b/pilot-env/bin/Activate.ps1 new file mode 100644 index 0000000..b49d77b --- /dev/null +++ b/pilot-env/bin/Activate.ps1 @@ -0,0 +1,247 @@ +<# +.Synopsis +Activate a Python virtual environment for the current PowerShell session. + +.Description +Pushes the python executable for a virtual environment to the front of the +$Env:PATH environment variable and sets the prompt to signify that you are +in a Python virtual environment. Makes use of the command line switches as +well as the `pyvenv.cfg` file values present in the virtual environment. + +.Parameter VenvDir +Path to the directory that contains the virtual environment to activate. The +default value for this is the parent of the directory that the Activate.ps1 +script is located within. + +.Parameter Prompt +The prompt prefix to display when this virtual environment is activated. By +default, this prompt is the name of the virtual environment folder (VenvDir) +surrounded by parentheses and followed by a single space (ie. '(.venv) '). + +.Example +Activate.ps1 +Activates the Python virtual environment that contains the Activate.ps1 script. + +.Example +Activate.ps1 -Verbose +Activates the Python virtual environment that contains the Activate.ps1 script, +and shows extra information about the activation as it executes. + +.Example +Activate.ps1 -VenvDir C:\Users\MyUser\Common\.venv +Activates the Python virtual environment located in the specified location. + +.Example +Activate.ps1 -Prompt "MyPython" +Activates the Python virtual environment that contains the Activate.ps1 script, +and prefixes the current prompt with the specified string (surrounded in +parentheses) while the virtual environment is active. + +.Notes +On Windows, it may be required to enable this Activate.ps1 script by setting the +execution policy for the user. You can do this by issuing the following PowerShell +command: + +PS C:\> Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope CurrentUser + +For more information on Execution Policies: +https://go.microsoft.com/fwlink/?LinkID=135170 + +#> +Param( + [Parameter(Mandatory = $false)] + [String] + $VenvDir, + [Parameter(Mandatory = $false)] + [String] + $Prompt +) + +<# Function declarations --------------------------------------------------- #> + +<# +.Synopsis +Remove all shell session elements added by the Activate script, including the +addition of the virtual environment's Python executable from the beginning of +the PATH variable. + +.Parameter NonDestructive +If present, do not remove this function from the global namespace for the +session. + +#> +function global:deactivate ([switch]$NonDestructive) { + # Revert to original values + + # The prior prompt: + if (Test-Path -Path Function:_OLD_VIRTUAL_PROMPT) { + Copy-Item -Path Function:_OLD_VIRTUAL_PROMPT -Destination Function:prompt + Remove-Item -Path Function:_OLD_VIRTUAL_PROMPT + } + + # The prior PYTHONHOME: + if (Test-Path -Path Env:_OLD_VIRTUAL_PYTHONHOME) { + Copy-Item -Path Env:_OLD_VIRTUAL_PYTHONHOME -Destination Env:PYTHONHOME + Remove-Item -Path Env:_OLD_VIRTUAL_PYTHONHOME + } + + # The prior PATH: + if (Test-Path -Path Env:_OLD_VIRTUAL_PATH) { + Copy-Item -Path Env:_OLD_VIRTUAL_PATH -Destination Env:PATH + Remove-Item -Path Env:_OLD_VIRTUAL_PATH + } + + # Just remove the VIRTUAL_ENV altogether: + if (Test-Path -Path Env:VIRTUAL_ENV) { + Remove-Item -Path env:VIRTUAL_ENV + } + + # Just remove VIRTUAL_ENV_PROMPT altogether. + if (Test-Path -Path Env:VIRTUAL_ENV_PROMPT) { + Remove-Item -Path env:VIRTUAL_ENV_PROMPT + } + + # Just remove the _PYTHON_VENV_PROMPT_PREFIX altogether: + if (Get-Variable -Name "_PYTHON_VENV_PROMPT_PREFIX" -ErrorAction SilentlyContinue) { + Remove-Variable -Name _PYTHON_VENV_PROMPT_PREFIX -Scope Global -Force + } + + # Leave deactivate function in the global namespace if requested: + if (-not $NonDestructive) { + Remove-Item -Path function:deactivate + } +} + +<# +.Description +Get-PyVenvConfig parses the values from the pyvenv.cfg file located in the +given folder, and returns them in a map. + +For each line in the pyvenv.cfg file, if that line can be parsed into exactly +two strings separated by `=` (with any amount of whitespace surrounding the =) +then it is considered a `key = value` line. The left hand string is the key, +the right hand is the value. + +If the value starts with a `'` or a `"` then the first and last character is +stripped from the value before being captured. + +.Parameter ConfigDir +Path to the directory that contains the `pyvenv.cfg` file. +#> +function Get-PyVenvConfig( + [String] + $ConfigDir +) { + Write-Verbose "Given ConfigDir=$ConfigDir, obtain values in pyvenv.cfg" + + # Ensure the file exists, and issue a warning if it doesn't (but still allow the function to continue). + $pyvenvConfigPath = Join-Path -Resolve -Path $ConfigDir -ChildPath 'pyvenv.cfg' -ErrorAction Continue + + # An empty map will be returned if no config file is found. + $pyvenvConfig = @{ } + + if ($pyvenvConfigPath) { + + Write-Verbose "File exists, parse `key = value` lines" + $pyvenvConfigContent = Get-Content -Path $pyvenvConfigPath + + $pyvenvConfigContent | ForEach-Object { + $keyval = $PSItem -split "\s*=\s*", 2 + if ($keyval[0] -and $keyval[1]) { + $val = $keyval[1] + + # Remove extraneous quotations around a string value. + if ("'""".Contains($val.Substring(0, 1))) { + $val = $val.Substring(1, $val.Length - 2) + } + + $pyvenvConfig[$keyval[0]] = $val + Write-Verbose "Adding Key: '$($keyval[0])'='$val'" + } + } + } + return $pyvenvConfig +} + + +<# Begin Activate script --------------------------------------------------- #> + +# Determine the containing directory of this script +$VenvExecPath = Split-Path -Parent $MyInvocation.MyCommand.Definition +$VenvExecDir = Get-Item -Path $VenvExecPath + +Write-Verbose "Activation script is located in path: '$VenvExecPath'" +Write-Verbose "VenvExecDir Fullname: '$($VenvExecDir.FullName)" +Write-Verbose "VenvExecDir Name: '$($VenvExecDir.Name)" + +# Set values required in priority: CmdLine, ConfigFile, Default +# First, get the location of the virtual environment, it might not be +# VenvExecDir if specified on the command line. +if ($VenvDir) { + Write-Verbose "VenvDir given as parameter, using '$VenvDir' to determine values" +} +else { + Write-Verbose "VenvDir not given as a parameter, using parent directory name as VenvDir." + $VenvDir = $VenvExecDir.Parent.FullName.TrimEnd("\\/") + Write-Verbose "VenvDir=$VenvDir" +} + +# Next, read the `pyvenv.cfg` file to determine any required value such +# as `prompt`. +$pyvenvCfg = Get-PyVenvConfig -ConfigDir $VenvDir + +# Next, set the prompt from the command line, or the config file, or +# just use the name of the virtual environment folder. +if ($Prompt) { + Write-Verbose "Prompt specified as argument, using '$Prompt'" +} +else { + Write-Verbose "Prompt not specified as argument to script, checking pyvenv.cfg value" + if ($pyvenvCfg -and $pyvenvCfg['prompt']) { + Write-Verbose " Setting based on value in pyvenv.cfg='$($pyvenvCfg['prompt'])'" + $Prompt = $pyvenvCfg['prompt']; + } + else { + Write-Verbose " Setting prompt based on parent's directory's name. (Is the directory name passed to venv module when creating the virtual environment)" + Write-Verbose " Got leaf-name of $VenvDir='$(Split-Path -Path $venvDir -Leaf)'" + $Prompt = Split-Path -Path $venvDir -Leaf + } +} + +Write-Verbose "Prompt = '$Prompt'" +Write-Verbose "VenvDir='$VenvDir'" + +# Deactivate any currently active virtual environment, but leave the +# deactivate function in place. +deactivate -nondestructive + +# Now set the environment variable VIRTUAL_ENV, used by many tools to determine +# that there is an activated venv. +$env:VIRTUAL_ENV = $VenvDir + +if (-not $Env:VIRTUAL_ENV_DISABLE_PROMPT) { + + Write-Verbose "Setting prompt to '$Prompt'" + + # Set the prompt to include the env name + # Make sure _OLD_VIRTUAL_PROMPT is global + function global:_OLD_VIRTUAL_PROMPT { "" } + Copy-Item -Path function:prompt -Destination function:_OLD_VIRTUAL_PROMPT + New-Variable -Name _PYTHON_VENV_PROMPT_PREFIX -Description "Python virtual environment prompt prefix" -Scope Global -Option ReadOnly -Visibility Public -Value $Prompt + + function global:prompt { + Write-Host -NoNewline -ForegroundColor Green "($_PYTHON_VENV_PROMPT_PREFIX) " + _OLD_VIRTUAL_PROMPT + } + $env:VIRTUAL_ENV_PROMPT = $Prompt +} + +# Clear PYTHONHOME +if (Test-Path -Path Env:PYTHONHOME) { + Copy-Item -Path Env:PYTHONHOME -Destination Env:_OLD_VIRTUAL_PYTHONHOME + Remove-Item -Path Env:PYTHONHOME +} + +# Add the venv to the PATH +Copy-Item -Path Env:PATH -Destination Env:_OLD_VIRTUAL_PATH +$Env:PATH = "$VenvExecDir$([System.IO.Path]::PathSeparator)$Env:PATH" diff --git a/pilot-env/bin/activate b/pilot-env/bin/activate new file mode 100644 index 0000000..93c7a82 --- /dev/null +++ b/pilot-env/bin/activate @@ -0,0 +1,69 @@ +# This file must be used with "source bin/activate" *from bash* +# you cannot run it directly + +deactivate () { + # reset old environment variables + if [ -n "${_OLD_VIRTUAL_PATH:-}" ] ; then + PATH="${_OLD_VIRTUAL_PATH:-}" + export PATH + unset _OLD_VIRTUAL_PATH + fi + if [ -n "${_OLD_VIRTUAL_PYTHONHOME:-}" ] ; then + PYTHONHOME="${_OLD_VIRTUAL_PYTHONHOME:-}" + export PYTHONHOME + unset _OLD_VIRTUAL_PYTHONHOME + fi + + # This should detect bash and zsh, which have a hash command that must + # be called to get it to forget past commands. Without forgetting + # past commands the $PATH changes we made may not be respected + if [ -n "${BASH:-}" -o -n "${ZSH_VERSION:-}" ] ; then + hash -r 2> /dev/null + fi + + if [ -n "${_OLD_VIRTUAL_PS1:-}" ] ; then + PS1="${_OLD_VIRTUAL_PS1:-}" + export PS1 + unset _OLD_VIRTUAL_PS1 + fi + + unset VIRTUAL_ENV + unset VIRTUAL_ENV_PROMPT + if [ ! "${1:-}" = "nondestructive" ] ; then + # Self destruct! + unset -f deactivate + fi +} + +# unset irrelevant variables +deactivate nondestructive + +VIRTUAL_ENV="/Users/duynguyen/www/gpt-pilot/pilot-env" +export VIRTUAL_ENV + +_OLD_VIRTUAL_PATH="$PATH" +PATH="$VIRTUAL_ENV/bin:$PATH" +export PATH + +# unset PYTHONHOME if set +# this will fail if PYTHONHOME is set to the empty string (which is bad anyway) +# could use `if (set -u; : $PYTHONHOME) ;` in bash +if [ -n "${PYTHONHOME:-}" ] ; then + _OLD_VIRTUAL_PYTHONHOME="${PYTHONHOME:-}" + unset PYTHONHOME +fi + +if [ -z "${VIRTUAL_ENV_DISABLE_PROMPT:-}" ] ; then + _OLD_VIRTUAL_PS1="${PS1:-}" + PS1="(pilot-env) ${PS1:-}" + export PS1 + VIRTUAL_ENV_PROMPT="(pilot-env) " + export VIRTUAL_ENV_PROMPT +fi + +# This should detect bash and zsh, which have a hash command that must +# be called to get it to forget past commands. Without forgetting +# past commands the $PATH changes we made may not be respected +if [ -n "${BASH:-}" -o -n "${ZSH_VERSION:-}" ] ; then + hash -r 2> /dev/null +fi diff --git a/pilot-env/bin/activate.csh b/pilot-env/bin/activate.csh new file mode 100644 index 0000000..d8732ba --- /dev/null +++ b/pilot-env/bin/activate.csh @@ -0,0 +1,26 @@ +# This file must be used with "source bin/activate.csh" *from csh*. +# You cannot run it directly. +# Created by Davide Di Blasi . +# Ported to Python 3.3 venv by Andrew Svetlov + +alias deactivate 'test $?_OLD_VIRTUAL_PATH != 0 && setenv PATH "$_OLD_VIRTUAL_PATH" && unset _OLD_VIRTUAL_PATH; rehash; test $?_OLD_VIRTUAL_PROMPT != 0 && set prompt="$_OLD_VIRTUAL_PROMPT" && unset _OLD_VIRTUAL_PROMPT; unsetenv VIRTUAL_ENV; unsetenv VIRTUAL_ENV_PROMPT; test "\!:*" != "nondestructive" && unalias deactivate' + +# Unset irrelevant variables. +deactivate nondestructive + +setenv VIRTUAL_ENV "/Users/duynguyen/www/gpt-pilot/pilot-env" + +set _OLD_VIRTUAL_PATH="$PATH" +setenv PATH "$VIRTUAL_ENV/bin:$PATH" + + +set _OLD_VIRTUAL_PROMPT="$prompt" + +if (! "$?VIRTUAL_ENV_DISABLE_PROMPT") then + set prompt = "(pilot-env) $prompt" + setenv VIRTUAL_ENV_PROMPT "(pilot-env) " +endif + +alias pydoc python -m pydoc + +rehash diff --git a/pilot-env/bin/activate.fish b/pilot-env/bin/activate.fish new file mode 100644 index 0000000..30f0628 --- /dev/null +++ b/pilot-env/bin/activate.fish @@ -0,0 +1,69 @@ +# This file must be used with "source /bin/activate.fish" *from fish* +# (https://fishshell.com/); you cannot run it directly. + +function deactivate -d "Exit virtual environment and return to normal shell environment" + # reset old environment variables + if test -n "$_OLD_VIRTUAL_PATH" + set -gx PATH $_OLD_VIRTUAL_PATH + set -e _OLD_VIRTUAL_PATH + end + if test -n "$_OLD_VIRTUAL_PYTHONHOME" + set -gx PYTHONHOME $_OLD_VIRTUAL_PYTHONHOME + set -e _OLD_VIRTUAL_PYTHONHOME + end + + if test -n "$_OLD_FISH_PROMPT_OVERRIDE" + set -e _OLD_FISH_PROMPT_OVERRIDE + # prevents error when using nested fish instances (Issue #93858) + if functions -q _old_fish_prompt + functions -e fish_prompt + functions -c _old_fish_prompt fish_prompt + functions -e _old_fish_prompt + end + end + + set -e VIRTUAL_ENV + set -e VIRTUAL_ENV_PROMPT + if test "$argv[1]" != "nondestructive" + # Self-destruct! + functions -e deactivate + end +end + +# Unset irrelevant variables. +deactivate nondestructive + +set -gx VIRTUAL_ENV "/Users/duynguyen/www/gpt-pilot/pilot-env" + +set -gx _OLD_VIRTUAL_PATH $PATH +set -gx PATH "$VIRTUAL_ENV/bin" $PATH + +# Unset PYTHONHOME if set. +if set -q PYTHONHOME + set -gx _OLD_VIRTUAL_PYTHONHOME $PYTHONHOME + set -e PYTHONHOME +end + +if test -z "$VIRTUAL_ENV_DISABLE_PROMPT" + # fish uses a function instead of an env var to generate the prompt. + + # Save the current fish_prompt function as the function _old_fish_prompt. + functions -c fish_prompt _old_fish_prompt + + # With the original prompt function renamed, we can override with our own. + function fish_prompt + # Save the return status of the last command. + set -l old_status $status + + # Output the venv prompt; color taken from the blue of the Python logo. + printf "%s%s%s" (set_color 4B8BBE) "(pilot-env) " (set_color normal) + + # Restore the return status of the previous command. + echo "exit $old_status" | . + # Output the original/"old" prompt. + _old_fish_prompt + end + + set -gx _OLD_FISH_PROMPT_OVERRIDE "$VIRTUAL_ENV" + set -gx VIRTUAL_ENV_PROMPT "(pilot-env) " +end diff --git a/pilot-env/bin/distro b/pilot-env/bin/distro new file mode 100755 index 0000000..e7b1433 --- /dev/null +++ b/pilot-env/bin/distro @@ -0,0 +1,8 @@ +#!/Users/duynguyen/www/gpt-pilot/pilot-env/bin/python3.11 +# -*- coding: utf-8 -*- +import re +import sys +from distro.distro import main +if __name__ == '__main__': + sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) + sys.exit(main()) diff --git a/pilot-env/bin/dotenv b/pilot-env/bin/dotenv new file mode 100755 index 0000000..6480193 --- /dev/null +++ b/pilot-env/bin/dotenv @@ -0,0 +1,8 @@ +#!/Users/duynguyen/www/gpt-pilot/pilot-env/bin/python3.11 +# -*- coding: utf-8 -*- +import re +import sys +from dotenv.__main__ import cli +if __name__ == '__main__': + sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) + sys.exit(cli()) diff --git a/pilot-env/bin/normalizer b/pilot-env/bin/normalizer new file mode 100755 index 0000000..a800dd9 --- /dev/null +++ b/pilot-env/bin/normalizer @@ -0,0 +1,8 @@ +#!/Users/duynguyen/www/gpt-pilot/pilot-env/bin/python3.11 +# -*- coding: utf-8 -*- +import re +import sys +from charset_normalizer.cli.normalizer import cli_detect +if __name__ == '__main__': + sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) + sys.exit(cli_detect()) diff --git a/pilot-env/bin/pip b/pilot-env/bin/pip new file mode 100755 index 0000000..c3089ba --- /dev/null +++ b/pilot-env/bin/pip @@ -0,0 +1,8 @@ +#!/Users/duynguyen/www/gpt-pilot/pilot-env/bin/python3.11 +# -*- coding: utf-8 -*- +import re +import sys +from pip._internal.cli.main import main +if __name__ == '__main__': + sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) + sys.exit(main()) diff --git a/pilot-env/bin/pip3 b/pilot-env/bin/pip3 new file mode 100755 index 0000000..c3089ba --- /dev/null +++ b/pilot-env/bin/pip3 @@ -0,0 +1,8 @@ +#!/Users/duynguyen/www/gpt-pilot/pilot-env/bin/python3.11 +# -*- coding: utf-8 -*- +import re +import sys +from pip._internal.cli.main import main +if __name__ == '__main__': + sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) + sys.exit(main()) diff --git a/pilot-env/bin/pip3.11 b/pilot-env/bin/pip3.11 new file mode 100755 index 0000000..c3089ba --- /dev/null +++ b/pilot-env/bin/pip3.11 @@ -0,0 +1,8 @@ +#!/Users/duynguyen/www/gpt-pilot/pilot-env/bin/python3.11 +# -*- coding: utf-8 -*- +import re +import sys +from pip._internal.cli.main import main +if __name__ == '__main__': + sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) + sys.exit(main()) diff --git a/pilot-env/bin/pwiz.py b/pilot-env/bin/pwiz.py new file mode 100755 index 0000000..3576c90 --- /dev/null +++ b/pilot-env/bin/pwiz.py @@ -0,0 +1,228 @@ +#!/Users/duynguyen/www/gpt-pilot/pilot-env/bin/python3.11 + +import datetime +import os +import sys +from getpass import getpass +from optparse import OptionParser + +from peewee import * +from peewee import print_ +from peewee import __version__ as peewee_version +from playhouse.cockroachdb import CockroachDatabase +from playhouse.reflection import * + + +HEADER = """from peewee import *%s + +database = %s('%s'%s) +""" + +BASE_MODEL = """\ +class BaseModel(Model): + class Meta: + database = database +""" + +UNKNOWN_FIELD = """\ +class UnknownField(object): + def __init__(self, *_, **__): pass +""" + +DATABASE_ALIASES = { + CockroachDatabase: ['cockroach', 'cockroachdb', 'crdb'], + MySQLDatabase: ['mysql', 'mysqldb'], + PostgresqlDatabase: ['postgres', 'postgresql'], + SqliteDatabase: ['sqlite', 'sqlite3'], +} + +DATABASE_MAP = dict((value, key) + for key in DATABASE_ALIASES + for value in DATABASE_ALIASES[key]) + +def make_introspector(database_type, database_name, **kwargs): + if database_type not in DATABASE_MAP: + err('Unrecognized database, must be one of: %s' % + ', '.join(DATABASE_MAP.keys())) + sys.exit(1) + + schema = kwargs.pop('schema', None) + DatabaseClass = DATABASE_MAP[database_type] + db = DatabaseClass(database_name, **kwargs) + return Introspector.from_database(db, schema=schema) + +def print_models(introspector, tables=None, preserve_order=False, + include_views=False, ignore_unknown=False, snake_case=True): + database = introspector.introspect(table_names=tables, + include_views=include_views, + snake_case=snake_case) + + db_kwargs = introspector.get_database_kwargs() + header = HEADER % ( + introspector.get_additional_imports(), + introspector.get_database_class().__name__, + introspector.get_database_name(), + ', **%s' % repr(db_kwargs) if db_kwargs else '') + print_(header) + + if not ignore_unknown: + print_(UNKNOWN_FIELD) + + print_(BASE_MODEL) + + def _print_table(table, seen, accum=None): + accum = accum or [] + foreign_keys = database.foreign_keys[table] + for foreign_key in foreign_keys: + dest = foreign_key.dest_table + + # In the event the destination table has already been pushed + # for printing, then we have a reference cycle. + if dest in accum and table not in accum: + print_('# Possible reference cycle: %s' % dest) + + # If this is not a self-referential foreign key, and we have + # not already processed the destination table, do so now. + if dest not in seen and dest not in accum: + seen.add(dest) + if dest != table: + _print_table(dest, seen, accum + [table]) + + print_('class %s(BaseModel):' % database.model_names[table]) + columns = database.columns[table].items() + if not preserve_order: + columns = sorted(columns) + primary_keys = database.primary_keys[table] + for name, column in columns: + skip = all([ + name in primary_keys, + name == 'id', + len(primary_keys) == 1, + column.field_class in introspector.pk_classes]) + if skip: + continue + if column.primary_key and len(primary_keys) > 1: + # If we have a CompositeKey, then we do not want to explicitly + # mark the columns as being primary keys. + column.primary_key = False + + is_unknown = column.field_class is UnknownField + if is_unknown and ignore_unknown: + disp = '%s - %s' % (column.name, column.raw_column_type or '?') + print_(' # %s' % disp) + else: + print_(' %s' % column.get_field()) + + print_('') + print_(' class Meta:') + print_(' table_name = \'%s\'' % table) + multi_column_indexes = database.multi_column_indexes(table) + if multi_column_indexes: + print_(' indexes = (') + for fields, unique in sorted(multi_column_indexes): + print_(' ((%s), %s),' % ( + ', '.join("'%s'" % field for field in fields), + unique, + )) + print_(' )') + + if introspector.schema: + print_(' schema = \'%s\'' % introspector.schema) + if len(primary_keys) > 1: + pk_field_names = sorted([ + field.name for col, field in columns + if col in primary_keys]) + pk_list = ', '.join("'%s'" % pk for pk in pk_field_names) + print_(' primary_key = CompositeKey(%s)' % pk_list) + elif not primary_keys: + print_(' primary_key = False') + print_('') + + seen.add(table) + + seen = set() + for table in sorted(database.model_names.keys()): + if table not in seen: + if not tables or table in tables: + _print_table(table, seen) + +def print_header(cmd_line, introspector): + timestamp = datetime.datetime.now() + print_('# Code generated by:') + print_('# python -m pwiz %s' % cmd_line) + print_('# Date: %s' % timestamp.strftime('%B %d, %Y %I:%M%p')) + print_('# Database: %s' % introspector.get_database_name()) + print_('# Peewee version: %s' % peewee_version) + print_('') + + +def err(msg): + sys.stderr.write('\033[91m%s\033[0m\n' % msg) + sys.stderr.flush() + +def get_option_parser(): + parser = OptionParser(usage='usage: %prog [options] database_name') + ao = parser.add_option + ao('-H', '--host', dest='host') + ao('-p', '--port', dest='port', type='int') + ao('-u', '--user', dest='user') + ao('-P', '--password', dest='password', action='store_true') + engines = sorted(DATABASE_MAP) + ao('-e', '--engine', dest='engine', choices=engines, + help=('Database type, e.g. sqlite, mysql, postgresql or cockroachdb. ' + 'Default is "postgresql".')) + ao('-s', '--schema', dest='schema') + ao('-t', '--tables', dest='tables', + help=('Only generate the specified tables. Multiple table names should ' + 'be separated by commas.')) + ao('-v', '--views', dest='views', action='store_true', + help='Generate model classes for VIEWs in addition to tables.') + ao('-i', '--info', dest='info', action='store_true', + help=('Add database information and other metadata to top of the ' + 'generated file.')) + ao('-o', '--preserve-order', action='store_true', dest='preserve_order', + help='Model definition column ordering matches source table.') + ao('-I', '--ignore-unknown', action='store_true', dest='ignore_unknown', + help='Ignore fields whose type cannot be determined.') + ao('-L', '--legacy-naming', action='store_true', dest='legacy_naming', + help='Use legacy table- and column-name generation.') + return parser + +def get_connect_kwargs(options): + ops = ('host', 'port', 'user', 'schema') + kwargs = dict((o, getattr(options, o)) for o in ops if getattr(options, o)) + if options.password: + kwargs['password'] = getpass() + return kwargs + + +if __name__ == '__main__': + raw_argv = sys.argv + + parser = get_option_parser() + options, args = parser.parse_args() + + if len(args) < 1: + err('Missing required parameter "database"') + parser.print_help() + sys.exit(1) + + connect = get_connect_kwargs(options) + database = args[-1] + + tables = None + if options.tables: + tables = [table.strip() for table in options.tables.split(',') + if table.strip()] + + engine = options.engine + if engine is None: + engine = 'sqlite' if os.path.exists(database) else 'postgresql' + + introspector = make_introspector(engine, database, **connect) + if options.info: + cmd_line = ' '.join(raw_argv[1:]) + print_header(cmd_line, introspector) + + print_models(introspector, tables, options.preserve_order, options.views, + options.ignore_unknown, not options.legacy_naming) diff --git a/pilot-env/bin/python b/pilot-env/bin/python new file mode 120000 index 0000000..6e7f3c7 --- /dev/null +++ b/pilot-env/bin/python @@ -0,0 +1 @@ +python3.11 \ No newline at end of file diff --git a/pilot-env/bin/python3 b/pilot-env/bin/python3 new file mode 120000 index 0000000..6e7f3c7 --- /dev/null +++ b/pilot-env/bin/python3 @@ -0,0 +1 @@ +python3.11 \ No newline at end of file diff --git a/pilot-env/bin/python3.11 b/pilot-env/bin/python3.11 new file mode 120000 index 0000000..3cf1fbd --- /dev/null +++ b/pilot-env/bin/python3.11 @@ -0,0 +1 @@ +/opt/homebrew/opt/python@3.11/bin/python3.11 \ No newline at end of file diff --git a/pilot-env/pyvenv.cfg b/pilot-env/pyvenv.cfg new file mode 100644 index 0000000..94293fc --- /dev/null +++ b/pilot-env/pyvenv.cfg @@ -0,0 +1,5 @@ +home = /opt/homebrew/opt/python@3.11/bin +include-system-site-packages = false +version = 3.11.4 +executable = /opt/homebrew/Cellar/python@3.11/3.11.4/Frameworks/Python.framework/Versions/3.11/bin/python3.11 +command = /opt/homebrew/opt/python@3.11/bin/python3.11 -m venv /Users/duynguyen/www/gpt-pilot/pilot-env diff --git a/pilot/utils/llm_connection.py b/pilot/utils/llm_connection.py index d72515e..4ef1c91 100644 --- a/pilot/utils/llm_connection.py +++ b/pilot/utils/llm_connection.py @@ -90,7 +90,7 @@ def num_tokens_from_functions(functions, model=model): def create_gpt_chat_completion(messages: List[dict], req_type, min_tokens=MIN_TOKENS_FOR_GPT_RESPONSE, function_calls=None): gpt_data = { - 'model': os.getenv('OPENAI_MODEL', 'gpt-4'), + 'model': os.getenv('MODEL_NAME', 'gpt-4'), 'n': 1, 'max_tokens': 4096, 'temperature': 1, @@ -101,6 +101,13 @@ def create_gpt_chat_completion(messages: List[dict], req_type, min_tokens=MIN_TO 'stream': True } + # delete some keys if using "OpenRouter" API + if os.getenv('ENDPOINT') == "OPENROUTER": + keys_to_delete = ['n', 'max_tokens', 'temperature', 'top_p', 'presence_penalty', 'frequency_penalty'] + for key in keys_to_delete: + if key in gpt_data: + del gpt_data[key] + if function_calls is not None: gpt_data['functions'] = function_calls['definitions'] if len(function_calls['definitions']) > 1: @@ -188,10 +195,14 @@ def stream_gpt_completion(data, req_type): # If yes, get the AZURE_ENDPOINT from .ENV file endpoint_url = os.getenv('AZURE_ENDPOINT') + '/openai/deployments/' + model + '/chat/completions?api-version=2023-05-15' headers = {'Content-Type': 'application/json', 'api-key': os.getenv('AZURE_API_KEY')} + elif endpoint == 'OPENROUTER': + # If so, send the request to the OpenRouter API endpoint + headers = {'Content-Type': 'application/json', 'Authorization': 'Bearer ' + os.getenv("OPENROUTER_API_KEY"), 'HTTP-Referer': 'http://localhost:3000', 'X-Title': 'GPT Pilot (LOCAL)'} + endpoint_url = os.getenv("OPENROUTER_ENDPOINT") or 'https://openrouter.ai/api/v1/chat/completions' else: # If not, send the request to the OpenAI endpoint headers = {'Content-Type': 'application/json', 'Authorization': 'Bearer ' + os.getenv("OPENAI_API_KEY")} - endpoint_url = 'https://api.openai.com/v1/chat/completions' + endpoint_url = os.getenv("OPENAI_ENDPOINT") or 'https://api.openai.com/v1/chat/completions' response = requests.post( endpoint_url,