AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 20:59:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(typo): misspelling (#5094)

Browse Source
This commit is contained in:
Pika 2024-06-12 17:01:21 +08:00 committed by GitHub
parent 95443bd551
commit ecc7f130b4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/model_runtime/model_providers/localai/llm/llm.py
View File

@ -50,7 +50,7 @@ from core.model_runtime.model_providers.__base.large_language_model import Large
from core.model_runtime.utils import helper from core.model_runtime.utils import helper
class LocalAILarguageModel(LargeLanguageModel): class LocalAILanguageModel(LargeLanguageModel):
def _invoke(self, model: str, credentials: dict, def _invoke(self, model: str, credentials: dict,
prompt_messages: list[PromptMessage], model_parameters: dict, prompt_messages: list[PromptMessage], model_parameters: dict,
tools: list[PromptMessageTool] | None = None, stop: list[str] | None = None, tools: list[PromptMessageTool] | None = None, stop: list[str] | None = None,

14
api/tests/integration_tests/model_runtime/localai/test_llm.py
View File

@ -13,11 +13,11 @@ from core.model_runtime.entities.message_entities import (
) )
from core.model_runtime.entities.model_entities import ParameterRule from core.model_runtime.entities.model_entities import ParameterRule
from core.model_runtime.errors.validate import CredentialsValidateFailedError from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.localai.llm.llm import LocalAILarguageModel from core.model_runtime.model_providers.localai.llm.llm import LocalAILanguageModel
def test_validate_credentials_for_chat_model(): def test_validate_credentials_for_chat_model():
model = LocalAILarguageModel() model = LocalAILanguageModel()
with pytest.raises(CredentialsValidateFailedError): with pytest.raises(CredentialsValidateFailedError):
model.validate_credentials( model.validate_credentials(
@ -37,7 +37,7 @@ def test_validate_credentials_for_chat_model():
) )
def test_invoke_completion_model(): def test_invoke_completion_model():
model = LocalAILarguageModel() model = LocalAILanguageModel()
response = model.invoke( response = model.invoke(
model='chinese-llama-2-7b', model='chinese-llama-2-7b',
@ -65,7 +65,7 @@ def test_invoke_completion_model():
assert response.usage.total_tokens > 0 assert response.usage.total_tokens > 0
def test_invoke_chat_model(): def test_invoke_chat_model():
model = LocalAILarguageModel() model = LocalAILanguageModel()
response = model.invoke( response = model.invoke(
model='chinese-llama-2-7b', model='chinese-llama-2-7b',
@ -93,7 +93,7 @@ def test_invoke_chat_model():
assert response.usage.total_tokens > 0 assert response.usage.total_tokens > 0
def test_invoke_stream_completion_model(): def test_invoke_stream_completion_model():
model = LocalAILarguageModel() model = LocalAILanguageModel()
response = model.invoke( response = model.invoke(
model='chinese-llama-2-7b', model='chinese-llama-2-7b',
@ -124,7 +124,7 @@ def test_invoke_stream_completion_model():
assert len(chunk.delta.message.content) > 0 if chunk.delta.finish_reason is None else True assert len(chunk.delta.message.content) > 0 if chunk.delta.finish_reason is None else True
def test_invoke_stream_chat_model(): def test_invoke_stream_chat_model():
model = LocalAILarguageModel() model = LocalAILanguageModel()
response = model.invoke( response = model.invoke(
model='chinese-llama-2-7b', model='chinese-llama-2-7b',
@ -155,7 +155,7 @@ def test_invoke_stream_chat_model():
assert len(chunk.delta.message.content) > 0 if chunk.delta.finish_reason is None else True assert len(chunk.delta.message.content) > 0 if chunk.delta.finish_reason is None else True
def test_get_num_tokens(): def test_get_num_tokens():
model = LocalAILarguageModel() model = LocalAILanguageModel()
num_tokens = model.get_num_tokens( num_tokens = model.get_num_tokens(
model='????', model='????',