Browse Source

agent template upgrade (#3060)

### What problem does this PR solve?
#3056

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
tags/v0.13.0
Kevin Hu 1 year ago
parent
commit
e56bd770ea
No account linked to committer's email address

+ 2
- 8
agent/component/retrieval.py View File

@@ -43,25 +43,19 @@ class RetrievalParam(ComponentParamBase):
self.check_decimal_float(self.similarity_threshold, "[Retrieval] Similarity threshold")
self.check_decimal_float(self.keywords_similarity_weight, "[Retrieval] Keywords similarity weight")
self.check_positive_number(self.top_n, "[Retrieval] Top N")
self.check_empty(self.kb_ids, "[Retrieval] Knowledge bases")


class Retrieval(ComponentBase, ABC):
component_name = "Retrieval"

def _run(self, history, **kwargs):
# query = []
# for role, cnt in history[::-1][:self._param.message_history_window_size]:
# if role != "user":continue
# query.append(cnt)
# # query = "\n".join(query)
# query = query[0]
query = self.get_input()
query = str(query["content"][0]) if "content" in query else ""

kbs = KnowledgebaseService.get_by_ids(self._param.kb_ids)
if not kbs:
raise ValueError("Can't find knowledgebases by {}".format(self._param.kb_ids))
return Retrieval.be_output("")

embd_nms = list(set([kb.embd_id for kb in kbs]))
assert len(embd_nms) == 1, "Knowledge bases use different embedding models."


+ 916
- 672
agent/templates/DB Assistant.json
File diff suppressed because it is too large
View File


+ 1489
- 661
agent/templates/HR_callout_zh.json
File diff suppressed because it is too large
View File


+ 871
- 582
agent/templates/customer_service.json
File diff suppressed because it is too large
View File


+ 1808
- 1667
agent/templates/general_chat_bot.json
File diff suppressed because it is too large
View File


+ 238
- 136
agent/templates/interpreter.json
File diff suppressed because it is too large
View File


+ 553
- 388
agent/templates/investment_advisor.json
File diff suppressed because it is too large
View File


+ 662
- 480
agent/templates/medical_consultation.json
File diff suppressed because it is too large
View File


+ 572
- 432
agent/templates/text2sql.json
File diff suppressed because it is too large
View File


+ 800
- 488
agent/templates/websearch_assistant.json
File diff suppressed because it is too large
View File


Loading…
Cancel
Save