types

Author: tripolskypetr

src/functions/test/overrideCompute.ts

@@ -3,7 +3,7 @@
  * @description Provides a function to override an existing compute schema with partial updates.
  */
 
-import { IComputeSchema } from "../../interfaces/Compute.interface";
+import { IComputeData, IComputeSchema } from "../../interfaces/Compute.interface";
 import swarm from "../../lib";
 import { GLOBAL_CONFIG } from "../../config/params";
 import beginContext from "../../utils/beginContext";
@@ -20,9 +20,9 @@ const METHOD_NAME = "function.test.overrideCompute";
  * @type TComputeSchema
  * @description Type for partial compute schema updates, requiring computeName and allowing other IComputeSchema properties.
  */
-type TComputeSchema = {
-  computeName: IComputeSchema["computeName"];
-} & Partial<IComputeSchema>;
+type TComputeSchema<T extends IComputeData = any> = {
+  computeName: IComputeSchema<T>["computeName"];
+} & Partial<IComputeSchema<T>>;
 
 /**
  * Function implementation
@@ -43,6 +43,6 @@ const overrideComputeInternal = beginContext((publicComputeSchema: TComputeSchem
  * @param {TComputeSchema} computeSchema - The partial compute schema with updates.
  * @returns {IComputeSchema} The updated compute schema.
  */
-export function overrideCompute(computeSchema: TComputeSchema) {
+export function overrideCompute<T extends IComputeData = any>(computeSchema: TComputeSchema<T>) {
   return overrideComputeInternal(computeSchema)
 }

src/functions/test/overrideOutline.ts

@@ -1,4 +1,4 @@
-import { IOutlineSchema } from "../../interfaces/Outline.interface";
+import { IOutlineData, IOutlineParam, IOutlineSchema } from "../../interfaces/Outline.interface";
 import swarm from "../../lib";
 import { GLOBAL_CONFIG } from "../../config/params";
 import beginContext from "../../utils/beginContext";
@@ -19,9 +19,12 @@ const METHOD_NAME = "function.test.overrideOutline";
  * @property {IOutlineSchema["outlineName"]} outlineName - The unique name of the outline to override.
  * @property {Partial<IOutlineSchema>} [partial] - Optional partial properties of the `IOutlineSchema` to override.
  */
-type TOutlineSchema = {
-  outlineName: IOutlineSchema["outlineName"];
-} & Partial<IOutlineSchema>;
+type TOutlineSchema<
+  Data extends IOutlineData = IOutlineData,
+  Param extends IOutlineParam = IOutlineParam
+> = {
+  outlineName: IOutlineSchema<Data, Param>["outlineName"];
+} & Partial<IOutlineSchema<Data, Param>>;
 
 /**
  * Internal implementation of the outline override logic, wrapped in a clean context.
@@ -35,7 +38,7 @@ const overrideOutlineInternal = beginContext(
       swarm.loggerService.log(METHOD_NAME, {
         outlineSchema: publicOutlineSchema,
       });
-    
+
     const outlineSchema = removeUndefined(publicOutlineSchema);
 
     return swarm.outlineSchemaService.override(
@@ -51,6 +54,9 @@ const overrideOutlineInternal = beginContext(
  * Logs the operation if logging is enabled in the global configuration.
  * @param {TOutlineSchema} outlineSchema - The partial outline schema containing the outline name and optional schema properties to override.
  */
-export function overrideOutline(outlineSchema: TOutlineSchema) {
+export function overrideOutline<
+  Data extends IOutlineData = IOutlineData,
+  Param extends IOutlineParam = IOutlineParam
+>(outlineSchema: TOutlineSchema<Data, Param>) {
   return overrideOutlineInternal(outlineSchema);
 }

src/functions/test/overrideTool.ts

@@ -1,19 +1,19 @@
-import { IAgentTool } from "../../interfaces/Agent.interface";
+import { IAgentTool, ToolValue } from "../../interfaces/Agent.interface";
 import swarm from "../../lib";
 import { GLOBAL_CONFIG } from "../../config/params";
 import beginContext from "../../utils/beginContext";
 import removeUndefined from "../../helpers/removeUndefined";
 
 const METHOD_NAME = "function.test.overrideTool";
 
-type TAgentTool = {
-  toolName: IAgentTool["toolName"];
-} & Partial<IAgentTool>;
+type TAgentTool<T extends any = Record<string, ToolValue>> = {
+  toolName: IAgentTool<T>["toolName"];
+} & Partial<IAgentTool<T>>;
 
 /**
  * Function implementation
  */
-const overrideToolInternal = beginContext((publicToolSchema: TAgentTool) => {
+const overrideToolInternal = beginContext((publicToolSchema: TAgentTool<unknown>) => {
   GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
     swarm.loggerService.log(METHOD_NAME, {
       toolSchema: publicToolSchema,
@@ -45,6 +45,6 @@ const overrideToolInternal = beginContext((publicToolSchema: TAgentTool) => {
  * });
  * // Logs the operation (if enabled) and updates the tool schema in the swarm.
  */
-export function overrideTool(toolSchema: TAgentTool) {
+export function overrideTool<T extends any = Record<string, ToolValue>>(toolSchema: TAgentTool<T>) {
   return overrideToolInternal(toolSchema);
 }

types.d.ts

@@ -12464,9 +12464,9 @@ type TSwarmSchema = {
  */
 declare function overrideSwarm(swarmSchema: TSwarmSchema): ISwarmSchema;
 
-type TAgentTool = {
-    toolName: IAgentTool["toolName"];
-} & Partial<IAgentTool>;
+type TAgentTool<T extends any = Record<string, ToolValue>> = {
+    toolName: IAgentTool<T>["toolName"];
+} & Partial<IAgentTool<T>>;
 /**
  * Overrides an existing tool schema in the swarm system with a new or partial schema.
  * This function updates the configuration of a tool identified by its `toolName`, applying the provided schema properties.
@@ -12488,7 +12488,7 @@ type TAgentTool = {
  * });
  * // Logs the operation (if enabled) and updates the tool schema in the swarm.
  */
-declare function overrideTool(toolSchema: TAgentTool): IAgentTool<Record<string, ToolValue>>;
+declare function overrideTool<T extends any = Record<string, ToolValue>>(toolSchema: TAgentTool<T>): IAgentTool<Record<string, ToolValue>>;
 
 /**
  * Type definition for a partial MCP schema, requiring at least an mcpName.
@@ -12538,15 +12538,15 @@ declare function overrideWiki(wikiSchema: TWikiSchema): IWikiSchema;
  * @type TComputeSchema
  * @description Type for partial compute schema updates, requiring computeName and allowing other IComputeSchema properties.
  */
-type TComputeSchema = {
-    computeName: IComputeSchema["computeName"];
-} & Partial<IComputeSchema>;
+type TComputeSchema<T extends IComputeData = any> = {
+    computeName: IComputeSchema<T>["computeName"];
+} & Partial<IComputeSchema<T>>;
 /**
  * Overrides an existing compute schema with provided partial updates.
  * @param {TComputeSchema} computeSchema - The partial compute schema with updates.
  * @returns {IComputeSchema} The updated compute schema.
  */
-declare function overrideCompute(computeSchema: TComputeSchema): IComputeSchema<any>;
+declare function overrideCompute<T extends IComputeData = any>(computeSchema: TComputeSchema<T>): IComputeSchema<any>;
 
 /**
  * @module overridePipeline
@@ -12568,16 +12568,16 @@ declare function overridePipeline<Payload extends object = any>(pipelineSchema:
  * @property {IOutlineSchema["outlineName"]} outlineName - The unique name of the outline to override.
  * @property {Partial<IOutlineSchema>} [partial] - Optional partial properties of the `IOutlineSchema` to override.
  */
-type TOutlineSchema = {
-    outlineName: IOutlineSchema["outlineName"];
-} & Partial<IOutlineSchema>;
+type TOutlineSchema<Data extends IOutlineData = IOutlineData, Param extends IOutlineParam = IOutlineParam> = {
+    outlineName: IOutlineSchema<Data, Param>["outlineName"];
+} & Partial<IOutlineSchema<Data, Param>>;
 /**
  * Overrides an existing outline schema in the swarm system by updating it with the provided partial schema.
  * Ensures the operation runs in a clean context using `beginContext` to avoid interference from existing method or execution contexts.
  * Logs the operation if logging is enabled in the global configuration.
  * @param {TOutlineSchema} outlineSchema - The partial outline schema containing the outline name and optional schema properties to override.
  */
-declare function overrideOutline(outlineSchema: TOutlineSchema): IOutlineSchema<any, any>;
+declare function overrideOutline<Data extends IOutlineData = IOutlineData, Param extends IOutlineParam = IOutlineParam>(outlineSchema: TOutlineSchema<Data, Param>): IOutlineSchema<any, any>;
 
 /**
  * Marks a client as online in the specified swarm.