1export const Endpoint = {
2  GOOGLE: 'https://www.google.com',
3  INIT: 'https://gemini.google.com/app',
4  GENERATE:
5    'https://gemini.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate',
6  ROTATE_COOKIES: 'https://accounts.google.com/RotateCookies',
7  UPLOAD: 'https://content-push.googleapis.com/upload',
8  BATCH_EXEC: 'https://gemini.google.com/_/BardChatUi/data/batchexecute',
9} as const;
10 
11export const GRPC = {
12  LIST_CHATS: 'MaZiqc',
13  READ_CHAT: 'hNvQHb',
14  LIST_GEMS: 'CNgdBe',
15  CREATE_GEM: 'oMH3Zd',
16  UPDATE_GEM: 'kHv0Vd',
17  DELETE_GEM: 'UXcSJb',
18} as const;
19 
20export const Headers = {
21  GEMINI: {
22    'Content-Type': 'application/x-www-form-urlencoded;charset=utf-8',
23    Host: 'gemini.google.com',
24    Origin: 'https://gemini.google.com',
25    Referer: 'https://gemini.google.com/',
26    'User-Agent':
27      'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
28    'X-Same-Domain': '1',
29  },
30  ROTATE_COOKIES: {
31    'Content-Type': 'application/json',
32  },
33  UPLOAD: {
34    'Push-ID': 'feeds/mcudyrk2a4khkz',
35  },
36} as const;
37 
38export const ErrorCode = {
39  TEMPORARY_ERROR_1013: 1013,
40  USAGE_LIMIT_EXCEEDED: 1037,
41  MODEL_INCONSISTENT: 1050,
42  MODEL_HEADER_INVALID: 1052,
43  IP_TEMPORARILY_BLOCKED: 1060,
44} as const;
45 
46export class Model {
47  static readonly UNSPECIFIED = new Model('unspecified', {}, false);
48  static readonly G_3_0_PRO = new Model(
49    'gemini-3.0-pro',
50    { 'x-goog-ext-525001261-jspb': '[1,null,null,null,"9d8ca3786ebdfbea",null,null,0,[4],null,null,1]' },
51    false,
52  );
53  static readonly G_3_0_FLASH = new Model(
54    'gemini-3.0-flash',
55    { 'x-goog-ext-525001261-jspb': '[1,null,null,null,"fbb127bbb056c959",null,null,0,[4],null,null,1]' },
56    false,
57  );
58  static readonly G_3_0_FLASH_THINKING = new Model(
59    'gemini-3.0-flash-thinking',
60    { 'x-goog-ext-525001261-jspb': '[1,null,null,null,"5bf011840784117a",null,null,0,[4],null,null,1]' },
61    false,
62  );
63  static readonly G_3_1_PRO_PREVIEW = new Model(
64    'gemini-3.1-pro-preview',
65    {},
66    false,
67  );
68 
69  constructor(
70    public readonly model_name: string,
71    public readonly model_header: Record<string, string>,
72    public readonly advanced_only: boolean,
73  ) {}
74 
75  static from_name(name: string): Model {
76    for (const model of [Model.UNSPECIFIED, Model.G_3_0_PRO, Model.G_3_0_FLASH, Model.G_3_0_FLASH_THINKING, Model.G_3_1_PRO_PREVIEW]) {
77      if (model.model_name === name) return model;
78    }
79 
80    throw new Error(
81      `Unknown model name: ${name}. Available models: ${[Model.UNSPECIFIED, Model.G_3_0_PRO, Model.G_3_0_FLASH, Model.G_3_0_FLASH_THINKING, Model.G_3_1_PRO_PREVIEW]
82        .map((m) => m.model_name)
83        .join(', ')}`,
84    );
85  }
86 
87  static from_dict(model_dict: { model_name?: unknown; model_header?: unknown }): Model {
88    if (!model_dict || typeof model_dict !== 'object') {
89      throw new Error("When passing a custom model as a dictionary, 'model_name' and 'model_header' keys must be provided.");
90    }
91 
92    if (!('model_name' in model_dict) || !('model_header' in model_dict)) {
93      throw new Error("When passing a custom model as a dictionary, 'model_name' and 'model_header' keys must be provided.");
94    }
95 
96    if (typeof model_dict.model_name !== 'string' || !model_dict.model_name.trim()) {
97      throw new Error("When passing a custom model as a dictionary, 'model_name' must be a non-empty string.");
98    }
99 
100    if (!model_dict.model_header || typeof model_dict.model_header !== 'object') {
101      throw new Error("When passing a custom model as a dictionary, 'model_header' must be a dictionary containing valid header strings.");
102    }
103 
104    const header: Record<string, string> = {};
105    for (const [k, v] of Object.entries(model_dict.model_header as Record<string, unknown>)) {
106      if (typeof v === 'string') header[k] = v;
107    }
108 
109    return new Model(model_dict.model_name, header, false);
110  }
111}
112 
113