Fix AI agent invocation handling for dict responses (#28)

* fix issue with AI agent invoke

* simplify agent response

* update poetry version

* update changelog

Author: Matanga1-2

CHANGELOG.md

@@ -5,6 +5,11 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.2.10] - 2025-06-04
+
+### Changed
+- Simplified AI agent response handling by removing conditional API validation.
+
 ## [0.1.8] - 2025-04-27
 
 ### Changed

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcp-server-port"
-version = "0.2.9"
+version = "0.2.10"
 authors = [
   { name = "Matan Grady", email = "[email protected]" }
 ]
@@ -32,7 +32,7 @@ include = [
 ]
 [tool.poetry]
 name = "mcp-server-port"
-version = "0.2.9"
+version = "0.2.10"
 description = "Port's MCP server"
 packages = [
     { include = "src" }

src/client/agent.py

@@ -29,10 +29,7 @@ async def trigger_agent(self, prompt: str) -> PortAgentTriggerResponse:
             raise PortError("Response missing required invocation identifier")
 
         try:
-            if config.api_validation_enabled:
-                return PortAgentTriggerResponse(**response_data)
-            else:
-                return PortAgentTriggerResponse.construct(**response_data)
+            return PortAgentTriggerResponse(**response_data)
         except Exception as e:
             logger.error(f"Failed to parse trigger agent response: {e}")
             logger.error(f"Response data: {response_data}")

src/tools/ai_agent/invoke_ai_agent.py

@@ -42,13 +42,10 @@ def __init__(self, port_client: PortClient):
         self.port_client = port_client
 
     async def invoke_ai_agent(self, props: InvokeAIAGentToolSchema) -> dict[str, Any]:
-        # Handle both dict input and schema object
         prompt = props.prompt
-        # Trigger agent
         logger.info(f"Invoking Port's AI agent with prompt: {prompt[:50]}{'...' if len(prompt) > 50 else ''}")
         response = await self.port_client.trigger_agent(prompt)
 
-        # Get identifier from the Pydantic response object
         identifier = response.invocation.identifier
 
         logger.info(f"Got invocation identifier: {identifier}")