@llumiverse/common 0.22.0 → 0.23.0-dev-20251118

package/src/types.ts

@@ -64,7 +64,7 @@ export const ProviderList: Record<Providers, ProviderParams> = {
     replicate:
     {
         id: Providers.replicate,
-        name: "Repicate",
+        name: "Replicate",
         requiresApiKey: true,
         requiresEndpointUrl: false,
         supportSearch: true,
@@ -152,20 +152,40 @@ export interface EmbeddingsResult {
 export interface ResultValidationError {
     code: 'validation_error' | 'json_error' | 'content_policy_violation';
     message: string;
-    data?: string;
+    data?: CompletionResult[];
 }
 
-//ResultT should be either JSONObject or string
+// ============== Result Types ===============
+
+export interface BaseResult {
+    type: "text" | "json" | "image";
+    value: any;
+}
+
+export interface TextResult extends BaseResult {
+    type: "text";
+    value: string;
+}
+
+export interface JsonResult extends BaseResult {
+    type: "json";
+}
+
+export interface ImageResult extends BaseResult {
+    type: "image";
+    value: string; // base64 data url or real url
+}
+
+export type CompletionResult = TextResult | JsonResult | ImageResult;
+
+
 //Internal structure used in driver implementation.
-export interface CompletionChunkObject<ResultT = any> {
-    result: ResultT;
+export interface CompletionChunkObject {
+    result: CompletionResult[];
     token_usage?: ExecutionTokenUsage;
     finish_reason?: "stop" | "length" | string;
 }
 
-//Internal structure used in driver implementation.
-export type CompletionChunk = CompletionChunkObject | string;
-
 export interface ToolDefinition {
     name: string,
     description?: string,
@@ -185,10 +205,9 @@ export interface ToolUse<ParamsT = JSONObject> {
     tool_input: ParamsT | null
 }
 
-//ResultT should be either JSONObject or string
-export interface Completion<ResultT = any> {
+export interface Completion {
     // the driver impl must return the result and optionally the token_usage. the execution time is computed by the extended abstract driver
-    result: ResultT;
+    result: CompletionResult[];
     token_usage?: ExecutionTokenUsage;
     /**
      * Contains the tools from which the model awaits information.
@@ -216,12 +235,6 @@ export interface Completion<ResultT = any> {
     conversation?: unknown;
 }
 
-export interface ImageGeneration {
-
-    images?: string[];
-
-}
-
 export interface ExecutionResponse<PromptT = any> extends Completion {
     prompt: PromptT;
     /**
@@ -239,11 +252,24 @@ export interface CompletionStream<PromptT = any> extends AsyncIterable<string> {
     completion: ExecutionResponse<PromptT> | undefined;
 }
 
+/**
+ * Minimal logger interface for LLM drivers.
+ * Follows pino v10 signature: supports message-only or object-first (never message-first).
+ * - Message-only: logger.info("message")
+ * - Object-first: logger.info({ data }, "message")
+ * - PREVENTS: logger.info("message", { data }) - compile error (objects not allowed in ...args)
+ * 
+ * Additional args must be primitives (string | number | boolean) for string interpolation.
+ */
 export interface Logger {
-    debug: (...obj: any[]) => void;
-    info: (...obj: any[]) => void;
-    warn: (...obj: any[]) => void;
-    error: (...obj: any[]) => void;
+    debug(msg: string, ...args: (string | number | boolean)[]): void;
+    debug<T>(obj: T, msg?: T extends string ? never : string, ...args: (string | number | boolean)[]): void;
+    info(msg: string, ...args: (string | number | boolean)[]): void;
+    info<T>(obj: T, msg?: T extends string ? never : string, ...args: (string | number | boolean)[]): void;
+    warn(msg: string, ...args: (string | number | boolean)[]): void;
+    warn<T>(obj: T, msg?: T extends string ? never : string, ...args: (string | number | boolean)[]): void;
+    error(msg: string, ...args: (string | number | boolean)[]): void;
+    error<T>(obj: T, msg?: T extends string ? never : string, ...args: (string | number | boolean)[]): void;
 }
 
 export interface DriverOptions {
@@ -533,7 +559,7 @@ export interface TrainingOptions {
 
 export interface TrainingPromptOptions {
     segments: PromptSegment[];
-    completion: string | JSONObject;
+    completion: CompletionResult[]
     model: string; // the model to train
     schema?: JSONSchema; // the result schema f any
 }