@stainless-api/docs-ai-chat 0.1.0-beta.40 → 0.1.0-beta.43

package/CHANGELOG.md

@@ -1,6 +1,28 @@
 # @stainless-api/docs-ai-chat
 
-## 1.0.0-beta.40
+## 0.1.0-beta.43
+
+### Patch Changes
+
+- Updated dependencies [21e50d3]
+  - @stainless-api/docs-ui@0.1.0-beta.77
+
+## 0.1.0-beta.42
+
+### Patch Changes
+
+- Updated dependencies [95fe66c]
+- Updated dependencies [5701494]
+  - @stainless-api/docs-ui@0.1.0-beta.76
+  - @stainless-api/ai-chat@0.1.0-beta.6
+
+## 0.1.0-beta.41
+
+### Patch Changes
+
+- 6e762e2: fix steelie for multi-turn conversations
+
+## 0.1.0-beta.40
 
 ### Patch Changes
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stainless-api/docs-ai-chat",
-  "version": "0.1.0-beta.40",
+  "version": "0.1.0-beta.43",
   "publishConfig": {
     "access": "public"
   },
@@ -8,12 +8,12 @@
   "peerDependencies": {
     "react": ">=19.0.0",
     "react-dom": ">=19.0.0",
-    "@stainless-api/docs-ui": "0.1.0-beta.75"
+    "@stainless-api/docs-ui": "0.1.0-beta.77"
   },
   "dependencies": {
     "@streamparser/json-whatwg": "^0.0.22",
     "zod": "^4.3.5",
-    "@stainless-api/ai-chat": "0.1.0-beta.5"
+    "@stainless-api/ai-chat": "0.1.0-beta.6"
   },
   "devDependencies": {
     "@types/react": "19.2.10",

package/src/api/index.ts

@@ -41,13 +41,13 @@ export async function* getChatResponse(
     query,
     project,
     language,
-    sessionId,
+    priorMessages,
     siteTitle,
   }: {
     query: string;
     project: string;
     language: DocsLanguage;
-    sessionId: string | undefined;
+    priorMessages: NonNullable<RequestBody['additionalContext']>['prior_messages'];
     siteTitle: string | undefined;
   },
   abortSignal: AbortSignal,
@@ -61,8 +61,8 @@ export async function* getChatResponse(
       query,
       sdk: { project, language },
       stream: true,
-      session_id: sessionId,
       additionalContext: {
+        prior_messages: priorMessages,
         intent: getPageContext({ siteTitle }),
       },
       browser_id: getClientId(),

package/src/api/schemas.ts

@@ -17,6 +17,12 @@ export const requestBody = z.object({
   session_id: z.string().optional(),
   additionalContext: z
     .object({
+      prior_messages: z.array(
+        z.object({
+          role: z.enum(['user', 'assistant']),
+          content: z.string(),
+        }),
+      ),
       code: z.string().optional(),
       intent: z.string().optional(),
       lsp: z.string().optional(),

package/src/hook.ts

@@ -130,7 +130,6 @@ export function useChat({
     return () => ac.abort('Component unmounted');
   }, []);
 
-  const sessionId = useRef<string | undefined>(undefined);
   const [chatMessages, dispatch] = useReducer(chatReducer, []);
 
   /** Send a message and stream back the response in chat */
@@ -143,23 +142,21 @@ export function useChat({
       let lastChunkType: ResponseChunk['type'] | undefined = undefined;
 
       try {
-        for await (const chunk of getChatResponse(
+        let chunk: ResponseChunk | undefined = undefined;
+        for await (chunk of getChatResponse(
           {
             query: question,
             project: projectId,
             language,
-            sessionId: sessionId.current,
+            priorMessages: chatMessages.filter(
+              (msg) => msg.role === 'user' || (msg.role === 'assistant' && msg.messageType === 'text'),
+            ),
             siteTitle,
           },
           abortController.current.signal,
         )) {
           if (abortController.current.signal.aborted) break;
 
-          // store session id at start of session
-          if (chunk.type === 'start_session') {
-            sessionId.current = chunk.session_id;
-          }
-
           // mark complete when text messages finish streaming
           if (lastChunkType === 'text' && chunk.type !== 'text') {
             dispatch({ type: 'completeMessage', id: currentResponseId });
@@ -204,6 +201,13 @@ export function useChat({
 
           lastChunkType = chunk.type;
         }
+        if (!chunk) {
+          dispatch({
+            type: 'addError',
+            respondingTo: userMessageId,
+            errorMessage: 'No response received. Please try again.',
+          });
+        }
       } catch {
         dispatch({
           type: 'addError',
@@ -212,7 +216,7 @@ export function useChat({
         });
       }
     },
-    [language, projectId, siteTitle],
+    [language, projectId, siteTitle, chatMessages],
   );
 
   return { chatMessages, sendMessage };