feat(model): Support llama3.1 models (#1744)

2025-09-12 12:37:14 +00:00 · 2024-07-24 11:00:05 +08:00
parent 4149252321
commit 3c5ed9d8c1
10 changed files with 125 additions and 2 deletions
--- a/dbgpt/agent/expand/resources/search_tool.py
+++ b/dbgpt/agent/expand/resources/search_tool.py
@@ -38,6 +38,7 @@ def baidu_search(
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:112.0) "
        "Gecko/20100101 Firefox/112.0"
    }
+    num_results = int(num_results)
    if num_results < 8:
        num_results = 8
    url = f"https://www.baidu.com/s?wd={query}&rn={num_results}"