fix(ask-baml): ask-baml-client is not correctly pointed at sage-backend (#2267)

<!-- ELLIPSIS_HIDDEN -->



> [!IMPORTANT]
> Fix environment variables and endpoints in ask-baml-client to
correctly point to sage-backend.
> 
>   - **Environment Variables**:
> - Change `TARGET_ENV` from `prod` to `production` in
`.github/workflows/publish-docs.yml`.
> - Change `NODE_ENV` to `PINECONE_ENV` in
`.github/workflows/publish-docs.yml`.
>   - **Endpoints and Constants**:
> - Replace hardcoded `API_ENDPOINT` with
`ALGOLIA_SEARCH_CREDENTIALS_ENDPOINT` in `AlgoliaSearch.tsx`.
> - Replace hardcoded `API_ENDPOINT` with `CHAT_ENDPOINT` in
`ChatBot.tsx`.
> - Replace hardcoded `FEEDBACK_API_ENDPOINT` with `FEEDBACK_ENDPOINT`
in `AssistantResponseFeedback.tsx`.
> - Define `SAGE_URL`, `CHAT_ENDPOINT`, `FEEDBACK_ENDPOINT`,
`ALGOLIA_SEARCH_CREDENTIALS_ENDPOINT`, and `ALGOLIA_SEARCH_INDEX_NAME`
in `constants.ts`.
>   - **File Renames**:
> - Rename `typescript/apps/sage-backend/app/api/ask-baml-chat/route.ts`
to `typescript/apps/sage-backend/app/api/ask-baml/chat/route.ts`.
> - Rename
`typescript/apps/sage-backend/app/api/ask-baml-feedback/route.ts` to
`typescript/apps/sage-backend/app/api/ask-baml/feedback/route.ts`.
>   - **Miscellaneous**:
> - Update `PINECONE_INDEX_NAME` logic in `pinecone-api.ts` to use
`PINECONE_ENV`.
> 
> <sup>This description was created by </sup>[<img alt="Ellipsis"
src="https://img.shields.io/badge/Ellipsis-blue?color=175173">](https://www.ellipsis.dev?ref=BoundaryML%2Fbaml&utm_source=github&utm_medium=referral)<sup>
for 927081e. You can
[customize](https://app.ellipsis.dev/BoundaryML/settings/summaries) this
summary. It will automatically update as commits are pushed.</sup>

<!-- ELLIPSIS_HIDDEN -->

Author: sxlijin

typescript/apps/ask-baml-client/src/AlgoliaSearch.tsx

@@ -6,9 +6,8 @@ import { useEffect, useRef, useState } from 'react';
 import { Configure, InstantSearch, useHits, useSearchBox } from 'react-instantsearch';
 import { z } from 'zod';
 import BamlLambWhite from './baml-lamb-white.svg';
+import { ALGOLIA_SEARCH_CREDENTIALS_ENDPOINT, ALGOLIA_SEARCH_INDEX_NAME } from './constants';
 
-const SEARCH_INDEX_NAME = 'fern_docs_search';
-const API_ENDPOINT = 'https://docs.boundaryml.com/api/fern-docs/search/v2/key';
 
 // Zod schema for API response validation
 const SearchCredentialsSchema = z.object({
@@ -21,7 +20,7 @@ type SearchCredentials = z.infer<typeof SearchCredentialsSchema>;
 // Function to fetch Algolia search credentials from API
 async function fetchSearchCredentials(): Promise<SearchCredentials> {
   try {
-    const response = await fetch(API_ENDPOINT);
+    const response = await fetch(ALGOLIA_SEARCH_CREDENTIALS_ENDPOINT);
     if (!response.ok) {
       throw new Error(`Failed to fetch search credentials: ${response.status}`);
     }
@@ -954,7 +953,7 @@ export default function AlgoliaSearch({
   return (
     <div ref={containerRef} style={{ position: 'relative', width: '100%' }}>
       <InstantSearch
-        indexName={SEARCH_INDEX_NAME}
+        indexName={ALGOLIA_SEARCH_INDEX_NAME}
         searchClient={searchClient}
         future={{
           preserveSharedStateOnUnmount: true,

typescript/apps/ask-baml-client/src/ChatBot.tsx

@@ -12,6 +12,7 @@ import {
   resetSessionAtom,
   sessionIdAtom,
 } from './store';
+import { CHAT_ENDPOINT } from './constants';
 
 const OPEN_BY_DEFAULT = true;
 const SESSION_STORAGE_KEY = 'baml-ai-context';
@@ -63,7 +64,7 @@ const serializeError = (
 };
 
 const postDocChat = async (req: QueryRequest) => {
-  const response = await fetch(API_ENDPOINT, {
+  const response = await fetch(CHAT_ENDPOINT, {
     method: 'POST',
     headers: {
       'Content-Type': 'application/json',
@@ -79,11 +80,6 @@ const postDocChat = async (req: QueryRequest) => {
   return QueryResponseSchema.parse(data);
 };
 
-const API_ENDPOINT =
-  process.env.NODE_ENV === 'development'
-    ? 'http://localhost:4000/api/doc-chat'
-    : 'https://boundary-sage-backend.vercel.app/api/ask-baml-chat';
-
 const ChatBot: React.FC<ChatBotProps> = ({ isOpen = OPEN_BY_DEFAULT, onClose }) => {
   const [messages, setMessages] = useAtom(messagesAtom);
   const sessionId = useAtomValue(sessionIdAtom);

typescript/apps/ask-baml-client/src/api-types.ts

@@ -0,0 +1,27 @@
+import { z } from 'zod';
+// IfChange
+export const QueryRequestSchema = z.object({
+  query: z.string(),
+  language_preference: z.string().optional(),
+  prev_messages: z.array(
+    z.object({
+      role: z.enum(['user', 'assistant']),
+      text: z.string(),
+    }),
+  ),
+});
+export type QueryRequest = z.infer<typeof QueryRequestSchema>;
+
+export const QueryResponseSchema = z.object({
+  ranked_docs: z.array(
+    z.object({
+      title: z.string(),
+      url: z.string(),
+      relevance: z.enum(['very-relevant', 'relevant', 'not-relevant']),
+    }),
+  ),
+  answer: z.string().optional().or(z.null()),
+  suggested_messages: z.array(z.string()).optional(),
+});
+export type QueryResponse = z.infer<typeof QueryResponseSchema>;
+// ThenChange baml/sage-backend/app/types.ts

typescript/apps/ask-baml-client/src/constants.ts

@@ -0,0 +1,10 @@
+const SAGE_URL =
+  process.env.NODE_ENV === 'development'
+    ? 'http://localhost:4000'
+    : 'https://boundary-sage-backend.vercel.app';
+export const CHAT_ENDPOINT = `${SAGE_URL}/api/ask-baml/chat`;
+export const FEEDBACK_ENDPOINT = `${SAGE_URL}/api/ask-baml/feedback`;
+
+export const ALGOLIA_SEARCH_CREDENTIALS_ENDPOINT =
+  'https://docs.boundaryml.com/api/fern-docs/search/v2/key';
+export const ALGOLIA_SEARCH_INDEX_NAME = 'fern_docs_search';

typescript/apps/ask-baml-client/src/lib/AssistantResponseFeedback.tsx

@@ -4,16 +4,12 @@ import { ThumbsDown, ThumbsUp } from 'lucide-react';
 import type React from 'react';
 import { useState } from 'react';
 import { messagesAtom, sessionIdAtom } from '../store';
+import { FEEDBACK_ENDPOINT } from '../constants';
 
 interface AssistantResponseFeedbackProps {
   messageId: string;
 }
 
-const FEEDBACK_API_ENDPOINT =
-  process.env.NODE_ENV === 'development'
-    ? 'http://localhost:4000/api/feedback'
-    : 'https://boundary-sage-backend.vercel.app/api/ask-baml-feedback';
-
 export const AssistantResponseFeedback: React.FC<AssistantResponseFeedbackProps> = ({
   messageId,
 }) => {
@@ -96,7 +92,7 @@ export const AssistantResponseFeedback: React.FC<AssistantResponseFeedbackProps>
         messages: filteredMessages,
       };
 
-      const response = await fetch(FEEDBACK_API_ENDPOINT, {
+      const response = await fetch(FEEDBACK_ENDPOINT, {
         method: 'POST',
         headers: {
           'Content-Type': 'application/json',

typescript/apps/sage-backend/app/api/ask-baml-chat/route.ts renamed to typescript/apps/sage-backend/app/api/ask-baml/chat/route.ts

@@ -2,7 +2,7 @@ import { NotionLogger } from '@/lib/notion-api';
 import { QueryRequestSchema } from '@baml/sage-interface';
 import type { NextRequest } from 'next/server';
 import { NextResponse } from 'next/server';
-import { submitQuery } from '../../actions/query';
+import { submitQuery } from '@/app/actions/query';
 
 const notionLogger = new NotionLogger();