graphlit-client 1.0.20250924001 → 1.0.20250924002

package/dist/client.js

@@ -6,7 +6,7 @@ import { RetryLink } from "@apollo/client/link/retry/index.js";
 import * as Types from "./generated/graphql-types.js";
 import * as Documents from "./generated/graphql-documents.js";
 import * as dotenv from "dotenv";
-import { getServiceType, getModelName } from "./model-mapping.js";
+import { getServiceType, getModelName, getModelEnum } from "./model-mapping.js";
 import { UIEventAdapter } from "./streaming/ui-event-adapter.js";
 import { formatMessagesForOpenAI, formatMessagesForAnthropic, formatMessagesForGoogle, formatMessagesForMistral, formatMessagesForBedrock, } from "./streaming/llm-formatters.js";
 import { streamWithOpenAI, streamWithAnthropic, streamWithGoogle, streamWithGroq, streamWithCerebras, streamWithCohere, streamWithMistral, streamWithBedrock, streamWithDeepseek, streamWithXai, } from "./streaming/providers.js";
@@ -2174,12 +2174,14 @@ class Graphlit {
             }
             // Create UI event adapter with model information
             const modelName = fullSpec ? getModelName(fullSpec) : undefined;
+            const modelEnum = fullSpec ? getModelEnum(fullSpec) : undefined;
             const serviceType = fullSpec ? getServiceType(fullSpec) : undefined;
             uiAdapter = new UIEventAdapter(onEvent, actualConversationId, {
                 smoothingEnabled: options?.smoothingEnabled ?? true,
                 chunkingStrategy: options?.chunkingStrategy ?? "word",
                 smoothingDelay: options?.smoothingDelay ?? 30,
-                model: modelName,
+                model: modelEnum,
+                modelName: modelName,
                 modelService: serviceType,
             });
             // Start the streaming conversation

package/dist/model-mapping.d.ts

@@ -16,3 +16,9 @@ export declare function isStreamingSupported(serviceType?: string): boolean;
  * @returns The service type string
  */
 export declare function getServiceType(specification: any): string | undefined;
+/**
+ * Get the model enum value from specification
+ * @param specification - The specification object
+ * @returns The model enum value
+ */
+export declare function getModelEnum(specification: any): string | undefined;

package/dist/model-mapping.js

@@ -264,3 +264,35 @@ export function isStreamingSupported(serviceType) {
 export function getServiceType(specification) {
     return specification?.serviceType;
 }
+/**
+ * Get the model enum value from specification
+ * @param specification - The specification object
+ * @returns The model enum value
+ */
+export function getModelEnum(specification) {
+    const serviceType = specification?.serviceType;
+    switch (serviceType) {
+        case Types.ModelServiceTypes.OpenAi:
+            return specification?.openAI?.model;
+        case Types.ModelServiceTypes.Anthropic:
+            return specification?.anthropic?.model;
+        case Types.ModelServiceTypes.Google:
+            return specification?.google?.model;
+        case Types.ModelServiceTypes.Groq:
+            return specification?.groq?.model;
+        case Types.ModelServiceTypes.Cerebras:
+            return specification?.cerebras?.model;
+        case Types.ModelServiceTypes.Cohere:
+            return specification?.cohere?.model;
+        case Types.ModelServiceTypes.Mistral:
+            return specification?.mistral?.model;
+        case Types.ModelServiceTypes.Bedrock:
+            return specification?.bedrock?.model;
+        case Types.ModelServiceTypes.Deepseek:
+            return specification?.deepseek?.model;
+        case Types.ModelServiceTypes.Xai:
+            return specification?.xai?.model;
+        default:
+            return undefined;
+    }
+}

package/dist/streaming/ui-event-adapter.d.ts

@@ -9,6 +9,7 @@ export declare class UIEventAdapter {
     private onEvent;
     private conversationId;
     private model?;
+    private modelName?;
     private modelService?;
     private tokenCount;
     private currentMessage;
@@ -39,6 +40,7 @@ export declare class UIEventAdapter {
         chunkingStrategy?: ChunkingStrategy;
         smoothingDelay?: number;
         model?: string;
+        modelName?: string;
         modelService?: string;
     });
     /**

package/dist/streaming/ui-event-adapter.js

@@ -7,7 +7,8 @@ import { ChunkBuffer } from "./chunk-buffer.js";
 export class UIEventAdapter {
     onEvent;
     conversationId;
-    model;
+    model; // This will now be the enum value
+    modelName; // This will be the actual model name (e.g., "claude-sonnet-4-0")
     modelService;
     tokenCount = 0;
     currentMessage = "";
@@ -38,6 +39,7 @@ export class UIEventAdapter {
         this.conversationId = conversationId;
         this.smoothingDelay = options.smoothingDelay ?? 30;
         this.model = options.model;
+        this.modelName = options.modelName;
         this.modelService = options.modelService;
         this.conversationStartTime = Date.now(); // Capture when conversation began
         if (options.smoothingEnabled) {
@@ -333,6 +335,7 @@ export class UIEventAdapter {
             tokens: tokens, // Now we have the actual LLM token count!
             toolCalls: Array.from(this.activeToolCalls.values()).map((t) => t.toolCall),
             model: this.model,
+            modelName: this.modelName,
             modelService: this.modelService,
         };
         // Add final timing metadata
@@ -514,6 +517,9 @@ export class UIEventAdapter {
         if (this.model) {
             message.model = this.model;
         }
+        if (this.modelName) {
+            message.modelName = this.modelName;
+        }
         if (this.modelService) {
             message.modelService = this.modelService;
         }

package/dist/types/ui-events.d.ts

@@ -20,6 +20,13 @@ export type ContextWindowEvent = {
     };
     timestamp: Date;
 };
+/**
+ * Extended conversation message with additional streaming metadata
+ */
+export type StreamingConversationMessage = Partial<ConversationMessage> & {
+    message: string;
+    modelName?: string;
+};
 /**
  * Simplified UI-focused streaming events using GraphQL types
  */
@@ -30,9 +37,7 @@ export type AgentStreamEvent = {
     model?: string;
 } | ContextWindowEvent | {
     type: "message_update";
-    message: Partial<ConversationMessage> & {
-        message: string;
-    };
+    message: StreamingConversationMessage;
     isStreaming: boolean;
     metrics?: {
         ttft?: number;
@@ -55,7 +60,7 @@ export type AgentStreamEvent = {
     isComplete: boolean;
 } | {
     type: "conversation_completed";
-    message: ConversationMessage;
+    message: StreamingConversationMessage;
     metrics?: {
         ttft?: number;
         totalTime: number;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphlit-client",
-  "version": "1.0.20250924001",
+  "version": "1.0.20250924002",
   "description": "Graphlit API Client for TypeScript",
   "type": "module",
   "main": "./dist/client.js",