diff --git a/python/core/components/llms/base.py b/python/core/components/llms/base.py index 27931789..346a4774 100644 --- a/python/core/components/llms/base.py +++ b/python/core/components/llms/base.py @@ -432,6 +432,9 @@ def completion( url = self.http_client.service_url(completion_url, self.base_url) response = self.http_client.session.post(url, json=request.params, headers=headers, timeout=timeout, stream=stream) + + return self.gene_response(response, stream) + @staticmethod def check_service_error(data: dict):