fix: bug (#743)

vesoft-inc · Jan 17, 2024 · 7a2bf18 · 7a2bf18
1 parent afec886
commit 7a2bf18
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/app/pages/Import/AIImport/Create.tsx b/app/pages/Import/AIImport/Create.tsx
@@ -53,12 +53,12 @@ const Create = observer((props: { visible: boolean; onCancel: () => void }) => {
  const type = types[subfix] || 1;
  const size = file.size * type;
  const schema = await llm.getSpaceSchema(space);
- const schemaBytesLength = getByteLength(schema) + 500; //prompt length about 500
+ const schemaBytesLength = getByteLength(schema) + 1000; //prompt length about 500
  const splitNums = Math.ceil(size / llm.config.maxContextLength);
  const splitBytesLength = Math.ceil(schemaBytesLength * splitNums);
  const totalBytesLength = splitBytesLength + file.size;
  // full connection
- const tokensNum = totalBytesLength / 4; //about 4 bytes per token
+ const tokensNum = totalBytesLength / 2; //about 4 bytes per token
  setTokens(tokensNum);
  }
  })();

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -23,7 +23,7 @@
  "@vesoft-inc/force-graph": "2.0.7",
  "@vesoft-inc/i18n": "^1.0.1",
  "@vesoft-inc/icons": "^1.7.0",
- "@vesoft-inc/nebula-explain-graph": "^1.0.10",
+ "@vesoft-inc/nebula-explain-graph": "^1.0.11",
  "@vesoft-inc/veditor": "^4.4.14",
  "antd": "^5.8.4",
  "axios": "^0.23.0",