wirejs-resources 0.1.150-llm → 0.1.151-llm

package/dist/services/llm.d.ts

@@ -10,15 +10,25 @@ export type ToolCall = {
         arguments: Record<string, any>;
     };
 };
-export type LLMMessage = {
-    /** Standard user or assistant message */
-    role: 'user' | 'assistant';
+/** User message in conversation */
+export type UserMessage = {
+    /** Always 'user' for user messages */
+    role: 'user';
     /** Text content of the message */
     content: string;
-    /** Tool calls requested by the assistant (only for assistant messages) */
+};
+/** Assistant message type - what LLM methods actually return */
+export type AssistantMessage = {
+    /** Always 'assistant' for LLM responses */
+    role: 'assistant';
+    /** Text content of the message */
+    content: string;
+    /** Tool calls requested by the assistant */
     tool_calls?: ToolCall[];
-} | {
-    /** Tool execution result message */
+};
+/** Tool execution result message */
+export type ToolMessage = {
+    /** Always 'tool' for tool execution results */
     role: 'tool';
     /** Result content from the tool execution */
     content: string;
@@ -27,6 +37,8 @@ export type LLMMessage = {
     /** ID linking this result to the original tool call */
     tool_call_id: string;
 };
+/** Union of all possible message types in conversation history */
+export type LLMMessage = UserMessage | AssistantMessage | ToolMessage;
 export type LLMChunk = {
     created_at: string;
     message: LLMMessage;
@@ -88,6 +100,6 @@ export declare class LLM extends Resource {
     private checkOllamaAvailable;
     private checkModelExists;
     private createStreamedString;
-    continueConversation({ history, onChunk, timeoutSeconds, systemPrompt, targetContextSize, models, tools, }: ContinueConversationOptions): Promise<LLMMessage>;
+    continueConversation({ history, onChunk, timeoutSeconds, systemPrompt, targetContextSize, models, tools, }: ContinueConversationOptions): Promise<AssistantMessage>;
 }
 export {};

package/dist/services/llm.js

@@ -40,11 +40,9 @@ export class LLM extends Resource {
                 tool_calls = chunk.message.tool_calls;
             }
         }
-        // Ensure we have a valid role, default to assistant if unexpected
-        const finalRole = (role === 'user' || role === 'assistant') ? role : 'assistant';
-        // Build the result message
+        // Build the assistant result message
         const result = {
-            role: finalRole,
+            role: 'assistant',
             content
         };
         if (tool_calls) {
@@ -161,7 +159,18 @@ export class LLM extends Resource {
                     }
                     else {
                         const chunk = await response.json();
-                        return chunk.message;
+                        // Ensure we return an assistant message
+                        const message = chunk.message;
+                        if (message.role === 'assistant') {
+                            return message;
+                        }
+                        else {
+                            // Fallback: create assistant message
+                            return {
+                                role: 'assistant',
+                                content: message.content
+                            };
+                        }
                     }
                 }
             }

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "wirejs-resources",
-	"version": "0.1.150-llm",
+	"version": "0.1.151-llm",
 	"description": "Basic services and server-side resources for wirejs apps",
 	"type": "module",
 	"main": "./dist/index.js",