awesome-bookmarks/shell_gpt
3
1
Fork 0
mirror of https://github.com/TheR1D/shell_gpt.git synced 2026-06-02 06:14:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Adding new GPT models (#263)

Browse Source
This commit is contained in:
William C. Wright
2023-06-14 07:37:46 -07:00
committed by GitHub
parent cb9782b74d
commit fd210e9cb6
2 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sgpt/config.py
+1 -1
View File
@@ -23,7 +23,7 @@ DEFAULT_CONFIG = {
"CHAT_CACHE_LENGTH": int(os.getenv("CHAT_CACHE_LENGTH", "100")),
"CACHE_LENGTH": int(os.getenv("CHAT_CACHE_LENGTH", "100")),
"REQUEST_TIMEOUT": int(os.getenv("REQUEST_TIMEOUT", "60")),
"DEFAULT_MODEL": os.getenv("DEFAULT_MODEL", ModelOptions.GPT3.value),
"DEFAULT_MODEL": os.getenv("DEFAULT_MODEL", ModelOptions.GPT35TURBO.value),
"OPENAI_API_HOST": os.getenv("OPENAI_API_HOST", "https://api.openai.com"),
"DEFAULT_COLOR": os.getenv("DEFAULT_COLOR", "magenta"),
"ROLE_STORAGE_PATH": os.getenv("ROLE_STORAGE_PATH", str(ROLE_STORAGE_PATH)),
sgpt/utils.py
+15 -2
View File
@@ -10,9 +10,22 @@ from click import BadParameter
class ModelOptions(str, Enum):
GPT3 = "gpt-3.5-turbo"
"""
Model endpoint compatibility
https://platform.openai.com/docs/models/model-endpoint-compatibility
"""
GPT4 = "gpt-4"
GPT4_32K = "gpt-4-32k"
GPT40314 = "gpt-4-0314" # discontinued 2023-09-13
GPT40613 = "gpt-4-0613"
GPT432k = "gpt-4-32k"
GPT432k0314 = "gpt-4-32k-0314" # discontinued 2023-09-13
GPT432k0613 = "gpt-4-32k-0613"
GPT35TURBO = "gpt-3.5-turbo"
GPT35TURBO16K = "gpt-3.5-turbo-16k"
GPT35TURBO16K0613 = "gpt-3.5-turbo-16k-0613"
GPT35TURBO0301 = "gpt-3.5-turbo-0301" # discontinued 2023-09-13
GPT35TURBO0613 = "gpt-3.5-turbo-0613"
def get_edited_prompt() -> str: