@harness-engineering/orchestrator 0.8.1 → 0.8.2

package/dist/index.js

@@ -4746,7 +4746,7 @@ var OpenAIBackend = class {
 };
 
 // src/agent/backends/gemini.ts
-var import_generative_ai = require("@google/generative-ai");
+var import_genai = require("@google/genai");
 var import_types13 = require("@harness-engineering/types");
 var import_core6 = require("@harness-engineering/core");
 var GeminiBackend = class {
@@ -4784,16 +4784,21 @@ var GeminiBackend = class {
     let cacheCreationTokens = 0;
     let cacheReadTokens = 0;
     try {
-      const genAI = new import_generative_ai.GoogleGenerativeAI(this.config.apiKey);
-      const model = genAI.getGenerativeModel({
+      const genAI = new import_genai.GoogleGenAI({ apiKey: this.config.apiKey });
+      const response = await genAI.models.generateContentStream({
         model: this.config.model,
+        contents: params.prompt,
         ...geminiSession.systemPrompt !== void 0 && {
-          systemInstruction: geminiSession.systemPrompt
+          config: { systemInstruction: geminiSession.systemPrompt }
         }
       });
-      const result = await model.generateContentStream(params.prompt);
-      for await (const chunk of result.stream) {
-        const text = chunk.text();
+      for await (const chunk of response) {
+        let text;
+        try {
+          text = chunk.text;
+        } catch {
+          text = void 0;
+        }
         if (text) {
           const event = {
             type: "text",
@@ -4823,7 +4828,13 @@ var GeminiBackend = class {
       return {
         success: false,
         sessionId: session.sessionId,
-        usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+        usage: {
+          inputTokens,
+          outputTokens,
+          totalTokens,
+          cacheCreationTokens,
+          cacheReadTokens
+        },
         error: errorMessage
       };
     }
@@ -4850,9 +4861,14 @@ var GeminiBackend = class {
     return (0, import_types13.Ok)(void 0);
   }
   async healthCheck() {
+    if (!this.config.apiKey) {
+      return (0, import_types13.Err)({
+        category: "agent_not_found",
+        message: "GEMINI_API_KEY is not set"
+      });
+    }
     try {
-      const genAI = new import_generative_ai.GoogleGenerativeAI(this.config.apiKey);
-      genAI.getGenerativeModel({ model: this.config.model });
+      new import_genai.GoogleGenAI({ apiKey: this.config.apiKey });
       return (0, import_types13.Ok)(void 0);
     } catch (err) {
       return (0, import_types13.Err)({

package/dist/index.mjs

@@ -4653,7 +4653,7 @@ var OpenAIBackend = class {
 };
 
 // src/agent/backends/gemini.ts
-import { GoogleGenerativeAI } from "@google/generative-ai";
+import { GoogleGenAI } from "@google/genai";
 import {
   Ok as Ok13,
   Err as Err10
@@ -4694,16 +4694,21 @@ var GeminiBackend = class {
     let cacheCreationTokens = 0;
     let cacheReadTokens = 0;
     try {
-      const genAI = new GoogleGenerativeAI(this.config.apiKey);
-      const model = genAI.getGenerativeModel({
+      const genAI = new GoogleGenAI({ apiKey: this.config.apiKey });
+      const response = await genAI.models.generateContentStream({
         model: this.config.model,
+        contents: params.prompt,
         ...geminiSession.systemPrompt !== void 0 && {
-          systemInstruction: geminiSession.systemPrompt
+          config: { systemInstruction: geminiSession.systemPrompt }
         }
       });
-      const result = await model.generateContentStream(params.prompt);
-      for await (const chunk of result.stream) {
-        const text = chunk.text();
+      for await (const chunk of response) {
+        let text;
+        try {
+          text = chunk.text;
+        } catch {
+          text = void 0;
+        }
         if (text) {
           const event = {
             type: "text",
@@ -4733,7 +4738,13 @@ var GeminiBackend = class {
       return {
         success: false,
         sessionId: session.sessionId,
-        usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+        usage: {
+          inputTokens,
+          outputTokens,
+          totalTokens,
+          cacheCreationTokens,
+          cacheReadTokens
+        },
         error: errorMessage
       };
     }
@@ -4760,9 +4771,14 @@ var GeminiBackend = class {
     return Ok13(void 0);
   }
   async healthCheck() {
+    if (!this.config.apiKey) {
+      return Err10({
+        category: "agent_not_found",
+        message: "GEMINI_API_KEY is not set"
+      });
+    }
     try {
-      const genAI = new GoogleGenerativeAI(this.config.apiKey);
-      genAI.getGenerativeModel({ model: this.config.model });
+      new GoogleGenAI({ apiKey: this.config.apiKey });
       return Ok13(void 0);
     } catch (err) {
       return Err10({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@harness-engineering/orchestrator",
-  "version": "0.8.1",
+  "version": "0.8.2",
   "description": "Orchestrator daemon for dispatching coding agents to issues",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -38,7 +38,7 @@
   "dependencies": {
     "@anthropic-ai/sdk": "^0.95.1",
     "@earendil-works/pi-coding-agent": "^0.74.1",
-    "@google/generative-ai": "^0.24.0",
+    "@google/genai": "^2.0.4",
     "bcryptjs": "^2.4.3",
     "better-sqlite3": "^12.10.0",
     "ink": "^4.4.1",
@@ -48,9 +48,9 @@
     "ws": "^8.21.0",
     "yaml": "^2.8.3",
     "zod": "^3.25.76",
-    "@harness-engineering/core": "0.29.0",
+    "@harness-engineering/core": "0.30.0",
     "@harness-engineering/graph": "0.11.0",
-    "@harness-engineering/intelligence": "0.2.7",
+    "@harness-engineering/intelligence": "0.3.0",
     "@harness-engineering/types": "0.16.0"
   },
   "devDependencies": {