wip

Author: tripolskypetr

src/contract/OutlineCompletion.contract.ts

@@ -1,18 +1,14 @@
-import { IBaseMessage } from "./BaseMessage.contract";
 import { IBaseCompletionArgs } from "./BaseCompletion.contract";
 import { OutlineName, IOutlineFormat } from "../interfaces/Outline.interface";
-import { OutlineMessageRole } from "./OutlineMessage.contract";
+import { IOutlineMessage, OutlineMessageRole } from "./OutlineMessage.contract";
 
 /**
  * Interface representing the arguments for outline (JSON) completions.
  * Extends base completion args with outline-specific fields for structured JSON output.
  * Used for completions that return data conforming to a predefined schema.
- * @template Message - The type of message, extending IBaseMessage with any role type. Defaults to IBaseMessage with string role.
  * @interface IOutlineCompletionArgs
  */
-export interface IOutlineCompletionArgs<
-  Message extends IBaseMessage<any> = IBaseMessage<OutlineMessageRole>
-> extends IBaseCompletionArgs<Message> {
+export interface IOutlineCompletionArgs extends IBaseCompletionArgs<IOutlineMessage> {
   /**
    * The outline schema name (required).
    * Defines the structure of the expected JSON response.

src/contract/SwarmCompletion.contract.ts

@@ -1,20 +1,16 @@
-import { IBaseMessage } from "./BaseMessage.contract";
 import { IBaseCompletionArgs } from "./BaseCompletion.contract";
 import { AgentName } from "../interfaces/Agent.interface";
 import { ITool } from "../model/Tool.model";
-import { SwarmMessageRole } from "./SwarmMessage.contract";
+import { ISwarmMessage, SwarmMessageRole } from "./SwarmMessage.contract";
 import { ExecutionMode } from "../interfaces/Session.interface";
 
 /**
  * Interface representing the arguments for swarm (chat) completions.
  * Extends base completion args with swarm-specific fields for agent-based interactions.
  * Used for agent completions with tool support, client tracking, and multi-agent context.
- * @template Message - The type of message, extending IBaseMessage with any role type. Defaults to IBaseMessage with string role.
  * @interface ISwarmCompletionArgs
  */
-export interface ISwarmCompletionArgs<
-  Message extends IBaseMessage<any> = IBaseMessage<SwarmMessageRole>
-> extends IBaseCompletionArgs<Message> {
+export interface ISwarmCompletionArgs extends IBaseCompletionArgs<ISwarmMessage> {
   /**
    * The agent name (required).
    * Identifies the agent context for the completion.

src/interfaces/Completion.interface.ts

@@ -1,9 +1,9 @@
 import { IBaseMessage } from "../contract/BaseMessage.contract";
 import { IBaseCompletionArgs } from "../contract/BaseCompletion.contract";
 import { addCompletion } from "../functions/setup/addCompletion";
-import { ISwarmCompletionArgs } from "../contract/SwarmCompletion.contract";
 import { ISwarmMessage } from "../contract/SwarmMessage.contract";
 import { IOutlineCompletionArgs } from "../contract/OutlineCompletion.contract";
+import { ISwarmCompletionArgs } from "src/contract/SwarmCompletion.contract";
 
 /**
  * Interface representing a completion mechanism.
@@ -29,8 +29,12 @@ export interface ICompletionCallbacks<Message extends IBaseMessage<any> = IBaseM
  * Interface representing the schema for configuring a completion mechanism.
  * Defines how completions are generated within the swarm.
  * @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.
 */
-export interface ICompletionSchema<Message extends IBaseMessage<string> = IBaseMessage<any>> {
+export 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;
 
@@ -39,12 +43,7 @@ export interface ICompletionSchema<Message extends IBaseMessage<string> = IBaseM
    * Generates a model response using the given context and tools.
    * @throws {Error} If completion generation fails (e.g., due to invalid arguments, model errors, or tool issues).
    */
-  getCompletion(
-    args:
-      | IBaseCompletionArgs<IBaseMessage<string>>
-      | IOutlineCompletionArgs<IBaseMessage<string>>
-      | ISwarmCompletionArgs<IBaseMessage<string>>
-  ): Promise<Message>;
+  getCompletion(args: Args): Promise<Message>;
 
   /*
    * Flag if the completion is a JSON completion.
@@ -65,3 +64,12 @@ export interface ICompletionSchema<Message extends IBaseMessage<string> = IBaseM
  * Used to identify and reference specific completion implementations.
 */
 export type CompletionName = string;
+
+
+addCompletion({
+  completionName: "test",
+  getCompletion: async (params: ISwarmCompletionArgs): Promise<ISwarmMessage> => {
+    params.messages[0].role === "assistant"
+    return null as never;
+  }
+})