update

baidubce · Dec 27, 2024 · a776d8a · a776d8a
1 parent e196696
commit a776d8a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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):