codemie-sdk-python 0.1.27__tar.gz → 0.1.28__tar.gz

{codemie_sdk_python-0.1.27 → codemie_sdk_python-0.1.28}/PKG-INFO

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

{codemie_sdk_python-0.1.27 → codemie_sdk_python-0.1.28}/pyproject.toml

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

{codemie_sdk_python-0.1.27 → codemie_sdk_python-0.1.28}/src/codemie_sdk/models/assistant.py

@@ -3,9 +3,9 @@
 import uuid
 from datetime import datetime
 from enum import Enum
-from typing import List, Optional, Any, Union
+from typing import List, Optional, Any, Union, Dict
 
-from pydantic import BaseModel, Field, ConfigDict
+from pydantic import BaseModel, Field, ConfigDict, model_validator
 
 from .common import User
 from .integration import Integration
@@ -171,6 +171,28 @@ class ChatMessage(BaseModel):
     message: Optional[str] = Field(default="")
 
 
+class ToolConfig(BaseModel):
+    name: str
+    tool_creds: Optional[Dict[str, Any]] = None
+    integration_id: Optional[str] = None
+
+    @model_validator(mode="after")
+    def validate_credentials_provided(self) -> "ToolConfig":
+        """
+        Validate that either tool_creds or integration_id is provided.
+
+        At least one of these fields must be specified for the tool configuration
+        to be valid. This ensures that the tool has a way to obtain credentials.
+        """
+        if not self.tool_creds and not self.integration_id:
+            raise ValueError("Either tool_creds or integration_id must be provided")
+        if self.tool_creds and self.integration_id:
+            raise ValueError(
+                "Either tool_creds or integration_id must be provided, but not both"
+            )
+        return self
+
+
 class AssistantChatRequest(BaseModel):
     """Model for chat request to assistant."""
 
@@ -197,6 +219,7 @@ class AssistantChatRequest(BaseModel):
     metadata: Optional[dict[str, Any]] = Field(
         default=None, description="Provide additional metadata"
     )
+    tools_config: Optional[List[ToolConfig]] = None
 
 
 class BaseModelResponse(BaseModel):