Browse Source

Ignore any warnings LiteLLM might emit on import (#1687)

மனோஜ்குமார் பழனிச்சாமி 1 year ago
parent
commit
b4cdebec06
2 changed files with 10 additions and 2 deletions
  1. 5 1
      opendevin/llm/llm.py
  2. 5 1
      opendevin/server/listen.py

+ 5 - 1
opendevin/llm/llm.py

@@ -1,6 +1,10 @@
 from functools import partial
 
-import litellm
+import warnings
+
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
+    import litellm
 from litellm import completion as litellm_completion
 from litellm import completion_cost as litellm_completion_cost
 from litellm.exceptions import (

+ 5 - 1
opendevin/server/listen.py

@@ -3,7 +3,11 @@ import shutil
 import uuid
 from pathlib import Path
 
-import litellm
+import warnings
+
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
+    import litellm
 from fastapi import Depends, FastAPI, Response, UploadFile, WebSocket, status
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse, RedirectResponse