|
42 | 42 | "APICallParamsSchema", |
43 | 43 | "APICallParamsSchemaProperties", |
44 | 44 | "Integration", |
| 45 | + "IntegrationMcpIntegrationDef", |
| 46 | + "IntegrationMcpIntegrationDefArguments", |
| 47 | + "IntegrationMcpIntegrationDefArgumentsMcpCallToolArguments", |
| 48 | + "IntegrationMcpIntegrationDefArgumentsMcpListToolsArguments", |
| 49 | + "IntegrationMcpIntegrationDefSetup", |
45 | 50 | "IntegrationGoogleSheetsIntegrationDefOutput", |
46 | 51 | "IntegrationGoogleSheetsIntegrationDefOutputArguments", |
47 | 52 | "IntegrationGoogleSheetsIntegrationDefOutputArgumentsGoogleSheetsReadArguments", |
@@ -108,6 +113,53 @@ class APICall(BaseModel): |
108 | 113 | timeout: Optional[int] = None |
109 | 114 |
|
110 | 115 |
|
| 116 | +class IntegrationMcpIntegrationDefArgumentsMcpCallToolArguments(BaseModel): |
| 117 | + tool_name: str |
| 118 | + |
| 119 | + arguments: Optional[object] = None |
| 120 | + |
| 121 | + timeout_seconds: Optional[int] = None |
| 122 | + |
| 123 | + |
| 124 | +class IntegrationMcpIntegrationDefArgumentsMcpListToolsArguments(BaseModel): |
| 125 | + dummy: Optional[str] = None |
| 126 | + |
| 127 | + |
| 128 | +IntegrationMcpIntegrationDefArguments: TypeAlias = Union[ |
| 129 | + IntegrationMcpIntegrationDefArgumentsMcpCallToolArguments, |
| 130 | + IntegrationMcpIntegrationDefArgumentsMcpListToolsArguments, |
| 131 | + None, |
| 132 | +] |
| 133 | + |
| 134 | + |
| 135 | +class IntegrationMcpIntegrationDefSetup(BaseModel): |
| 136 | + transport: Literal["sse", "http"] |
| 137 | + |
| 138 | + args: Optional[List[str]] = None |
| 139 | + |
| 140 | + command: Optional[str] = None |
| 141 | + |
| 142 | + cwd: Optional[str] = None |
| 143 | + |
| 144 | + env: Optional[Dict[str, str]] = None |
| 145 | + |
| 146 | + http_headers: Optional[Dict[str, str]] = None |
| 147 | + |
| 148 | + http_url: Optional[str] = None |
| 149 | + |
| 150 | + |
| 151 | +class IntegrationMcpIntegrationDef(BaseModel): |
| 152 | + arguments: Optional[IntegrationMcpIntegrationDefArguments] = None |
| 153 | + """Arguments to call a named tool on the MCP server""" |
| 154 | + |
| 155 | + method: Optional[str] = None |
| 156 | + |
| 157 | + provider: Optional[Literal["mcp"]] = None |
| 158 | + |
| 159 | + setup: Optional[IntegrationMcpIntegrationDefSetup] = None |
| 160 | + """Setup parameters for MCP integration""" |
| 161 | + |
| 162 | + |
111 | 163 | class IntegrationGoogleSheetsIntegrationDefOutputArgumentsGoogleSheetsReadArguments(BaseModel): |
112 | 164 | range: str |
113 | 165 |
|
@@ -240,6 +292,7 @@ class IntegrationGoogleSheetsIntegrationDefOutput(BaseModel): |
240 | 292 | ArxivIntegrationDef, |
241 | 293 | UnstructuredIntegrationDef, |
242 | 294 | AlgoliaIntegrationDef, |
| 295 | + IntegrationMcpIntegrationDef, |
243 | 296 | IntegrationGoogleSheetsIntegrationDefOutput, |
244 | 297 | None, |
245 | 298 | ] |
|
0 commit comments