patch

Author: tripolskypetr

src/functions/setup/addCompletion.ts

@@ -3,12 +3,14 @@ import swarm from "../../lib";
 import { GLOBAL_CONFIG } from "../../config/params";
 import beginContext from "../../utils/beginContext";
 import mapCompletionSchema from "../../helpers/mapCompletionSchema";
+import { IBaseMessage } from "../../contract/BaseMessage.contract";
+import { IBaseCompletionArgs } from "../../contract/BaseCompletion.contract";
 
 const METHOD_NAME = "function.setup.addCompletion";
 
 /**
  * Function implementation
-*/
+ */
 const addCompletionInternal = beginContext(
   (completionPublicSchema: ICompletionSchema) => {
     // Log the operation details if logging is enabled in GLOBAL_CONFIG
@@ -48,7 +50,12 @@ const addCompletionInternal = beginContext(
  * const completionSchema = { completionName: "OpenAI", model: "gpt-3.5-turbo" };
  * const completionName = addCompletion(completionSchema);
  * console.log(completionName); // Outputs "OpenAI"
-*/
-export function addCompletion(completionSchema: ICompletionSchema) {
+ */
+export function addCompletion<
+  Message extends IBaseMessage<any> = IBaseMessage<string>,
+  Args extends IBaseCompletionArgs<IBaseMessage<string>> = IBaseCompletionArgs<
+    IBaseMessage<string>
+  >
+>(completionSchema: ICompletionSchema<Message, Args>) {
   return addCompletionInternal(completionSchema);
 }

src/functions/target/chat.ts

@@ -8,8 +8,8 @@ import {
 import { randomString } from "functools-kit";
 import { errorSubject } from "../../config/emitters";
 import { IBaseMessage } from "../../contract/BaseMessage.contract";
-import { ISwarmMessage } from "src/contract/SwarmMessage.contract";
-import { ISwarmCompletionArgs } from "src/contract/SwarmCompletion.contract";
+import { ISwarmMessage } from "../../contract/SwarmMessage.contract";
+import { ISwarmCompletionArgs } from "../../contract/SwarmCompletion.contract";
 
 const METHOD_NAME = "function.target.chat";
 

src/interfaces/Completion.interface.ts

@@ -1,6 +1,27 @@
 import { IBaseMessage } from "../contract/BaseMessage.contract";
 import { IBaseCompletionArgs } from "../contract/BaseCompletion.contract";
 
+/*
+import { addCompletion } from "../functions/setup/addCompletion";
+import { ISwarmMessage } from "../contract/SwarmMessage.contract";
+import { IOutlineCompletionArgs } from "../contract/OutlineCompletion.contract";
+import { ISwarmCompletionArgs } from "../contract/SwarmCompletion.contract";
+
+
+addCompletion({
+  completionName: "test",
+  getCompletion: async (
+    params: ISwarmCompletionArgs
+  ): Promise<ISwarmMessage> => {
+    params.messages[0].role === "assistant";
+    return null as never;
+  },
+  callbacks: {
+    onComplete: (args) => {}
+  }
+});
+*/
+
 /**
  * Interface representing a completion mechanism.
  * Extends the completion schema to provide a complete API for generating model responses.
@@ -15,9 +36,9 @@ export interface ICompletion extends ICompletionSchema {}
  */
 export interface ICompletionCallbacks<
   Message extends IBaseMessage<any> = IBaseMessage<string>,
-  Args extends IBaseCompletionArgs<
+  Args extends IBaseCompletionArgs<IBaseMessage<string>> = IBaseCompletionArgs<
     IBaseMessage<string>
-  > = IBaseCompletionArgs<Message>
+  >
 > {
   /**
    * Optional callback triggered after a completion is successfully generated.
@@ -34,9 +55,9 @@ export interface ICompletionCallbacks<
  */
 export interface ICompletionSchema<
   Message extends IBaseMessage<string> = IBaseMessage<any>,
-  Args extends IBaseCompletionArgs<
+  Args extends IBaseCompletionArgs<IBaseMessage<string>> = IBaseCompletionArgs<
     IBaseMessage<string>
-  > = IBaseCompletionArgs<Message>
+  >
 > {
   /** The unique name of the completion mechanism within the swarm.*/
   completionName: CompletionName;

types.d.ts

@@ -2515,7 +2515,7 @@ interface ICompletion extends ICompletionSchema {
  * Provides hooks for post-completion actions.
  * @template Message - The type of message, extending IBaseMessage with any role type. Defaults to IBaseMessage with string role.
  */
-interface ICompletionCallbacks<Message extends IBaseMessage<any> = IBaseMessage<string>, Args extends IBaseCompletionArgs<IBaseMessage<string>> = IBaseCompletionArgs<Message>> {
+interface ICompletionCallbacks<Message extends IBaseMessage<any> = IBaseMessage<string>, Args extends IBaseCompletionArgs<IBaseMessage<string>> = IBaseCompletionArgs<IBaseMessage<string>>> {
     /**
      * Optional callback triggered after a completion is successfully generated.
      * Useful for logging, output processing, or triggering side effects.
@@ -2528,7 +2528,7 @@ interface ICompletionCallbacks<Message extends IBaseMessage<any> = IBaseMessage<
  * @template Message - The type of message, extending IBaseMessage with any role type. Defaults to IBaseMessage for maximum flexibility.
  * @template Args - The type of completion arguments, defaults to any completion args type.
  */
-interface ICompletionSchema<Message extends IBaseMessage<string> = IBaseMessage<any>, Args extends IBaseCompletionArgs<IBaseMessage<string>> = IBaseCompletionArgs<Message>> {
+interface ICompletionSchema<Message extends IBaseMessage<string> = IBaseMessage<any>, Args extends IBaseCompletionArgs<IBaseMessage<string>> = IBaseCompletionArgs<IBaseMessage<string>>> {
     /** The unique name of the completion mechanism within the swarm.*/
     completionName: CompletionName;
     /**
@@ -4824,7 +4824,7 @@ declare class CompletionSchemaService {
      * Supports dynamic updates to completion schemas used by AgentSchemaService, ClientAgent, and other swarm components.
      * @throws {Error} If the key does not exist in the registry (inherent to ToolRegistry.override behavior).
      */
-    override: (key: CompletionName, value: Partial<ICompletionSchema>) => ICompletionSchema<IBaseMessage<any>, IBaseCompletionArgs<IBaseMessage<any>>>;
+    override: (key: CompletionName, value: Partial<ICompletionSchema>) => ICompletionSchema<IBaseMessage<any>, IBaseCompletionArgs<IBaseMessage<string>>>;
     /**
      * Retrieves a completion schema from the registry by its name.
      * Fetches the schema from ToolRegistry using the provided key, logging the operation via LoggerService if GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO is true.
@@ -10330,8 +10330,8 @@ declare function addAgent(agentSchema: IAgentSchema): string;
  * const completionSchema = { completionName: "OpenAI", model: "gpt-3.5-turbo" };
  * const completionName = addCompletion(completionSchema);
  * console.log(completionName); // Outputs "OpenAI"
-*/
-declare function addCompletion(completionSchema: ICompletionSchema): string;
+ */
+declare function addCompletion<Message extends IBaseMessage<any> = IBaseMessage<string>, Args extends IBaseCompletionArgs<IBaseMessage<string>> = IBaseCompletionArgs<IBaseMessage<string>>>(completionSchema: ICompletionSchema<Message, Args>): string;
 
 /**
  * Adds a new swarm to the system for managing client sessions.
@@ -10540,7 +10540,7 @@ type TCompletionSchema = {
  * });
  * // Logs the operation (if enabled) and updates the completion schema in the swarm.
 */
-declare function overrideCompletion(completionSchema: TCompletionSchema): ICompletionSchema<IBaseMessage<any>, IBaseCompletionArgs<IBaseMessage<any>>>;
+declare function overrideCompletion(completionSchema: TCompletionSchema): ICompletionSchema<IBaseMessage<any>, IBaseCompletionArgs<IBaseMessage<string>>>;
 
 /**
  * Type representing a partial embedding schema with required embeddingName.
@@ -11853,7 +11853,7 @@ declare function getAgent(agentName: AgentName): IAgentSchemaInternal;
  * @function getCompletion
  * @param {CompletionName} completionName - The name of the completion.
 */
-declare function getCompletion(completionName: CompletionName): ICompletionSchema<IBaseMessage<any>, IBaseCompletionArgs<IBaseMessage<any>>>;
+declare function getCompletion(completionName: CompletionName): ICompletionSchema<IBaseMessage<any>, IBaseCompletionArgs<IBaseMessage<string>>>;
 
 /**
  * Retrieves a compute schema by its name from the swarm's compute schema service.