codemie-sdk-python 0.1.150__py3-none-any.whl → 0.1.152__py3-none-any.whl

codemie_sdk/auth/credentials.py

@@ -1,7 +1,5 @@
 """Authentication credentials module for CodeMie SDK."""
 
-import os
-
 import requests
 from typing import Optional
 
@@ -38,17 +36,13 @@ class KeycloakCredentials:
         self.password = password
         self.verify_ssl = verify_ssl
 
-        if os.getenv("ENV", "").lower() != "local" and not (
-            (client_id and client_secret) or (username and password)
-        ):
+        if not ((client_id and client_secret) or (username and password)):
             raise ValueError(
                 "Either client credentials (client_id, client_secret) or "
                 "user credentials (username, password) must be provided"
             )
 
     def get_token(self) -> str:
-        if os.getenv("ENV", "").lower() == "local":
-            return ""
         """Get access token using either client credentials or password grant."""
         url = (
             f"{self.server_url}/realms/{self.realm_name}/protocol/openid-connect/token"

codemie_sdk/client/client.py

@@ -51,6 +51,7 @@ class CodeMieClient:
 
         self._token: Optional[str] = None
         self._api_domain = codemie_api_domain.rstrip("/")
+        self._is_localhost = self._is_localhost_domain(self._api_domain)
         self._verify_ssl = verify_ssl
         if not verify_ssl:
             import requests
@@ -58,8 +59,8 @@ class CodeMieClient:
 
             requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
 
-        # Initialize token first
-        self._token = self.auth.get_token()
+        # Initialize token
+        self._token = "" if self._is_localhost else self.auth.get_token()
 
         # Initialize services with verify_ssl parameter and token
         self.assistants = AssistantService(
@@ -84,13 +85,23 @@ class CodeMieClient:
     @property
     def token(self) -> str:
         """Get current token or fetch new one if not available."""
-        if not self._token:
-            self._token = self.auth.get_token()
+        self._token = "" if self._is_localhost else self.auth.get_token()
         return self._token
 
+    @staticmethod
+    def _is_localhost_domain(domain: str) -> bool:
+        """Check if the domain is a localhost variant."""
+        domain_lower = domain.lower()
+        localhost_patterns = [
+            "localhost",
+            "127.0.0.1",
+            "0.0.0.0",
+        ]
+        return any(pattern in domain_lower for pattern in localhost_patterns)
+
     def refresh_token(self) -> str:
         """Force token refresh."""
-        self._token = self.auth.get_token()
+        self._token = "" if self._is_localhost else self.auth.get_token()
         # Update token in services
         self.assistants = AssistantService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl

codemie_sdk/utils/http.py

@@ -1,6 +1,5 @@
 """HTTP utilities for CodeMie SDK."""
 
-import os
 from typing import TypeVar, Type, Optional, Any, Union, Dict, List, get_origin, get_args
 from pydantic import BaseModel
 import requests
@@ -46,6 +45,18 @@ class ApiRequestHandler:
         self._base_url = base_url.rstrip("/")
         self._token = token
         self._verify_ssl = verify_ssl
+        self._is_localhost = self._is_localhost_domain(base_url)
+
+    @staticmethod
+    def _is_localhost_domain(domain: str) -> bool:
+        """Check if the domain is a localhost variant."""
+        domain_lower = domain.lower()
+        localhost_patterns = [
+            "localhost",
+            "127.0.0.1",
+            "0.0.0.0",
+        ]
+        return any(pattern in domain_lower for pattern in localhost_patterns)
 
     def _get_headers(self, exclude_content_type: bool = False) -> dict:
         """Gets request headers with auth token.
@@ -58,9 +69,7 @@ class ApiRequestHandler:
             headers["Content-Type"] = "application/json"
 
         headers["Authorization"] = (
-            f"Bearer {self._token}"
-            if os.getenv("ENV", "").lower() != "local"
-            else "dev-codemie-user"
+            f"Bearer {self._token}" if not self._is_localhost else "dev-codemie-user"
         )
         return headers
 

{codemie_sdk_python-0.1.150.dist-info → codemie_sdk_python-0.1.152.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.150
+Version: 0.1.152
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com

{codemie_sdk_python-0.1.150.dist-info → codemie_sdk_python-0.1.152.dist-info}/RECORD

@@ -1,8 +1,8 @@
 codemie_sdk/__init__.py,sha256=leeACpTc2113BsY_IgO-ceFHPrQKC2WhuOfuuyyIqDE,567
 codemie_sdk/auth/__init__.py,sha256=IksEj223xEZtJ-cQ0AT9L0Bs9psIJ8QNzDXrPTUQ3xQ,126
-codemie_sdk/auth/credentials.py,sha256=vcvt2nvb4_tSCdR7On498wNW4D-DRIuEJU6l3QG8QN0,4471
+codemie_sdk/auth/credentials.py,sha256=u2eLNsD8fELTgreQghVb0g0kk82zchUkux0M5lzq-u8,4320
 codemie_sdk/client/__init__.py,sha256=yf6C39MmrJ6gK9ZHMhBeynKwUUYVSUTQbKxU8-4qpKg,101
-codemie_sdk/client/client.py,sha256=8XlFsa2OEKdawipbUcX3L7umVHXKh9nzQ5-CXsXpMXI,4691
+codemie_sdk/client/client.py,sha256=6yXXNYF4m1vh_0UMPENaxL39hmt_wdPF_b5NtsKMJQQ,5174
 codemie_sdk/exceptions.py,sha256=XoVPyognx-JmyVxLHkZPAcX1CMi1OoT1diBFJLU54so,1183
 codemie_sdk/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_sdk/models/assistant.py,sha256=w0Oc4ArB-Rf6KiBjhVMVJgbmXpd48Tdt-dGLJJ7qJHw,10369
@@ -26,7 +26,7 @@ codemie_sdk/services/workflow.py,sha256=0d5grAsBIH2NZY66UbVhlDqEFYxEYSBByj3GNgcz
 codemie_sdk/services/workflow_execution.py,sha256=aGoT3rdTmh5-doAsrmBBjLEuOfvL5aqeo3g9th1_aAw,3647
 codemie_sdk/services/workflow_execution_state.py,sha256=tXoaa8yT09xgYEUNiHhVULe76TwGwVgZupMIUyyLxdo,2070
 codemie_sdk/utils/__init__.py,sha256=BXAJJfAzO89-kMYvWWo9wSNhSbGgF3vB1In9sePFhMM,109
-codemie_sdk/utils/http.py,sha256=sdUdMYtexkYz0dQ7ysdwJlyYaE-QHdTZekQNDJpX48s,9134
-codemie_sdk_python-0.1.150.dist-info/METADATA,sha256=BJvuYsxOxDmvFjDTXsEf06M6LeuP8KVT1Ah38_Lj0Hs,24494
-codemie_sdk_python-0.1.150.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.150.dist-info/RECORD,,
+codemie_sdk/utils/http.py,sha256=ZtXb-hlwqXmjSTKRncgO-9MYoXnmHt2Dljno0kPy7mA,9506
+codemie_sdk_python-0.1.152.dist-info/METADATA,sha256=v_XHbDe_rIZkhrsXeWuDxpeqJaDCiTRhbCUAvdmT6iM,24494
+codemie_sdk_python-0.1.152.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.152.dist-info/RECORD,,