Skip to content

Commit d048a37

Browse files
committed
make configuration handling to streamline model and provider extraction from ai_config
1 parent f0e8aa0 commit d048a37

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

packages/ai-providers/server-ai-langchain/src/ldai_langchain/langchain_provider.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -228,14 +228,14 @@ def create_langchain_model(ai_config: AIConfigKind) -> BaseChatModel:
228228
"""
229229
from langchain.chat_models import init_chat_model
230230

231-
model_name = ai_config.model.name if ai_config.model else ''
232-
provider = ai_config.provider.name if ai_config.provider else ''
233-
parameters = {}
231+
config_dict = ai_config.to_dict()
232+
model_dict = config_dict.get('model') or {}
233+
provider_dict = config_dict.get('provider') or {}
234234

235-
if ai_config.model and hasattr(ai_config.model, '_parameters') and ai_config.model._parameters:
236-
parameters = ai_config.model._parameters.copy()
235+
model_name = model_dict.get('name', '')
236+
provider = provider_dict.get('name', '')
237+
parameters = model_dict.get('parameters') or {}
237238

238-
# Use LangChain's universal init_chat_model to support multiple providers
239239
return init_chat_model(
240240
model_name,
241241
model_provider=LangChainProvider.map_provider(provider),

0 commit comments

Comments
 (0)