Merge pull request #2 from rkihacker/main

remove model prefix for claude
This commit is contained in:
kqlio67 2024-11-02 17:06:04 +00:00 committed by GitHub
commit e030923fd6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -29,28 +29,46 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
image_models = ['ImageGeneration']
models = [
default_model,
'blackboxai-pro',
*image_models,
"llama-3.1-8b",
'llama-3.1-70b',
'llama-3.1-405b',
'gpt-4o',
'gemini-pro',
'gemini-1.5-flash',
'claude-sonnet-3.5',
'PythonAgent',
'JavaAgent',
'JavaScriptAgent',
'HTMLAgent',
'GoogleCloudAgent',
'AndroidDeveloper',
'SwiftDeveloper',
'Next.jsAgent',
'MongoDBAgent',
'PyTorchAgent',
'ReactAgent',
'XcodeAgent',
'AngularJSAgent',
'blackboxai-pro',
*image_models,
"llama-3.1-8b",
'llama-3.1-70b',
'llama-3.1-405b',
'gpt-4o',
'gemini-pro',
'gemini-1.5-flash',
'claude-sonnet-3.5',
'PythonAgent',
'JavaAgent',
'JavaScriptAgent',
'HTMLAgent',
'GoogleCloudAgent',
'AndroidDeveloper',
'SwiftDeveloper',
'Next.jsAgent',
'MongoDBAgent',
'PyTorchAgent',
'ReactAgent',
'XcodeAgent',
'AngularJSAgent',
'HerokuAgent',
'GodotAgent',
'GoAgent',
'GitlabAgent',
'GitAgent',
'RepoMap',
'FlaskAgent',
'FirebaseAgent',
'FastAPIAgent',
'ErlangAgent',
'ElectronAgent',
'DockerAgent',
'DigitalOceanAgent',
'BitbucketAgent',
'AzureAgent',
'FlutterAgent',
'YoutubeAgent',
'builderAgent',
]
agentMode = {
@ -58,6 +76,7 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
}
trendingAgentMode = {
"blackboxai": {},
"blackboxai": {},
"gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
"llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
@ -77,6 +96,24 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
'ReactAgent': {'mode': True, 'id': "React Agent"},
'XcodeAgent': {'mode': True, 'id': "Xcode Agent"},
'AngularJSAgent': {'mode': True, 'id': "AngularJS Agent"},
'HerokuAgent': {'mode': True, 'id': "Heroku Agent"},
'GodotAgent': {'mode': True, 'id': "Godot Agent"},
'GoAgent': {'mode': True, 'id': "Go Agent"},
'GitlabAgent': {'mode': True, 'id': "Gitlab Agent"},
'GitAgent': {'mode': True, 'id': "Git Agent"},
'RepoMap': {'mode': True, 'id': "RepoMap"},
'FlaskAgent': {'mode': True, 'id': "Flask Agent"},
'FirebaseAgent': {'mode': True, 'id': "Firebase Agent"},
'FastAPIAgent': {'mode': True, 'id': "FastAPI Agent"},
'ErlangAgent': {'mode': True, 'id': "Erlang Agent"},
'ElectronAgent': {'mode': True, 'id': "Electron Agent"},
'DockerAgent': {'mode': True, 'id': "Docker Agent"},
'DigitalOceanAgent': {'mode': True, 'id': "DigitalOcean Agent"},
'BitbucketAgent': {'mode': True, 'id': "Bitbucket Agent"},
'AzureAgent': {'mode': True, 'id': "Azure Agent"},
'FlutterAgent': {'mode': True, 'id': "Flutter Agent"},
'YoutubeAgent': {'mode': True, 'id': "Youtube Agent"},
'builderAgent': {'mode': True, 'id': "builder Agent"},
}
userSelectedModel = {
@ -88,7 +125,6 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
model_prefixes = {
'gpt-4o': '@GPT-4o',
'gemini-pro': '@Gemini-PRO',
'claude-sonnet-3.5': '@Claude-Sonnet-3.5',
'PythonAgent': '@Python Agent',
'JavaAgent': '@Java Agent',
'JavaScriptAgent': '@JavaScript Agent',
@ -102,8 +138,25 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
'ReactAgent': '@React Agent',
'XcodeAgent': '@Xcode Agent',
'AngularJSAgent': '@AngularJS Agent',
'HerokuAgent': '@Heroku Agent',
'GodotAgent': '@Godot Agent',
'GoAgent': '@Go Agent',
'GitlabAgent': '@Gitlab Agent',
'GitAgent': '@Git Agent',
'blackboxai-pro': '@BLACKBOXAI-PRO',
'ImageGeneration': '@Image Generation',
'FlaskAgent': '@Flask Agent',
'FirebaseAgent': '@Firebase Agent',
'FastAPIAgent': '@FastAPI Agent',
'ErlangAgent': '@Erlang Agent',
'ElectronAgent': '@Electron Agent',
'DockerAgent': '@Docker Agent',
'DigitalOceanAgent': '@DigitalOcean Agent',
'BitbucketAgent': '@Bitbucket Agent',
'AzureAgent': '@Azure Agent',
'FlutterAgent': '@Flutter Agent',
'YoutubeAgent': '@Youtube Agent',
'builderAgent': '@builder Agent',
}
model_referers = {