@3-/aiapi 0.1.39 → 0.1.41
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/gemini.js +6 -6
- package/package.json +1 -1
package/gemini.js
CHANGED
|
@@ -3,7 +3,7 @@ import sleep from '@3-/sleep';
|
|
|
3
3
|
|
|
4
4
|
import merge from 'lodash-es/merge.js';
|
|
5
5
|
|
|
6
|
-
export default (token_li) => {
|
|
6
|
+
export default (token_li, model = 'gemini-2.5-pro') => {
|
|
7
7
|
var _NEXT_TOKEN, _nextToken, nextToken;
|
|
8
8
|
token_li.sort(() => {
|
|
9
9
|
return Math.random() - 0.5;
|
|
@@ -21,7 +21,7 @@ export default (token_li) => {
|
|
|
21
21
|
return _NEXT_TOKEN.next().value;
|
|
22
22
|
};
|
|
23
23
|
return async(text, schema, system, option) => {
|
|
24
|
-
var body, err, error, message, r, retryed, status;
|
|
24
|
+
var body, err, error, message, parts, r, retryed, status;
|
|
25
25
|
body = {
|
|
26
26
|
contents: [
|
|
27
27
|
{
|
|
@@ -52,7 +52,7 @@ export default (token_li) => {
|
|
|
52
52
|
retryed = token_li.length;
|
|
53
53
|
while (true) {
|
|
54
54
|
try {
|
|
55
|
-
r = (await fetch('https://generativelanguage.googleapis.com/v1beta/models/
|
|
55
|
+
r = (await fetch('https://generativelanguage.googleapis.com/v1beta/models/' + model + ':generateContent', {
|
|
56
56
|
headers: {
|
|
57
57
|
'X-goog-api-key': nextToken(),
|
|
58
58
|
'Content-Type': 'application/json'
|
|
@@ -95,12 +95,12 @@ export default (token_li) => {
|
|
|
95
95
|
r = ((await r.json())).candidates[0];
|
|
96
96
|
if (schema) {
|
|
97
97
|
try {
|
|
98
|
-
|
|
99
|
-
if (!
|
|
98
|
+
({parts} = r.content);
|
|
99
|
+
if (!parts) {
|
|
100
100
|
console.warn('miss parts', r.content);
|
|
101
101
|
continue;
|
|
102
102
|
}
|
|
103
|
-
r = JSON.parse(
|
|
103
|
+
r = JSON.parse(parts[0].text);
|
|
104
104
|
} catch (error1) {
|
|
105
105
|
err = error1;
|
|
106
106
|
console.error(body, r, err);
|