|
@@ -29,18 +29,26 @@ class GiteeAILargeLanguageModel(OAIAPICompatLargeLanguageModel):
|
|
|
user: Optional[str] = None,
|
|
|
) -> Union[LLMResult, Generator]:
|
|
|
self._add_custom_parameters(credentials, model, model_parameters)
|
|
|
- return super()._invoke(model, credentials, prompt_messages, model_parameters, tools, stop, stream, user)
|
|
|
+ return super()._invoke(
|
|
|
+ GiteeAILargeLanguageModel.MODEL_TO_IDENTITY.get(model, model),
|
|
|
+ credentials,
|
|
|
+ prompt_messages,
|
|
|
+ model_parameters,
|
|
|
+ tools,
|
|
|
+ stop,
|
|
|
+ stream,
|
|
|
+ user,
|
|
|
+ )
|
|
|
|
|
|
def validate_credentials(self, model: str, credentials: dict) -> None:
|
|
|
- self._add_custom_parameters(credentials, None)
|
|
|
- super().validate_credentials(model, credentials)
|
|
|
+ self._add_custom_parameters(credentials, model, None)
|
|
|
+ super().validate_credentials(GiteeAILargeLanguageModel.MODEL_TO_IDENTITY.get(model, model), credentials)
|
|
|
|
|
|
- def _add_custom_parameters(self, credentials: dict, model: Optional[str]) -> None:
|
|
|
+ def _add_custom_parameters(self, credentials: dict, model: Optional[str], model_parameters: dict) -> None:
|
|
|
if model is None:
|
|
|
model = "Qwen2-72B-Instruct"
|
|
|
|
|
|
- model_identity = GiteeAILargeLanguageModel.MODEL_TO_IDENTITY.get(model, model)
|
|
|
- credentials["endpoint_url"] = f"https://ai.gitee.com/api/serverless/{model_identity}/"
|
|
|
+ credentials["endpoint_url"] = "https://ai.gitee.com/v1"
|
|
|
if model.endswith("completions"):
|
|
|
credentials["mode"] = LLMMode.COMPLETION.value
|
|
|
else:
|