update

InternLM · Oct 24, 2024 · 573c7b5 · 573c7b5
1 parent 5e6055c
commit 573c7b5
Showing 1 changed file with 10 additions and 12 deletions.
diff --git a/lagent/llms/openai.py b/lagent/llms/openai.py
@@ -243,7 +243,7 @@ def _chat(self, messages: List[dict], **gen_params) -> str:
                         response['error'])
                     self.logger.error(errmsg)
             except Exception as error:
-                errmsg = str(error) + ' ' + str(traceback.format_exc())
+                errmsg = str(error) + '\n' + str(traceback.format_exc())
                 self.logger.error(errmsg)
             max_num_retries += 1
 
@@ -293,10 +293,9 @@ def streaming(raw_response):
                                 return
                             yield choice['delta'].get('content', '')
                     except Exception as exc:
-                        print(
-                            f'response {decoded} lead to exception of {str(exc)}'
-                        )
-                        raise
+                        msg = f'response {decoded} lead to exception of {str(exc)}'
+                        self.logger.error(msg)
+                        raise Exception(msg) from exc
 
         assert isinstance(messages, list)
 
@@ -359,7 +358,7 @@ def streaming(raw_response):
                         response['error'])
                     self.logger.error(errmsg)
             except Exception as error:
-                errmsg = str(error) + ' ' + str(traceback.format_exc())
+                errmsg = str(error) + '\n' + str(traceback.format_exc())
                 self.logger.error(errmsg)
             max_num_retries += 1
 
@@ -694,7 +693,7 @@ async def _chat(self, messages: List[dict], **gen_params) -> str:
                         response['error'])
                     self.logger.error(errmsg)
             except Exception as error:
-                errmsg = str(error) + ' ' + str(traceback.format_exc())
+                errmsg = str(error) + '\n' + str(traceback.format_exc())
                 self.logger.error(errmsg)
             max_num_retries += 1
 
@@ -743,10 +742,9 @@ async def streaming(raw_response):
                                 return
                             yield choice['delta'].get('content', '')
                     except Exception as exc:
-                        self.logger.error(
-                            f'response {decoded} lead to exception of {str(exc)}'
-                        )
-                        raise
+                        msg = f'response {decoded} lead to exception of {str(exc)}'
+                        self.logger.error(msg)
+                        raise Exception(msg) from exc
 
         assert isinstance(messages, list)
 
@@ -814,7 +812,7 @@ async def streaming(raw_response):
                         response['error'])
                     self.logger.error(errmsg)
             except Exception as error:
-                errmsg = str(error) + ' ' + str(traceback.format_exc())
+                errmsg = str(error) + '\n' + str(traceback.format_exc())
                 self.logger.error(errmsg)
             max_num_retries += 1