Skip to content

Commit 5876019

Browse files
Remove support for a few models
These models have annoyingly big dependencies or dependencies that seem to fail to build under certain Python environments, so removing support for them for now.
1 parent dab6083 commit 5876019

File tree

2 files changed

+1
-13
lines changed

2 files changed

+1
-13
lines changed

pyproject.toml

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "git-ai-summarize"
7-
version = "0.1.2"
7+
version = "0.1.3"
88
authors = [{ name = "Kevin Beaulieu", email = "[email protected]" }]
99
description = "AI-powered git commands for summarizing changes"
1010
readme = "README.md"
@@ -19,14 +19,11 @@ dependencies = [
1919
"langchain-anthropic",
2020
"langchain-openai",
2121
"langchain-google-genai",
22-
"langchain-cohere",
2322
"langchain-mistralai",
2423
"langchain-fireworks",
2524
"langchain-together",
2625
"langchain-google-vertexai",
2726
"langchain-groq",
28-
"langchain-aws",
29-
"langchain-huggingface",
3027
"langchain-nvidia-ai-endpoints",
3128
"langchain-ollama",
3229
"langchain-ai21",

src/git_ai_summarize/models.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,11 @@
33
from langchain_anthropic import ChatAnthropic
44
from langchain_openai import ChatOpenAI
55
from langchain_google_genai import ChatGoogleGenerativeAI
6-
from langchain_cohere import ChatCohere
76
from langchain_mistralai import ChatMistralAI
87
from langchain_fireworks import ChatFireworks
98
from langchain_together import ChatTogether
109
from langchain_google_vertexai import ChatVertexAI
1110
from langchain_groq import ChatGroq
12-
from langchain_aws import ChatBedrock
13-
from langchain_huggingface import ChatHuggingFace
1411
from langchain_nvidia_ai_endpoints import ChatNVIDIA
1512
from langchain_ollama import ChatOllama
1613
from langchain_ai21 import ChatAI21
@@ -27,14 +24,11 @@ def get_supported_providers() -> List[str]:
2724
"anthropic",
2825
"openai",
2926
"google",
30-
"cohere",
3127
"mistralai",
3228
"fireworks",
3329
"together",
3430
"vertexai",
3531
"groq",
36-
"bedrock",
37-
"huggingface",
3832
"nvidia_ai",
3933
"ollama",
4034
"ai21",
@@ -51,14 +45,11 @@ def get_model(provider_name: str | None, model_name: str | None) -> BaseChatMode
5145
"anthropic": (ChatAnthropic, "ANTHROPIC_API_KEY", "https://www.anthropic.com", "anthropic_api_key"),
5246
"openai": (ChatOpenAI, "OPENAI_API_KEY", "https://platform.openai.com/account/api-keys", "openai_api_key"),
5347
"google": (ChatGoogleGenerativeAI, "GOOGLE_API_KEY", "https://developers.generativeai.google/", "google_api_key"),
54-
"cohere": (ChatCohere, "COHERE_API_KEY", "https://dashboard.cohere.ai/api-keys", "cohere_api_key"),
5548
"mistral": (ChatMistralAI, "MISTRAL_API_KEY", "https://console.mistral.ai/api-keys/", "mistral_api_key"),
5649
"fireworks": (ChatFireworks, "FIREWORKS_API_KEY", "https://app.fireworks.ai/", "fireworks_api_key"),
5750
"together": (ChatTogether, "TOGETHER_API_KEY", "https://api.together.xyz/", "together_api_key"),
5851
"vertex": (ChatVertexAI, "GOOGLE_APPLICATION_CREDENTIALS", "https://cloud.google.com/vertex-ai", None),
5952
"groq": (ChatGroq, "GROQ_API_KEY", "https://console.groq.com/", "groq_api_key"),
60-
"bedrock": (ChatBedrock, "AWS_ACCESS_KEY_ID", "https://aws.amazon.com/bedrock/", None),
61-
"huggingface": (ChatHuggingFace, "HUGGINGFACE_API_KEY", "https://huggingface.co/settings/tokens", "huggingface_api_key"),
6253
"nvidia": (ChatNVIDIA, "NVIDIA_API_KEY", "https://api.nvidia.com/", "nvidia_api_key"),
6354
"ollama": (ChatOllama, None, "https://ollama.ai/", None),
6455
"ai21": (ChatAI21, "AI21_API_KEY", "https://www.ai21.com/studio", "ai21_api_key"),

0 commit comments

Comments
 (0)