codemie-sdk-python 0.1.10__tar.gz → 0.1.12__tar.gz

{codemie_sdk_python-0.1.10 → codemie_sdk_python-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.10
+Version: 0.1.12
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
@@ -57,6 +57,12 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
 ```sh
 pip install codemie-sdk
 ```
+OR
+```sh
+poetry install
+```
+
+### If you want to run only tests, go to ## Running tests section
 
 ## Usage
 
@@ -740,7 +746,8 @@ For providing credentials please contact AI/Run CodeMie Team: Vadym_Vlasenko@epa
 
 ## Running tests
 
-For running tests on custom environment you should create .env file in the root directory
+For running tests on custom environment you should create .env file in the root directory,
+ask QA team: anton_yeromin@epam.com to provide all needed testing credentials
 
 ``` properties
 

{codemie_sdk_python-0.1.10 → codemie_sdk_python-0.1.12}/README.md

@@ -44,6 +44,12 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
 ```sh
 pip install codemie-sdk
 ```
+OR
+```sh
+poetry install
+```
+
+### If you want to run only tests, go to ## Running tests section
 
 ## Usage
 
@@ -727,7 +733,8 @@ For providing credentials please contact AI/Run CodeMie Team: Vadym_Vlasenko@epa
 
 ## Running tests
 
-For running tests on custom environment you should create .env file in the root directory
+For running tests on custom environment you should create .env file in the root directory,
+ask QA team: anton_yeromin@epam.com to provide all needed testing credentials
 
 ``` properties
 

{codemie_sdk_python-0.1.10 → codemie_sdk_python-0.1.12}/pyproject.toml

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

{codemie_sdk_python-0.1.10 → codemie_sdk_python-0.1.12}/src/codemie_sdk/models/assistant.py

@@ -177,6 +177,30 @@ class Context(BaseModel):
     name: str
 
 
+class MCPServerConfig(BaseModel):
+    model_config = ConfigDict(extra="ignore")
+
+    command: str
+    args: Optional[list[str]]
+    env: Optional[dict[str, Any]] = None
+    auth_token: Optional[str] = None
+
+
+class MCPServerDetails(BaseModel):
+    model_config = ConfigDict(extra="ignore")
+
+    name: str
+    description: Optional[str]
+    enabled: bool
+    config: Optional[MCPServerConfig] = None
+    mcp_connect_url: Optional[str] = None
+    tools_tokens_size_limit: Optional[int] = None
+    command: Optional[str] = None
+    arguments: Optional[str] = None
+    settings: Optional[Integration] = None
+    mcp_connect_auth_token: Optional[Integration] = None
+
+
 class SystemPromptHistory(BaseModel):
     """Model for system prompt history."""
 
@@ -221,6 +245,8 @@ class Assistant(AssistantBase):
     top_p: Optional[float] = None
     context: List[Context] = Field(default_factory=list)
     user_abilities: Optional[List[Any]] = None
+    mcp_servers: List[MCPServerDetails] = Field(default_factory=list)
+    assistant_ids: List[str] = Field(default_factory=list)
 
 
 class AssistantRequestBase(AssistantBase):
@@ -232,7 +258,7 @@ class AssistantRequestBase(AssistantBase):
     project: str
     context: List[Context] = Field(default_factory=list)
     llm_model_type: str
-    toolkits: List[ToolKitDetails]
+    toolkits: List[ToolKitDetails] = Field(default_factory=list)
     user_prompts: List[str] = Field(default_factory=list)
     shared: bool = False
     is_react: bool = False
@@ -240,6 +266,8 @@ class AssistantRequestBase(AssistantBase):
     slug: Optional[str] = None
     temperature: Optional[float] = None
     top_p: Optional[float] = None
+    mcp_servers: List[MCPServerDetails] = Field(default_factory=list)
+    assistant_ids: List[str] = Field(default_factory=list)
 
 
 class AssistantCreateRequest(AssistantRequestBase):