Merge branch 'main' into ci/starklings-evaluation

Author: ijusttookadnatest

packages/backend/src/config/provider/gemini.ts

@@ -17,7 +17,7 @@ export const loadGeminiChatModels = async () => {
       'Gemini Flash 2.5': new ChatGoogleGenerativeAI({
         temperature: 0.7,
         apiKey: geminiApiKey,
-        modelName: 'gemini-2.5-flash-preview-04-17',
+        modelName: 'gemini-2.5-flash',
       }),
     };
 

packages/backend/src/routes/chatCompletionHandler.ts

@@ -132,6 +132,11 @@ export async function handleChatCompletion(
       let responseContent = '';
 
       agent.on('data', (data: any) => {
+        // Check if response is already finished
+        if (res.destroyed || res.writableEnded) {
+          return;
+        }
+
         const parsed = JSON.parse(data);
 
         if (parsed.type === 'response') {
@@ -162,14 +167,24 @@ export async function handleChatCompletion(
 
       agent.on('error', (error: any) => {
         console.error('Agent error:', error);
-        res.write(`data: ${JSON.stringify({ error: error.message })}\n\n`);
-        res.end();
+        if (!res.destroyed && !res.writableEnded) {
+          res.write(`data: ${JSON.stringify({ error: error.message })}\n\n`);
+          res.end();
+        }
       });
 
       agent.on('end', () => {
+        // Check if response is already finished
+        if (res.destroyed || res.writableEnded) {
+          return;
+        }
+
         const tokenUsage = TokenTracker.getSessionTokenUsage();
 
-        res.setHeader('x-total-tokens', tokenUsage.totalTokens.toString());
+        // Check if headers haven't been sent yet
+        if (!res.headersSent) {
+          res.setHeader('x-total-tokens', tokenUsage.totalTokens.toString());
+        }
 
         const finalChunk = {
           id: uuidv4(),
@@ -210,16 +225,25 @@ export async function handleChatCompletion(
 
       agent.on('error', (error: any) => {
         console.error('Agent error:', error);
-        res.status(500).json({
-          error: {
-            message: error.message,
-            type: 'server_error',
-            code: 'internal_error',
-          },
-        });
+
+        // Check if headers haven't been sent yet
+        if (!res.headersSent) {
+          res.status(500).json({
+            error: {
+              message: error.message,
+              type: 'server_error',
+              code: 'internal_error',
+            },
+          });
+        }
       });
 
       agent.on('end', () => {
+        // Check if headers haven't been sent yet
+        if (res.headersSent) {
+          return;
+        }
+
         const tokenUsage = TokenTracker.getSessionTokenUsage();
 
         res.setHeader('x-total-tokens', tokenUsage.totalTokens.toString());
@@ -258,6 +282,11 @@ export async function handleChatCompletion(
   } catch (error) {
     console.error('Error in chat completion:', error);
 
+    // Check if headers haven't been sent yet
+    if (res.headersSent) {
+      return;
+    }
+
     // Map common errors to OpenAI error format
     if (error instanceof Error) {
       const errorResponse: any = {

packages/ingester/asciidoc/oz-ui-bundle.zip

@@ -2,10 +2,7 @@ import * as fs from 'fs/promises';
 import * as path from 'path';
 import downdoc from 'downdoc';
 import { Document } from '@langchain/core/documents';
-import {
-  BookChunk,
-  DocumentSource,
-} from '@cairo-coder/agents/types/index';
+import { BookChunk, DocumentSource } from '@cairo-coder/agents/types/index';
 import { BaseIngester } from '../BaseIngester';
 import {
   BookConfig,
@@ -68,7 +65,7 @@ export abstract class AsciiDocIngester extends BaseIngester {
 
     // Run Antora to generate the documentation
     logger.info('Running Antora to build documentation');
-    const antoraCommand = `antora ${this.playbookPath}`;
+    const antoraCommand = `antora ${this.playbookPath} --fetch`;
 
     // Find the package root by looking for package.json
     let packageRoot = __dirname;

packages/ingester/src/ingesters/AsciiDocIngester.ts

@@ -8,6 +8,8 @@ import { logger } from '@cairo-coder/agents/utils/index';
 import { AsciiDocIngesterConfig } from './AsciiDocIngester';
 import { AsciiDocIngester } from './AsciiDocIngester';
 
+const OZ_VERSION = 'v2.0.0';
+
 /**
  * Ingester for the OpenZeppelin documentation
  *
@@ -90,7 +92,7 @@ export class OpenZeppelinDocsIngester extends AsciiDocIngester {
             // This is the special handling for OpenZeppelin docs
             const adaptedFullPageName = path.join(
               'contracts-cairo',
-              '1.0.0',
+              OZ_VERSION,
               relativePath,
             );