@iinm/plain-agent 1.0.0

package/src/providers/gemini.d.ts

@@ -0,0 +1,208 @@
+/* Model */
+export type GeminiModelConfig = {
+  // https://ai.google.dev/gemini-api/docs/models
+  model: string;
+  requestConfig?: {
+    generationConfig: GeminiGenerationConfig;
+    safetySettings?: GeminiSafetySetting[];
+  };
+};
+
+// https://ai.google.dev/api/generate-content
+export type GeminiGenerationConfig = {
+  /**
+   * If less than 1, it tends to cause loops in Gemini 3.
+   * https://ai.google.dev/gemini-api/docs/troubleshooting?hl=ja#repetitive-tokens
+   */
+  temperature?: number;
+  maxOutputTokens?: number;
+  thinkingConfig?: {
+    includeThoughts: boolean;
+  } & (
+    | {
+        thinkingBudget?: number;
+      }
+    | {
+        thinkingLevel?: "low" | "medium" | "high";
+      }
+  );
+};
+
+export type GeminiSafetySetting = {
+  /**
+   * - HARM_CATEGORY_SEXUALLY_EXPLICIT
+   * - HARM_CATEGORY_HATE_SPEECH
+   * - HARM_CATEGORY_HARASSMENT
+   * - HARM_CATEGORY_DANGEROUS_CONTENT
+   */
+  category: string;
+
+  /**
+   * - BLOCK_NONE
+   * ...
+   */
+  threshold: string;
+};
+
+/* Input */
+export type GeminiGenerateContentInput = {
+  generationConfig: GeminiGenerationConfig;
+  safetySettings: GeminiSafetySetting[];
+  system_instruction?: {
+    parts: GeminiContentPartText[];
+  };
+  cachedContent?: string;
+  contents: (GeminiUserContent | GeminiModelContent | GeminiFunctionContent)[];
+  tools?: GeminiToolDefinition[];
+  toolConfig?: GeminiToolConfig;
+};
+
+/* Content */
+export type GeminiContent =
+  | GeminiSystemContent
+  | GeminiModelContent
+  | GeminiUserContent
+  | GeminiFunctionContent;
+
+export type GeminiSystemContent = {
+  role: "system";
+  parts: GeminiContentPartText[];
+};
+
+export type GeminiModelContent = {
+  role: "model";
+  parts?: (GeminiContentPartText | GeminiContentPartFunctionCall)[];
+};
+
+export type GeminiUserContent = {
+  role: "user";
+  parts: (GeminiContentPartText | GeminiContentPartInlineData)[];
+};
+
+export type GeminiFunctionContent = {
+  role: "user";
+  parts: GeminiContentPartFunctionResponse[];
+};
+
+/* Content Parts */
+export type GeminiContentPartText = {
+  text: string;
+  thoughtSignature?: string;
+  thought?: boolean;
+};
+
+export type GeminiContentPartFunctionCall = {
+  functionCall: {
+    name: string;
+    args: Record<string, unknown>;
+  };
+  thoughtSignature?: string;
+};
+
+export type GeminiContentPartFunctionResponse = {
+  functionResponse: {
+    name: string;
+    response: {
+      name: string;
+      content: (GeminiContentPartText | GeminiContentPartInlineData)[];
+    };
+  };
+};
+
+export type GeminiContentPartInlineData = {
+  inline_data: {
+    mime_type: string;
+    data: string;
+  };
+};
+
+/**
+ * Tool
+ * ref. https://cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1beta1/FunctionDeclaration
+ */
+export type GeminiToolDefinition = {
+  functionDeclarations: {
+    name: string;
+    description: string;
+    parametersJsonSchema: Record<string, unknown>;
+  }[];
+};
+
+export type GeminiToolConfig = {
+  functionCallingConfig?: {
+    mode?: "AUTO" | "ANY" | "NONE" | "VALIDATED";
+  };
+};
+
+/* Output */
+export type GeminiGeneratedContent = {
+  candidates?: GeminiGeneratedContentCandidate[];
+  usageMetadata: GeminiUsageMetadata;
+  modelVersion: string;
+};
+
+export type GeminiGeneratedContentCandidate = {
+  content: GeminiModelContent;
+  /**
+   * - STOP
+   * ...
+   */
+  finishReason?: "MALFORMED_FUNCTION_CALL" | string;
+  finishMessage?: string;
+  safetyRatings?: GeminiSafetyRating[];
+};
+
+export type GeminiSafetyRating = {
+  category: string;
+  probability: string;
+};
+
+export type GeminiTokensDetail = {
+  /**
+   * - TEXT
+   */
+  modality: string;
+  tokenCount: number;
+};
+
+export type GeminiUsageMetadata = {
+  totalTokenCount: number;
+  promptTokenCount: number;
+  cachedContentTokenCount?: number;
+  candidatesTokenCount?: number;
+  thoughtsTokenCount?: number;
+  promptTokensDetails: GeminiTokensDetail[];
+  cachedTokenDetails?: GeminiTokensDetail[];
+};
+
+/* Caching */
+export type GeminiCreateCachedContentInput = {
+  model: string;
+
+  /**
+   * e.g., 600s
+   */
+  ttl: string;
+
+  system_instruction?: {
+    parts: GeminiContentPartText[];
+  };
+  contents: (GeminiUserContent | GeminiModelContent | GeminiFunctionContent)[];
+  tools?: GeminiToolDefinition[];
+  toolConfig?: GeminiToolConfig;
+};
+
+export type GeminiCachedContents = {
+  contents: unknown[];
+  tools: unknown[];
+  createTime: string;
+  updateTime: string;
+  usageMetadata: GeminiUsageMetadata;
+  expireTime: string;
+  ttl: string;
+  name: string;
+  displayName: string;
+  model: string;
+  systemInstruction: unknown;
+  toolConfig: unknown;
+};