@ibgib/web-gib 0.0.5 → 0.0.7

package/src/witness/agent/gemini/gemini-types.mts

@@ -1,8 +1,37 @@
-import { Content, } from "@google/generative-ai";
-import type { StartChatParams, ChatSession, FunctionCallingMode, } from "@google/generative-ai";
+import { Content, FunctionCallingConfigMode, GoogleGenAI, } from "@google/genai";
 import { PromptInfo, } from "../agent-types.mjs";
 import { TextSource } from "../agent-constants.mjs";
 
+// #region @google/generative-ai to @google/genai shims
+// the changes in @google/genai are a little bassackwards, in that they no
+// longer expose the ability to constrain parts at compile time. these shims
+// enable this.
+
+import { type Part } from '@google/genai';
+
+/** Utility type to make a specific key required, without using 'never' for others. */
+type AssertRequired<T, K extends keyof T> = T & Required<Pick<T, K>>;
+
+// Define your shims:
+
+/** Represents a Part where the 'text' property is guaranteed to be a string. */
+export type TextPart = AssertRequired<Part, 'text'>;
+
+/** Represents a Part where the 'functionCall' property is guaranteed to be present. */
+export type FunctionCallPart = AssertRequired<Part, 'functionCall'>;
+
+/** Represents a Part where the 'functionResponse' property is guaranteed to be present. */
+export type FunctionResponsePart = AssertRequired<Part, 'functionResponse'>;
+
+// // If you need the other top-level types:
+// export type GenerateContentResult = Parameters<NonNullable<ConstructorParameters<typeof GoogleGenAI>[0]>['generateContent']>['0'];
+
+// The EnhancedGenerateContentResponse maps directly to the main response type now
+import { type GenerateContentResponse } from '@google/genai';
+export type EnhancedGenerateContentResponse = GenerateContentResponse;
+
+// #endregion @google/generative-ai to @google/genai shims
+
 export interface PromptInfoEntryGemini {
     /**
      * The source of the text, in terms of our code definition.
@@ -36,23 +65,19 @@ export interface PromptInfoGemini extends PromptInfo {
      * recent.
      *
      * NOTE: We will be using this for the gemini api's "chat" function.
-     *
-     * @see {@link StartChatParams.history}
      */
     chatPromptHistoryParts: PromptInfoEntryGemini[];
     /**
     * The most recent chat message
-    *
-    * @see {@link ChatSession.sendMessage}
     */
     chatPromptCurrent: PromptInfoEntryGemini;
     /**
      * if truthy, this will override the default function calling mode when
      * prompting the LLM.
      *
-     * if falsy, will use the default (FunctionCallingMode.ANY)
+     * if falsy, will use the default (FunctionCallingConfigMode.ANY)
      *
-     * @see {@link FunctionCallingMode}
+     * @see {@link FunctionCallingConfigMode}
      *
      * ## notes
      *
@@ -61,7 +86,7 @@ export interface PromptInfoGemini extends PromptInfo {
      * to just ask the model a "normal" chat llm question (for translating some
      * text).
      */
-    overrideFunctionCallingMode?: FunctionCallingMode;
+    overrideFunctionCallingMode?: FunctionCallingConfigMode;
 }
 
 /**
@@ -79,10 +104,11 @@ export type PromptAPIResult_Gemini =
 
 /**
  * This is a "kluge" IMO because it wraps the response part to be able to handle
- * all responses of functions. This is a limitation of the Gemini API which requires an
- * object as the response part's value. So if a function returns "undefined" or an array,
- * it throws an error. So this is a wrapper.
+ * all responses of functions. This is a limitation of the Gemini API which
+ * requires an object as the response part's value. So if a function returns
+ * "undefined" or an array, it throws an error. So this is a wrapper.
  */
 export interface FunctionResponsePart_ResponseWrapper {
     value: any | undefined;
+    [key: string]: any; // This satisfies the compiler for the *external* library
 }