瀏覽代碼

Fix generate component reset_index and update text2sql template prompt (#2031)

### What problem does this PR solve?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
- [x] Documentation Update

---------

Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
tags/v0.10.0
H 1 年之前
父節點
當前提交
17ada637db
No account linked to committer's email address
共有 2 個文件被更改,包括 2 次插入2 次删除
  1. 1
    1
      agent/component/generate.py
  2. 1
    1
      agent/templates/text2sql.json

+ 1
- 1
agent/component/generate.py 查看文件

@@ -66,7 +66,7 @@ class Generate(ComponentBase):
return cpnts

def set_cite(self, retrieval_res, answer):
retrieval_res.dropna(subset=["vector", "content_ltks"], inplace=True)
retrieval_res = retrieval_res.dropna(subset=["vector", "content_ltks"]).reset_index(drop=True)
if "empty_response" in retrieval_res.columns:
retrieval_res["empty_response"].fillna("", inplace=True)
answer, idx = retrievaler.insert_citations(answer, [ck["content_ltks"] for _, ck in retrieval_res.iterrows()],

+ 1
- 1
agent/templates/text2sql.json 查看文件

@@ -51,7 +51,7 @@
}
],
"presence_penalty": 0.4,
"prompt": "##The user provides a question and you provide SQL. You will only respond with SQL code and not with any explanations.\n\n##Respond with only SQL code. Do not answer with any explanations -- just the code.\n\n##You may use the following DDL statements as a reference for what tables might be available. Use responses to past questions also to guide you: {ddl_input}.\n\n##You may use the following documentation as a reference for what tables might be available. Use responses to past questions also to guide you: {db_input}.\n\n##You may use the following SQL statements as a reference for what tables might be available. Use responses to past questions also to guide you: {sql_input}.",
"prompt": "Role: You are an intelligent DB assistant that can help humans generate SQL statements for problems.\n\n##The user provides a question and you provide SQL. You will only respond with SQL code and not with any explanations.\n\n##You may use the following DDL statements as a reference for what tables might be available. Use responses to past questions also to guide you: {ddl_input}.\n\n##You may use the following documentation as a reference for what tables might be available. Use responses to past questions also to guide you: {db_input}.\n\n##You may use the following SQL statements as a reference for what tables might be available. Use responses to past questions also to guide you: {sql_input}.\n\n##Respond with only SQL code. Do not answer with any explanations -- just the code.",
"temperature": 0.1,
"top_p": 0.3
}

Loading…
取消
儲存