djcotto commited on
Commit
7e508e0
·
1 Parent(s): 6fa6cfb

fix: make model_provider call consistent in app_settings

Browse files
Files changed (4) hide show
  1. app.py +1 -1
  2. config.py +1 -0
  3. conversation/generate.py +1 -1
  4. 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="bedrock_converse",
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="bedrock_converse",
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:"