youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 01:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

add llm node

Browse Source
This commit is contained in:
takatost
2024-03-12 22:12:03 +08:00
parent 4f5c052dc8
commit 3f59a579d7
17 changed files with 697 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/model_manager.py
View File

@ -24,11 +24,11 @@ class ModelInstance:
"""
def __init__(self, provider_model_bundle: ProviderModelBundle, model: str) -> None:
self._provider_model_bundle = provider_model_bundle
self.provider_model_bundle = provider_model_bundle
self.model = model
self.provider = provider_model_bundle.configuration.provider.provider
self.credentials = self._fetch_credentials_from_bundle(provider_model_bundle, model)
self.model_type_instance = self._provider_model_bundle.model_type_instance
self.model_type_instance = self.provider_model_bundle.model_type_instance
def _fetch_credentials_from_bundle(self, provider_model_bundle: ProviderModelBundle, model: str) -> dict:
"""