Where are the codes and data for tool matching?

TsinghuaDatabaseGroup / DB-GPT

An LLM Based Diagnosis System (https://arxiv.org/pdf/2312.01454.pdf)

http://dbgpt.dbmind.cn/

Apache License 2.0

536 stars 77 forks source link

Where are the codes and data for tool matching? #61

Open hongjin-su opened 9 months ago

hongjin-su commented 9 months ago

As titled, thanks!

zhouxh19 commented 9 months ago

We have added the tool matching function, i.e., api embedding during tool registration (./multiagents/tools/api_retrieval.py) and api matching at the start of tree search (./multiagents/tools/retriever.py).

We will (1) release the local embedding model (sentence_transformer) after adding sufficient optimization tools and (2) further test for the most suitable time to apply the "tool matching" function (e.g., at the beginning of diagnosis or for every llm inference).

zhouxh19 commented 9 months ago

Pls. the training data for localized text embedding model will be released soon