codemie-sdk-python 0.1.427__tar.gz → 0.1.428__tar.gz

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/PKG-INFO

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

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.427"
+version = "0.1.428"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/auth/credentials.py

@@ -1,12 +1,15 @@
 """Authentication credentials module for CodeMie SDK."""
 
+import time
 import requests
-from typing import Optional
+from typing import Callable, Optional, Union
 
 
 class KeycloakCredentials:
     """Keycloak authentication credentials handler."""
 
+    _TOKEN_EXPIRY_BUFFER_SECONDS = 30
+
     def __init__(
         self,
         server_url: str,
@@ -15,7 +18,7 @@ class KeycloakCredentials:
         client_secret: Optional[str] = None,
         username: Optional[str] = None,
         password: Optional[str] = None,
-        external_token: Optional[str] = None,
+        external_token: Optional[Union[str, Callable[[], str]]] = None,
         external_idp: Optional[str] = None,
         verify_ssl: bool = True,
     ):
@@ -41,9 +44,15 @@ class KeycloakCredentials:
         self.external_token = external_token
         self.external_idp = external_idp
         self.verify_ssl = verify_ssl
+        self._cached_token: Optional[str] = None
+        self._token_expires_at: float = 0.0
 
     def get_token(self) -> str:
-        """Get access token using either client credentials or password grant."""
+        """Get access token, returning cached value if still valid."""
+        now = time.monotonic()
+        if self._cached_token and now < self._token_expires_at:
+            return self._cached_token
+
         if not (
             (self.client_id and self.client_secret)
             or (self.username and self.password)
@@ -57,28 +66,31 @@ class KeycloakCredentials:
             f"{self.server_url}/realms/{self.realm_name}/protocol/openid-connect/token"
         )
 
-        if self.username and self.password:
-            # Use Resource Owner Password Credentials flow
-            payload = {
-                "grant_type": "password",
-                "username": self.username,
-                "password": self.password,
-                "client_id": self.client_id
-                or "codemie-sdk",  # Use default client if not specified
-            }
-            if self.client_secret:
-                payload["client_secret"] = self.client_secret
-        elif self.external_token and self.external_idp:
+        if self.external_token and self.external_idp:
+            subject_token = (
+                self.external_token()
+                if callable(self.external_token)
+                else self.external_token
+            )
             payload = {
                 "client_id": self.client_id,
                 "client_secret": self.client_secret,
                 "grant_type": "urn:ietf:params:oauth:grant-type:token-exchange",
-                "subject_token": self.external_token,
+                "subject_token": subject_token,
+                "subject_token_type": "urn:ietf:params:oauth:token-type:access_token",
                 "subject_issuer": self.external_idp,
                 "requested_token_type": "urn:ietf:params:oauth:token-type:access_token",
             }
+        elif self.username and self.password:
+            payload = {
+                "grant_type": "password",
+                "username": self.username,
+                "password": self.password,
+                "client_id": self.client_id or "codemie-sdk",
+            }
+            if self.client_secret:
+                payload["client_secret"] = self.client_secret
         else:
-            # Use Client Credentials flow
             payload = {
                 "grant_type": "client_credentials",
                 "client_id": self.client_id,
@@ -87,7 +99,11 @@ class KeycloakCredentials:
 
         response = requests.post(url, data=payload, verify=self.verify_ssl)
         response.raise_for_status()
-        return response.json()["access_token"]
+        data = response.json()
+        self._cached_token = data["access_token"]
+        expires_in = data.get("expires_in", 300)
+        self._token_expires_at = now + expires_in - self._TOKEN_EXPIRY_BUFFER_SECONDS
+        return self._cached_token
 
     def exchange_token_for_user(self, email: str, access_token: str) -> str:
         """Exchange service account token for user token."""
@@ -118,7 +134,7 @@ class KeycloakCredentials:
         payload = {
             "client_id": self.client_id,
             "client_secret": self.client_secret,
-            "grant_type": "urn:ietf:params:oauth:grant-datasource_type:token-exchange",
+            "grant_type": "urn:ietf:params:oauth:grant-type:token-exchange",
             "subject_token": service_account_token,
             "requested_subject": user_id,
         }

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/client/client.py

@@ -78,62 +78,68 @@ class CodeMieClient:
 
             requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
 
-        # Initialize token
+        # Initialize token; keep string copy for backward-compat callers reading _token directly
         self._token = "" if self._is_localhost else self.auth.get_token()
+        # Services receive the bound method so each request resolves a fresh (cached) token
+        _token_source = "" if self._is_localhost else self.auth.get_token
 
         # Initialize services with verify_ssl parameter and token
-        self.admin = AdminService(self._api_domain, self._token, verify_ssl=verify_ssl)
+        self.admin = AdminService(
+            self._api_domain, _token_source, verify_ssl=verify_ssl
+        )
         self.analytics = AnalyticsService(
-            self._api_domain, self._token, verify_ssl=verify_ssl
+            self._api_domain, _token_source, verify_ssl=verify_ssl
         )
         self.assistants = AssistantService(
-            self._api_domain, self._token, verify_ssl=verify_ssl
+            self._api_domain, _token_source, verify_ssl=verify_ssl
         )
         self.categories = CategoryService(
-            self._api_domain, self._token, verify_ssl=verify_ssl
+            self._api_domain, _token_source, verify_ssl=verify_ssl
         )
-        self.llms = LLMService(self._api_domain, self._token, verify_ssl=verify_ssl)
+        self.llms = LLMService(self._api_domain, _token_source, verify_ssl=verify_ssl)
         self.mermaid = MermaidService(
-            self._api_domain, self._token, verify_ssl=verify_ssl
+            self._api_domain, _token_source, verify_ssl=verify_ssl
         )
         self.integrations = IntegrationService(
-            self._api_domain, self._token, verify_ssl=verify_ssl
+            self._api_domain, _token_source, verify_ssl=verify_ssl
         )
         self.projects = ProjectService(
-            self._api_domain, self._token, verify_ssl=verify_ssl
+            self._api_domain, _token_source, verify_ssl=verify_ssl
+        )
+        self.skills = SkillService(
+            self._api_domain, _token_source, verify_ssl=verify_ssl
         )
-        self.skills = SkillService(self._api_domain, self._token, verify_ssl=verify_ssl)
-        self.tasks = TaskService(self._api_domain, self._token, verify_ssl=verify_ssl)
-        self.users = UserService(self._api_domain, self._token, verify_ssl=verify_ssl)
+        self.tasks = TaskService(self._api_domain, _token_source, verify_ssl=verify_ssl)
+        self.users = UserService(self._api_domain, _token_source, verify_ssl=verify_ssl)
         self.datasources = DatasourceService(
-            self._api_domain, self._token, verify_ssl=verify_ssl
+            self._api_domain, _token_source, verify_ssl=verify_ssl
         )
         self.workflows = WorkflowService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
         self.conversations = ConversationService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
         self.files = FileOperationService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
         self.webhook = WebhookService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
         self.vendor_assistants = VendorAssistantService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
         self.vendor_workflows = VendorWorkflowService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
         self.vendor_knowledgebases = VendorKnowledgeBaseService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
         self.vendor_guardrails = VendorGuardrailService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
         self.codemie_guardrails = CodemieGuardrailService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
 
     @property
@@ -156,66 +162,9 @@ class CodeMieClient:
 
     def refresh_token(self) -> str:
         """Force token refresh."""
+        # Invalidate cache so next get_token() call re-fetches from Keycloak
+        self.auth._cached_token = None
+        self.auth._token_expires_at = 0.0
         self._token = "" if self._is_localhost else self.auth.get_token()
-        # Update token in services
-        self.admin = AdminService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.analytics = AnalyticsService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.assistants = AssistantService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.categories = CategoryService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.llms = LLMService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.mermaid = MermaidService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.integrations = IntegrationService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.skills = SkillService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.tasks = TaskService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.users = UserService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.datasources = DatasourceService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.workflows = WorkflowService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.conversations = ConversationService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.files = FileOperationService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.webhook = WebhookService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.vendor_assistants = VendorAssistantService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.vendor_workflows = VendorWorkflowService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.vendor_knowledgebases = VendorKnowledgeBaseService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.vendor_guardrails = VendorGuardrailService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
-        self.codemie_guardrails = CodemieGuardrailService(
-            self._api_domain, self._token, verify_ssl=self._verify_ssl
-        )
+        # Services already hold self.auth.get_token callable, so no reconstruction needed
         return self._token

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/admin.py

@@ -7,13 +7,13 @@ from ..models.admin import (
     ApplicationCreateRequest,
     ApplicationCreateResponse,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class AdminService:
     """Service for managing CodeMie applications/projects."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the admin service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/analytics.py

@@ -9,13 +9,13 @@ from ..models.analytics import (
     TabularResponse,
     UsersListResponse,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class AnalyticsService:
     """Service for managing CodeMie analytics."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the analytics service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/assistant.py

@@ -26,13 +26,13 @@ from ..models.assistant import (
     ReactionRequest,
 )
 from ..models.common import PaginationParams
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class AssistantService:
     """Service for managing CodeMie assistants."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the assistant service.
 
         Args:
@@ -373,7 +373,11 @@ class AssistantService:
             Chat response or streaming response
         """
         pydantic_schema = None
-        if issubclass(request.output_schema, BaseModel):
+        if (
+            request.output_schema is not None
+            and inspect.isclass(request.output_schema)
+            and issubclass(request.output_schema, BaseModel)
+        ):
             pydantic_schema = deepcopy(request.output_schema)
             request.output_schema = request.output_schema.model_json_schema()
 

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/categories.py

@@ -9,13 +9,13 @@ from ..models.categories import (
     CategoryResponse,
     CategoryListResponse,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class CategoryService:
     """Service for managing CodeMie assistant categories."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the category service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/codemie_guardrails.py

@@ -4,13 +4,13 @@ from ..models.guardrails import (
     GuardrailAssignmentRequest,
     GuardrailAssignmentResponse,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class CodemieGuardrailService:
     """Service for managing CodeMie guardrail assignments."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the CodeMie guardrail service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/conversation.py

@@ -19,13 +19,13 @@ from ..models.conversation import (
     SharedConversationResponse,
 )
 from ..models.assistant import AssistantChatRequest
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class ConversationService:
     """Service for managing user conversations."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the conversation service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/datasource.py

@@ -20,13 +20,13 @@ from ..models.datasource import (
     ElasticsearchStatsResponse,
 )
 from ..models.assistant import AssistantListResponse
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class DatasourceService:
     """Service for managing CodeMie Datasources."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the datasource service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/files.py

@@ -5,13 +5,13 @@ from pathlib import Path
 from typing import List
 
 from ..models.file_operation import FileBulkCreateResponse
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class FileOperationService:
     """Service for managing file operations."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the file operation service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/integration.py

@@ -10,13 +10,13 @@ from ..models.integration import (
     IntegrationType,
     IntegrationTestRequest,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class IntegrationService:
     """Service for managing CodeMie integrations (both user and project settings)."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the integration service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/llm.py

@@ -3,13 +3,13 @@
 from typing import List
 
 from ..models.llm import LLMModel
-from ..utils.http import ApiRequestHandler
+from ..utils.http import ApiRequestHandler, TokenSource
 
 
 class LLMService:
     """Service for managing LLM models."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the LLM models service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/mermaid.py

@@ -9,13 +9,13 @@ from ..models.mermaid import (
     ContentType,
     ResponseType,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class MermaidService:
     """Service for generating mermaid diagrams."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the mermaid service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/project.py

@@ -33,13 +33,13 @@ from ..models.project import (
     ProjectUpdateRequest,
     ProjectUpdateResponse,
 )
-from ..utils.http import ApiRequestHandler
+from ..utils.http import ApiRequestHandler, TokenSource
 
 
 class ProjectService:
     """Service for managing projects in CodeMie."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the ProjectService.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/skill.py

@@ -14,13 +14,13 @@ from ..models.skill import (
     SkillListResponse,
     SkillUpdateRequest,
 )
-from ..utils.http import ApiRequestHandler
+from ..utils.http import ApiRequestHandler, TokenSource
 
 
 class SkillService:
     """Service for managing CodeMie skills."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the skill service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/task.py

@@ -1,13 +1,13 @@
 """Task service implementation."""
 
 from ..models.task import BackgroundTaskEntity
-from ..utils.http import ApiRequestHandler
+from ..utils.http import ApiRequestHandler, TokenSource
 
 
 class TaskService:
     """Service for managing background tasks."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the Task service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/user.py

@@ -1,13 +1,13 @@
 """User service implementation."""
 
 from ..models.user import User, UserData
-from ..utils.http import ApiRequestHandler
+from ..utils.http import ApiRequestHandler, TokenSource
 
 
 class UserService:
     """Service for managing user profile and preferences."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the User service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/vendor_assistant.py

@@ -13,13 +13,13 @@ from ..models.vendor_assistant import (
     VendorAssistantInstallResponse,
     VendorAssistantUninstallResponse,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class VendorAssistantService:
     """Service for managing cloud vendor assistant settings (AWS, Azure, GCP)."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the vendor service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/vendor_guardrail.py

@@ -13,13 +13,13 @@ from ..models.vendor_guardrail import (
     VendorGuardrailInstallResponse,
     VendorGuardrailUninstallResponse,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class VendorGuardrailService:
     """Service for managing cloud vendor guardrail settings (AWS, Azure, GCP)."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the vendor guardrail service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/vendor_knowledgebase.py

@@ -11,13 +11,13 @@ from ..models.vendor_knowledgebase import (
     VendorKnowledgeBaseInstallResponse,
     VendorKnowledgeBaseUninstallResponse,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class VendorKnowledgeBaseService:
     """Service for managing cloud vendor knowledge base settings (AWS, Azure, GCP)."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the vendor knowledge base service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/vendor_workflow.py

@@ -12,13 +12,13 @@ from ..models.vendor_workflow import (
     VendorWorkflowInstallResponse,
     VendorWorkflowUninstallResponse,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class VendorWorkflowService:
     """Service for managing cloud vendor workflow settings (AWS, Azure, GCP)."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the vendor workflow service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/webhook.py

@@ -1,13 +1,13 @@
 import requests
 
 from typing import Dict, Any
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class WebhookService:
     """Webhook service implementation."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the conversation service.
 
         Args:

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/services/workflow.py

@@ -10,13 +10,13 @@ from ..models.workflow import (
     WorkflowUpdateRequest,
     Workflow,
 )
-from ..utils import ApiRequestHandler
+from ..utils import ApiRequestHandler, TokenSource
 
 
 class WorkflowService:
     """Service for managing CodeMie workflows."""
 
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
         """Initialize the workflow service.
 
         Args:

codemie_sdk_python-0.1.428/src/codemie_sdk/utils/__init__.py

@@ -0,0 +1,5 @@
+"""Utility modules for CodeMie SDK."""
+
+from .http import ApiRequestHandler, TokenSource
+
+__all__ = ["ApiRequestHandler", "TokenSource"]

{codemie_sdk_python-0.1.427 → codemie_sdk_python-0.1.428}/src/codemie_sdk/utils/http.py

@@ -1,12 +1,25 @@
 """HTTP utilities for CodeMie SDK."""
 
-from typing import TypeVar, Type, Optional, Any, Dict, List, get_origin, get_args
+from typing import (
+    TypeVar,
+    Type,
+    Optional,
+    Any,
+    Dict,
+    List,
+    Callable,
+    Union,
+    get_origin,
+    get_args,
+)
 import requests
 import logging
 from functools import wraps
 
 T = TypeVar("T")
 
+TokenSource = Union[str, Callable[[], str]]
+
 logger = logging.getLogger(__name__)
 
 
@@ -33,12 +46,17 @@ def log_request(func):
 class ApiRequestHandler:
     """Handles HTTP requests with consistent error handling and response parsing."""
 
-    def __init__(self, base_url: str, token: str, verify_ssl: bool = True):
+    def __init__(
+        self,
+        base_url: str,
+        token: Union[str, Callable[[], str]],
+        verify_ssl: bool = True,
+    ):
         """Initialize the API request handler.
 
         Args:
             base_url: Base URL for the API
-            token: Authentication token
+            token: Authentication token string or callable that returns one
             verify_ssl: Whether to verify SSL certificates
         """
         self._base_url = base_url.rstrip("/")
@@ -72,11 +90,11 @@ class ApiRequestHandler:
             headers["Content-Type"] = "application/json"
 
         if include_auth:
-            headers["Authorization"] = (
-                f"Bearer {self._token}"
-                if not self._is_localhost
-                else "dev-codemie-user"
-            )
+            if self._is_localhost:
+                headers["Authorization"] = "dev-codemie-user"
+            else:
+                token_value = self._token() if callable(self._token) else self._token
+                headers["Authorization"] = f"Bearer {token_value}"
         return headers
 
     def _parse_response(

codemie_sdk_python-0.1.427/src/codemie_sdk/utils/__init__.py

@@ -1,5 +0,0 @@
-"""Utility modules for CodeMie SDK."""
-
-from .http import ApiRequestHandler
-
-__all__ = ["ApiRequestHandler"]