Added whatismyip

.idea/misc.xml

@@ -1,6 +1,9 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" languageLevel="JDK_21" default="true" project-jdk-name="21" project-jdk-type="JavaSDK">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.12 (LLM-Engine)" />
+  </component>
+  <component name="ProjectRootManager" version="2" languageLevel="JDK_21" default="true" project-jdk-name="Python 3.12 (LLM-Engine)" project-jdk-type="Python SDK">
     <output url="file://$PROJECT_DIR$/out" />
   </component>
 </project>

main/main.py

@@ -1,12 +1,13 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from typing import Optional, Dict, Any, Union
+from typing import Optional, Union
 import torch
 import logging
 from pathlib import Path
 from litgpt.api import LLM
 import os
 import uvicorn
+import whatismyip
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -169,6 +170,7 @@ def main():
     port = int(os.getenv("LLM_ENGINE_PORT", "8001"))
 
     # Start the server
+    print(whatismyip.whatismyip())
     uvicorn.run(
         app,
         host=host,

requirements.txt

@@ -63,3 +63,4 @@ typing_extensions==4.12.2
 urllib3==2.2.3
 uvicorn==0.27.0
 yarl==1.17.1
+whatismyip==2024.2.20