diff --git a/pkg/api/http/service/model.py b/pkg/api/http/service/model.py index afeae3eb..3a4998e2 100644 --- a/pkg/api/http/service/model.py +++ b/pkg/api/http/service/model.py @@ -185,7 +185,6 @@ class EmbeddingModelsService: runtime_embedding_model = await self.ap.model_mgr.init_runtime_embedding_model(model_data) await runtime_embedding_model.requester.invoke_embedding( - query=None, model=runtime_embedding_model, input_text='Hello, world!', extra_args={}, diff --git a/pkg/provider/modelmgr/requester.py b/pkg/provider/modelmgr/requester.py index 9742a52c..0abebfa5 100644 --- a/pkg/provider/modelmgr/requester.py +++ b/pkg/provider/modelmgr/requester.py @@ -100,7 +100,6 @@ class ProviderAPIRequester(metaclass=abc.ABCMeta): async def invoke_embedding( self, - query: core_entities.Query, model: RuntimeEmbeddingModel, input_text: str, extra_args: dict[str, typing.Any] = {}, diff --git a/pkg/provider/modelmgr/requesters/chatcmpl.py b/pkg/provider/modelmgr/requesters/chatcmpl.py index 98d1f13a..5dadab7d 100644 --- a/pkg/provider/modelmgr/requesters/chatcmpl.py +++ b/pkg/provider/modelmgr/requesters/chatcmpl.py @@ -144,7 +144,6 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester): async def invoke_embedding( self, - query: core_entities.Query, model: requester.RuntimeEmbeddingModel, input_text: str, extra_args: dict[str, typing.Any] = {},