### 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
| @@ -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()], | |||
| @@ -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 | |||
| } | |||