intentkit 0.7.5.dev12__py3-none-any.whl → 0.7.5.dev13__py3-none-any.whl

intentkit/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.7.5-dev12"
+__version__ = "0.7.5-dev13"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

intentkit/models/agent.py

@@ -1415,16 +1415,7 @@ class Agent(AgentCreate, AgentPublicInfo):
                 skills_properties = skills_property.get("properties", {})
 
                 if skills_properties:
-                    # Load all skills from the database
-                    # Query all skills grouped by category with enabled status
-                    stmt = select(
-                        SkillTable.category,
-                        func.bool_or(SkillTable.enabled).label("any_enabled"),
-                    ).group_by(SkillTable.category)
-                    result = await db.execute(stmt)
-                    category_status = {row.category: row.any_enabled for row in result}
-
-                    # Query all skills with their price levels for adding x-price-level fields
+                    # Query all enabled skills with their price levels for adding x-price-level fields
                     skills_stmt = select(
                         SkillTable.category,
                         SkillTable.config_name,
@@ -1434,8 +1425,12 @@ class Agent(AgentCreate, AgentPublicInfo):
                     skills_result = await db.execute(skills_stmt)
                     skills_data = {}
                     category_price_levels = {}
+                    enabled_categories = set()
 
                     for row in skills_result:
+                        # Collect enabled categories using a set for deduplication
+                        enabled_categories.add(row.category)
+
                         if row.category not in skills_data:
                             skills_data[row.category] = {}
                             category_price_levels[row.category] = []
@@ -1458,11 +1453,8 @@ class Agent(AgentCreate, AgentPublicInfo):
 
                     # Process each skill in the schema
                     for skill_category in skill_keys:
-                        if skill_category not in category_status:
-                            # If category not found in database, remove it from schema
-                            skills_properties.pop(skill_category, None)
-                        elif not category_status[skill_category]:
-                            # If category exists but all skills are disabled, remove it
+                        if skill_category not in enabled_categories:
+                            # If category not found in enabled categories, remove it from schema
                             skills_properties.pop(skill_category, None)
                         elif filter_owner_api_skills and cls._is_agent_owner_only_skill(
                             skills_properties[skill_category]

intentkit/skills/venice_audio/venice_audio.py

@@ -9,7 +9,7 @@ from pydantic import BaseModel, Field
 from intentkit.abstracts.skill import SkillStoreABC
 from intentkit.skills.venice_audio.base import VeniceAudioBaseTool
 from intentkit.skills.venice_audio.input import AllowedAudioFormat, VeniceAudioInput
-from intentkit.utils.s3 import FileType, store_file_bytes
+from intentkit.utils.s3 import store_file
 
 logger = logging.getLogger(__name__)
 
@@ -48,7 +48,7 @@ class VeniceAudioTool(VeniceAudioBaseTool):
     ) -> Dict[str, Any]:
         """
         Generates audio using the configured voice model via Venice AI TTS /audio/speech endpoint.
-        Stores the resulting audio using store_file_bytes.
+        Stores the resulting audio using the generic S3 helper.
         Returns a dictionary containing audio details on success, or API error details on failure.
         """
         context = self.get_context()
@@ -155,11 +155,32 @@ class VeniceAudioTool(VeniceAudioBaseTool):
                     key = f"{self.category}/{voice_model}/{audio_hash}.{file_extension}"
 
                     size_limit = 1024 * 20  # 20Mb Size limit
-                    stored_url = await store_file_bytes(
-                        file_bytes=audio_bytes,
+                    audio_size = len(audio_bytes)
+                    if audio_size > size_limit:
+                        message = f"Generated audio exceeds the allowed size of {size_limit} bytes."
+                        logger.error(
+                            "Failed to store audio (Voice: %s): %s",
+                            voice_model,
+                            message,
+                        )
+                        return {
+                            "error": True,
+                            "error_type": "FileSizeLimitExceeded",
+                            "message": message,
+                            "voice_model": voice_model,
+                            "requested_format": final_response_format,
+                        }
+
+                    mime_type = (
+                        content_type_header.split(";", 1)[0].strip()
+                        if content_type_header
+                        else None
+                    )
+                    stored_url = await store_file(
+                        content=audio_bytes,
                         key=key,
-                        file_type=FileType.AUDIO,
-                        size_limit_bytes=size_limit,
+                        content_type=mime_type,
+                        size=audio_size,
                     )
 
                     if not stored_url:
@@ -182,7 +203,7 @@ class VeniceAudioTool(VeniceAudioBaseTool):
                     return {
                         "audio_url": stored_url,
                         "audio_bytes_sha256": audio_hash,
-                        "content_type": content_type_header,
+                        "content_type": mime_type or content_type_header,
                         "voice_model": voice_model,
                         "tts_engine": tts_model_id,
                         "speed": speed if speed is not None else 1.0,

intentkit/utils/s3.py

@@ -182,6 +182,59 @@ class FileType(str, Enum):
     PDF = "pdf"
 
 
+async def store_file(
+    content: bytes,
+    key: str,
+    content_type: Optional[str] = None,
+    size: Optional[int] = None,
+) -> str:
+    """Store raw file bytes with automatic content type detection."""
+    if not _client or not _bucket or not _prefix or not _cdn_url:
+        logger.info("S3 not initialized. Cannot store file bytes.")
+        return ""
+
+    if not content:
+        raise ValueError("File content cannot be empty")
+
+    actual_size = len(content)
+    if size is not None and size != actual_size:
+        raise ValueError(
+            f"Provided size {size} does not match actual content size {actual_size} bytes"
+        )
+
+    effective_size = size if size is not None else actual_size
+
+    detected_content_type = content_type
+    if not detected_content_type:
+        kind = filetype.guess(content)
+        detected_content_type = (
+            kind.mime if kind and kind.mime else "application/octet-stream"
+        )
+
+    prefixed_key = f"{_prefix}{key}"
+    file_obj = BytesIO(content)
+
+    logger.info(
+        "Uploading file to S3 with content type %s and size %s bytes",
+        detected_content_type,
+        effective_size,
+    )
+
+    _client.upload_fileobj(
+        file_obj,
+        _bucket,
+        prefixed_key,
+        ExtraArgs={
+            "ContentType": detected_content_type,
+            "ContentDisposition": "inline",
+        },
+    )
+
+    cdn_url = f"{_cdn_url}/{prefixed_key}"
+    logger.info("File uploaded successfully to %s", cdn_url)
+    return cdn_url
+
+
 async def store_file_bytes(
     file_bytes: bytes,
     key: str,

{intentkit-0.7.5.dev12.dist-info → intentkit-0.7.5.dev13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.7.5.dev12
+Version: 0.7.5.dev13
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestalnetwork/intentkit
 Project-URL: Repository, https://github.com/crestalnetwork/intentkit

{intentkit-0.7.5.dev12.dist-info → intentkit-0.7.5.dev13.dist-info}/RECORD

@@ -1,4 +1,4 @@
-intentkit/__init__.py,sha256=RnzjeFMcq6VPro5wJazURqVrmSzAFezYc9pQfYskF5Y,384
+intentkit/__init__.py,sha256=WCRarWbBBC8guMPMfmalWRblSMCx17YzO1FqXtdlpzo,384
 intentkit/abstracts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 intentkit/abstracts/agent.py,sha256=108gb5W8Q1Sy4G55F2_ZFv2-_CnY76qrBtpIr0Oxxqk,1489
 intentkit/abstracts/api.py,sha256=ZUc24vaQvQVbbjznx7bV0lbbQxdQPfEV8ZxM2R6wZWo,166
@@ -21,7 +21,7 @@ intentkit/core/credit.py,sha256=b4f4T6G6eeBTMe0L_r8awWtXgUnqiog4IUaymDPYym0,7558
 intentkit/core/engine.py,sha256=DoGqfhNK35sFeqMlFLyKR_NUDf1YXf4yihSfw7kufmU,36177
 intentkit/core/node.py,sha256=7h9zgDSd928bzUi3m3EZnKkhbwqlbRAQUr_uz7gKB5Y,8880
 intentkit/core/prompt.py,sha256=ssiyKHDNIjQOLU0KwUlIFX3jy51TqgeKOxrwnW4HBkw,15875
-intentkit/models/agent.py,sha256=ATRNAlmO1YN995kQa_IVAW7TUPOxAcDZDRTdHX95X_A,67442
+intentkit/models/agent.py,sha256=usZjgzW7mb0uQ1SIVFM944sTB0ul5-Xr8FOC0YTKswE,66942
 intentkit/models/agent_data.py,sha256=5zq3EPKnygT2P1OHc2IfEmL8hXkjeBND6sJ0JJsvQJg,28370
 intentkit/models/agent_public.json,sha256=0X8Bd2WOobDJLsok8avWNzmzu4uvKSGEyy6Myn53eT4,2802
 intentkit/models/agent_schema.json,sha256=hFGUE57JIiN8V4olgLf2LBXPejA2QLiQoFc6riM1UFo,17139
@@ -357,7 +357,7 @@ intentkit/skills/venice_audio/__init__.py,sha256=E39CeSUHRoGV9IxNUtwo-nCp0Yfq5rv
 intentkit/skills/venice_audio/base.py,sha256=1oxHrmDXpBZbA7q2BqatlpuY-X3uwhK22T8QZjKmkqw,4845
 intentkit/skills/venice_audio/input.py,sha256=nWsbW40Bd-Q331PpBZ-NBcsgw3pqEWsWIvY0lsbNR4A,1887
 intentkit/skills/venice_audio/schema.json,sha256=g3cb9zJvinnRrilvNIAdG8qivgGyB7sHCEC8392MXhY,5268
-intentkit/skills/venice_audio/venice_audio.py,sha256=2EfCkC-MH1p-evTf594yHKNiJiCmAlIPLuYgXF48Hlk,10365
+intentkit/skills/venice_audio/venice_audio.py,sha256=3xIUbNZ_ZZ5J2FYJzGaJ0ZCwX_IpAouVuoTyl0JI72Q,11305
 intentkit/skills/venice_audio/venice_logo.jpg,sha256=XDnVdh3V8UuJjAeSmp4rbqhTKn0yjRI6M_6z1DSs8cA,13342
 intentkit/skills/venice_image/README.md,sha256=79NWUOaYWeFLOhR7m424pLIZ7VT0fhkxSBcohoy8Oh0,5017
 intentkit/skills/venice_image/__init__.py,sha256=fPq9OS8er67cn3zyo9H_7iEj5cSUu7nOmqhhjwG9xNs,5841
@@ -415,10 +415,10 @@ intentkit/utils/chain.py,sha256=PubXjJM_kt6NCwzpKm5uXQ596QvKFLfr2L_DP9ntHAs,1516
 intentkit/utils/error.py,sha256=fZOZ9CPrU08veSyDBoNf44UjkojpsyFMEr2SxEdlSZA,4898
 intentkit/utils/logging.py,sha256=bhwZi5vscjBTd9kaNp_L6ijrfv9Sl3lsr4ARaUB4Iec,2389
 intentkit/utils/random.py,sha256=DymMxu9g0kuQLgJUqalvgksnIeLdS-v0aRk5nQU0mLI,452
-intentkit/utils/s3.py,sha256=9trQNkKQ5VgxWsewVsV8Y0q_pXzGRvsCYP8xauyUYkg,8549
+intentkit/utils/s3.py,sha256=A8Nsx5QJyLsxhj9g7oHNy2-m24tjQUhC9URm8Qb1jFw,10057
 intentkit/utils/slack_alert.py,sha256=s7UpRgyzLW7Pbmt8cKzTJgMA9bm4EP-1rQ5KXayHu6E,2264
 intentkit/utils/tx.py,sha256=2yLLGuhvfBEY5n_GJ8wmIWLCzn0FsYKv5kRNzw_sLUI,1454
-intentkit-0.7.5.dev12.dist-info/METADATA,sha256=qFBZAvX5suIuqP5k7ndeAXh7mhyDtAh5awVKf9-4IcU,6410
-intentkit-0.7.5.dev12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-intentkit-0.7.5.dev12.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
-intentkit-0.7.5.dev12.dist-info/RECORD,,
+intentkit-0.7.5.dev13.dist-info/METADATA,sha256=Y360u6Lp9aLakwP2Kj9eN-RmhQbwPyCT5j1VgxoA42k,6410
+intentkit-0.7.5.dev13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+intentkit-0.7.5.dev13.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
+intentkit-0.7.5.dev13.dist-info/RECORD,,