fix: make model_provider call consistent in app_settings
Browse files- app.py +1 -1
- config.py +1 -0
- conversation/generate.py +1 -1
- pyproject.toml +1 -0
app.py
CHANGED
|
@@ -39,7 +39,7 @@ embeddings = BedrockEmbeddings(
|
|
| 39 |
|
| 40 |
llm = init_chat_model(
|
| 41 |
app_settings.llm_model,
|
| 42 |
-
model_provider=
|
| 43 |
region_name=app_settings.llm_region,
|
| 44 |
aws_access_key_id=app_settings.aws_access_key_id,
|
| 45 |
aws_secret_access_key=app_settings.aws_secret_access_key,
|
|
|
|
| 39 |
|
| 40 |
llm = init_chat_model(
|
| 41 |
app_settings.llm_model,
|
| 42 |
+
model_provider=app_settings.model_provider,
|
| 43 |
region_name=app_settings.llm_region,
|
| 44 |
aws_access_key_id=app_settings.aws_access_key_id,
|
| 45 |
aws_secret_access_key=app_settings.aws_secret_access_key,
|
config.py
CHANGED
|
@@ -30,6 +30,7 @@ class AppSettings(Settings):
|
|
| 30 |
# backend
|
| 31 |
llm_model: str
|
| 32 |
llm_region: str
|
|
|
|
| 33 |
embedding_model: str
|
| 34 |
embedding_size: int
|
| 35 |
vector_db_url: str
|
|
|
|
| 30 |
# backend
|
| 31 |
llm_model: str
|
| 32 |
llm_region: str
|
| 33 |
+
model_provider: str
|
| 34 |
embedding_model: str
|
| 35 |
embedding_size: int
|
| 36 |
vector_db_url: str
|
conversation/generate.py
CHANGED
|
@@ -20,7 +20,7 @@ logger = structlog.get_logger(__name__)
|
|
| 20 |
|
| 21 |
llm = init_chat_model(
|
| 22 |
app_settings.llm_model,
|
| 23 |
-
model_provider=
|
| 24 |
region_name=app_settings.llm_region,
|
| 25 |
aws_access_key_id=app_settings.aws_access_key_id,
|
| 26 |
aws_secret_access_key=app_settings.aws_secret_access_key,
|
|
|
|
| 20 |
|
| 21 |
llm = init_chat_model(
|
| 22 |
app_settings.llm_model,
|
| 23 |
+
model_provider=app_settings.model_provider,
|
| 24 |
region_name=app_settings.llm_region,
|
| 25 |
aws_access_key_id=app_settings.aws_access_key_id,
|
| 26 |
aws_secret_access_key=app_settings.aws_secret_access_key,
|
pyproject.toml
CHANGED
|
@@ -25,6 +25,7 @@ dependencies = [
|
|
| 25 |
# shared
|
| 26 |
llm_model = "anthropic.claude-3-5-sonnet-20240620-v1:0"
|
| 27 |
llm_region = "eu-central-1"
|
|
|
|
| 28 |
embedding_model = "cohere.embed-multilingual-v3"
|
| 29 |
embedding_size = 1024
|
| 30 |
vector_db_url = ":memory:"
|
|
|
|
| 25 |
# shared
|
| 26 |
llm_model = "anthropic.claude-3-5-sonnet-20240620-v1:0"
|
| 27 |
llm_region = "eu-central-1"
|
| 28 |
+
model_provider = "bedrock_converse"
|
| 29 |
embedding_model = "cohere.embed-multilingual-v3"
|
| 30 |
embedding_size = 1024
|
| 31 |
vector_db_url = ":memory:"
|