jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: remove openllm pypi package because of this package too large (#931)

Browse source
This commit is contained in:
takatost 2023-08-21 02:12:28 +08:00 • committed by GitHub
commit 6c832ee328
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 97 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/model_providers/models/llm/openllm_model.py
View file

@ -1,13 +1,13 @@
from typing import List, Optional, Any
from langchain.callbacks.manager import Callbacks
from langchain.llms import OpenLLM
from langchain.schema import LLMResult
from core.model_providers.error import LLMBadRequestError
from core.model_providers.models.llm.base import BaseLLM
from core.model_providers.models.entity.message import PromptMessage
from core.model_providers.models.entity.model_params import ModelMode, ModelKwargs
from core.third_party.langchain.llms.openllm import OpenLLM
class OpenLLMModel(BaseLLM):
@ -19,7 +19,7 @@ class OpenLLMModel(BaseLLM):
client = OpenLLM(
server_url=self.credentials.get('server_url'),
callbacks=self.callbacks,
**self.provider_model_kwargs
llm_kwargs=self.provider_model_kwargs
)
return client