循环等待.(再次)

Question

这个问题对我来说就像是噩梦的重演。该项目是使用gpt3训练聊天机器人，我正在尝试嵌入。

我有文档，我正在尝试为不同的部分创建嵌入。根据我的测试，getEmbeddings((似乎返回了一个值。然而，当我的循环结束时，我的部分似乎没有嵌入(只有一个……嵌入需要花钱，所以我只运行一次循环，直到它工作为止(。

我做错了什么？

begin();
async function begin(){
let sections = [];
let inputFileName = "sourceDocument.jsonl";
let filecontents = fs.readFileSync(inputFileName,{encoding:'utf-8'})
let inputRows = [] = String(filecontents).split(/r?n/);
for(let i = 0; i < inputRows.length; i++) {
let row = inputRows[i];
if(row.trim().length == 0) continue;
let dataObject = JSON.parse(row);

let text = dataObject.completion;
let section = new Section();
let titleend =  text.indexOf("'>")

//console.log(dataObject);

section.title = text.substring(1,titleend);
section.text = String(text).substring( (titleend + 2), (text.length - 10) );
section.embedding = await getEmbedding(section.text);
sections.push(section);

break;
}
console.log(sections[0]);   
}
async function getEmbedding(textSample){
const embeddingModel = "text-search-davinci-doc-001";
if(DEBUGGING){
console.log(`DEBUG: getEmbedding(${textSample})`);
}
const OPENAI_REQEST_HEADERS = {
"Content-Type":"application/json",
"Authorization":`Bearer ${OPENAI_API_KEY}`
}
let data = {
"input"         :   textSample,
"model"         :   embeddingModel
};
let res;
await axios.post(EMBEDDINGS, data, {headers:OPENAI_REQEST_HEADERS})
.then((response) => {
if(DEBUGGING) {
console.log("   Embedding:  " + response.data.data[0].embedding);
}
let embedding = response.data.data[0].embedding;
return embedding;
})
.catch((error) => {
console.log("Error posting to OpenAI:");
console.log(error.response.data);
})
}

Answer 1

要么使用async/await，要么使用Promises的原生then模式，但尽量避免同时使用这两种模式-这会让人感到困惑，这就是你出错的地方。return语句从then方法返回，但不从外部getEmbedding方法本身返回。

有两种方法可以解决这个问题——一种是返回axios.post的结果，但另一种更容易阅读的方法在整个中使用async/await

async function getEmbedding(textSample){
const embeddingModel = "text-search-davinci-doc-001";
if(DEBUGGING){
console.log(`DEBUG: getEmbedding(${textSample})`);
}
const OPENAI_REQEST_HEADERS = {
"Content-Type":"application/json",
"Authorization":`Bearer ${OPENAI_API_KEY}`
}
let data = {
"input"         :   textSample,
"model"         :   embeddingModel
};
try{
const response = await axios.post(EMBEDDINGS, data, {headers:OPENAI_REQEST_HEADERS});
if(DEBUGGING) {
console.log("   Embedding:  " + response.data.data[0].embedding);
}
return response.data.data[0].embedding;
}
catch(error){
console.log("Error posting to OpenAI:");
console.log(error.response.data);
}
}