codemie-sdk-python 0.1.37__py3-none-any.whl → 0.1.38__py3-none-any.whl

codemie_sdk/__init__.py

@@ -19,5 +19,5 @@ Basic usage:
 
 from .client.client import CodeMieClient
 
-__version__ = "0.1.11"
+__version__ = "0.1.19"
 __all__ = ["CodeMieClient"]

codemie_sdk/models/assistant.py

@@ -41,6 +41,7 @@ class ContextType(str, Enum):
 
     KNOWLEDGE_BASE = "knowledge_base"
     CODE = "code"
+    PROVIDER = "provider"
 
 
 class Context(BaseModel):
@@ -89,6 +90,9 @@ class SystemPromptHistory(BaseModel):
 class AssistantBase(BaseModel):
     """Base model for assistant with common fields."""
 
+    def __getitem__(self, key):
+        return getattr(self, key)
+
     model_config = ConfigDict(extra="ignore")
 
     id: Optional[str] = None
@@ -127,6 +131,9 @@ class Assistant(AssistantBase):
 class AssistantRequestBase(AssistantBase):
     """Base model for assistant requests with common request fields."""
 
+    def __getitem__(self, key):
+        return getattr(self, key)
+
     model_config = ConfigDict(extra="ignore", use_enum_values=True)
 
     system_prompt: str
@@ -244,3 +251,15 @@ class BaseModelResponse(BaseModel):
         populate_by_name = True
         # Preserve alias on export
         alias_generator = None
+
+
+class EnvVars(BaseModel):
+    azure_openai_url: Optional[str] = None
+    azure_openai_api_key: Optional[str] = None
+    openai_api_type: Optional[str] = None
+    openai_api_version: Optional[str] = None
+    models_env: Optional[str] = None
+
+
+class ExportAssistantPayload(BaseModel):
+    env_vars: Optional[EnvVars] = None

codemie_sdk/models/common.py

@@ -14,6 +14,9 @@ class TokensUsage(BaseModel):
 class User(BaseModel):
     """Model representing the User."""
 
+    def __getitem__(self, key):
+        return getattr(self, key)
+
     model_config = ConfigDict(extra="ignore")
 
     user_id: str = Field(None)

codemie_sdk/services/assistant.py

@@ -14,6 +14,8 @@ from ..models.assistant import (
     AssistantChatRequest,
     BaseModelResponse,
     AssistantBase,
+    Context,
+    ExportAssistantPayload,
 )
 from ..models.common import PaginationParams
 from ..utils import ApiRequestHandler
@@ -122,6 +124,17 @@ class AssistantService:
             "/v1/assistants/tools", List[ToolKitDetails], wrap_response=False
         )
 
+    def get_context(self, project_name: str) -> List[Context]:
+        """Get list of available contexts.
+
+        Args: project_name: Name of the project to retrieve context for
+
+        Returns:
+            All available assistants context
+        """
+        params = {"project_name": project_name}
+        return self._api.get("/v1/assistants/context", List[Context], params=params)
+
     def delete(self, assistant_id: str) -> dict:
         """Delete an assistant by ID.
 
@@ -183,3 +196,20 @@ class AssistantService:
             response = self._api.post_multipart("/v1/files/", dict, files=files)
 
         return response
+
+    def export(self, assistant_id: str, request: ExportAssistantPayload):
+        """Export an assistant.
+
+        Args:
+            assistant_id: ID of the assistant to export
+            request: Export request details
+
+        Returns:
+             input stream of the exported assistant file"""
+
+        return self._api.post(
+            f"/v1/assistants/id/{assistant_id}/export",
+            response_model=Any,
+            stream=True,
+            json_data=request.model_dump(exclude_none=True),
+        )

{codemie_sdk_python-0.1.37.dist-info → codemie_sdk_python-0.1.38.dist-info}/METADATA

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

{codemie_sdk_python-0.1.37.dist-info → codemie_sdk_python-0.1.38.dist-info}/RECORD

@@ -1,11 +1,11 @@
-codemie_sdk/__init__.py,sha256=xfhrTFdvRmLSsrdQksX5GhnoQEj6tvkeytkrbB20Jj8,567
+codemie_sdk/__init__.py,sha256=P-U2Xw3rFixA2tX-JjVS7zpqo6WR8YnKMiIJWZnky6A,567
 codemie_sdk/auth/__init__.py,sha256=IksEj223xEZtJ-cQ0AT9L0Bs9psIJ8QNzDXrPTUQ3xQ,126
 codemie_sdk/auth/credentials.py,sha256=u2eLNsD8fELTgreQghVb0g0kk82zchUkux0M5lzq-u8,4320
 codemie_sdk/client/__init__.py,sha256=yf6C39MmrJ6gK9ZHMhBeynKwUUYVSUTQbKxU8-4qpKg,101
 codemie_sdk/client/client.py,sha256=dVcWqd-ruWd8GOZ3_33C3LnHgUyyxIoCKxKYPYbBPq8,4373
 codemie_sdk/exceptions.py,sha256=XoVPyognx-JmyVxLHkZPAcX1CMi1OoT1diBFJLU54so,1183
-codemie_sdk/models/assistant.py,sha256=RCVuYcKI_DzGyObJ3e1l4rrcVev-1Hq7crW9XY5pvus,7479
-codemie_sdk/models/common.py,sha256=V4sJCzwFTF8BPlrd2OKI44Em5HmPmn2Nm8fQNsnBZ_Q,1128
+codemie_sdk/models/assistant.py,sha256=HA9fHCDjU5BJ2HEtyXbCoUzBhEkSoJv9Ujx-XWf0Zi0,7963
+codemie_sdk/models/common.py,sha256=gmZ-ps8TbaieNKr0kUKoQEjhVrHD2CAYomOpZQRatH8,1195
 codemie_sdk/models/datasource.py,sha256=u6X1xXZ3ZR9CWH0B0bSw3IjC7KyEwtRKHIc6ZuhJwsg,10813
 codemie_sdk/models/integration.py,sha256=yT_hwETh4kghhH__VyGGXLrDWAYLgCSB97vdZH-BjEs,1831
 codemie_sdk/models/llm.py,sha256=ppb9-1dx1UFhRuJpSR3ij7H6Pfhe9nO4C4BEOIbToy4,1192
@@ -13,7 +13,7 @@ codemie_sdk/models/task.py,sha256=J4ZFRY3s8qBGrqB5NLQF0rMbInLh4s7OEZ0ZfmnW0Ho,14
 codemie_sdk/models/user.py,sha256=Q0rjimZh-IbeaPfq6b6fk6ZaCtwLqWHEIlU863suCS4,1777
 codemie_sdk/models/workflow.py,sha256=h3ZM-2bBucRd9CzZUYDTU0rGR1ihEN7cxkHgQmJfJVc,2394
 codemie_sdk/models/workflow_state.py,sha256=CMYFQZ7sy4QxmnWmc83TFfqP7TG_3rW5MdH5fxsS9kY,1251
-codemie_sdk/services/assistant.py,sha256=D3aYFPd97wZIpKylAmxF6qtMwSeKyM54OqeKoDOw1U4,5554
+codemie_sdk/services/assistant.py,sha256=xuzdel93VlOOy_xCmuptlwpChQWOvuPbpBoB2nF-Sx0,6499
 codemie_sdk/services/datasource.py,sha256=gIP9D-sq4733x9f0mWL87bMI1pWYxIsv5eYXkAs0O6Q,7150
 codemie_sdk/services/integration.py,sha256=vJnSkXk2C2l0ahX2SUsuA7fKhY2hTuAByynx5Lgh7Ls,4864
 codemie_sdk/services/llm.py,sha256=0-e4_7RvLHs2giCyoQ5U4KDTh6p5VXgPKNxnDP9ZDFU,1100
@@ -24,6 +24,6 @@ codemie_sdk/services/workflow_execution.py,sha256=aGoT3rdTmh5-doAsrmBBjLEuOfvL5a
 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=YzE53y1r7PBUKcEQt-gGYduRY06I8xzqVRzlj9f_voA,10084
-codemie_sdk_python-0.1.37.dist-info/METADATA,sha256=fG3QkgniRdV_krD5gSTzg3YOMVBOHQCE47cA7VP8mLY,21614
-codemie_sdk_python-0.1.37.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.37.dist-info/RECORD,,
+codemie_sdk_python-0.1.38.dist-info/METADATA,sha256=vL4AX66td_KbKT-bXm4wKR_TZLsxKRdxTG2h9-G7iqU,21614
+codemie_sdk_python-0.1.38.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.38.dist-info/RECORD,,