Merge pull request #208 from Scale3-Labs/ali/url-fix

fix api host

src/langtrace_python_sdk/extensions/langtrace_filesystem.py

@@ -27,6 +27,8 @@ def __new__(cls, value):
 
 
 class LangTraceFile(io.BytesIO):
+    _host: str = os.environ.get("LANGTRACE_API_HOST", None) or LANGTRACE_REMOTE_URL
+
     def __init__(self, fs: "LangTraceFileSystem", path: str, mode: OpenMode):
         super().__init__()
         self.fs = fs
@@ -65,7 +67,7 @@ def upload_to_server(self, file_data: bytes) -> None:
             else:
                 print(Fore.GREEN + "Sending results to Langtrace" + Fore.RESET)
             response = requests.post(
-                url=f"{LANGTRACE_REMOTE_URL}/api/run",
+                url=f"{self._host}/api/run",
                 data=json.dumps(data),
                 headers={
                     "Content-Type": "application/json",
@@ -80,6 +82,7 @@ def upload_to_server(self, file_data: bytes) -> None:
 
 
 class LangTraceFileSystem(AbstractFileSystem):
+    _host: str = os.environ.get("LANGTRACE_API_HOST", None) or LANGTRACE_REMOTE_URL
     protocol = "langtracefs"
     sep = "/"
 
@@ -112,7 +115,7 @@ def fetch_file_from_api(self, dataset_id: str) -> bytes:
                 + Fore.RESET
             )
             response = requests.get(
-                url=f"{LANGTRACE_REMOTE_URL}/api/dataset/download?id={dataset_id}",
+                url=f"{self._host}/api/dataset/download?id={dataset_id}",
                 headers={
                     "Content-Type": "application/json",
                     "x-api-key": os.environ.get("LANGTRACE_API_KEY"),

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "2.1.19"
+__version__ = "2.1.20"