codemie-sdk-python 0.1.37__tar.gz → 0.1.38__tar.gz

{codemie_sdk_python-0.1.37 → codemie_sdk_python-0.1.38}/PKG-INFO

@@ -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 → codemie_sdk_python-0.1.38}/pyproject.toml

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

{codemie_sdk_python-0.1.37 → codemie_sdk_python-0.1.38}/src/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_python-0.1.37 → codemie_sdk_python-0.1.38}/src/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_python-0.1.37 → codemie_sdk_python-0.1.38}/src/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_python-0.1.37 → codemie_sdk_python-0.1.38}/src/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),
+        )