mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-29 12:09:31 +08:00
Refact: switch from google-generativeai to google-genai (#13140)
### What problem does this PR solve? Refact: switch from oogle-generativeai to google-genai #13132 Refact: commnet out unused pywencai. ### Type of change - [x] Refactoring
This commit is contained in:
@@ -18,7 +18,7 @@ import os
|
||||
import time
|
||||
from abc import ABC
|
||||
import pandas as pd
|
||||
import pywencai
|
||||
# import pywencai
|
||||
|
||||
from agent.tools.base import ToolParamBase, ToolMeta, ToolBase
|
||||
from common.connection_utils import timeout
|
||||
@@ -84,7 +84,8 @@ class WenCai(ToolBase, ABC):
|
||||
|
||||
try:
|
||||
wencai_res = []
|
||||
res = pywencai.get(query=kwargs["query"], query_type=self._param.query_type, perpage=self._param.top_n)
|
||||
# res = pywencai.get(query=kwargs["query"], query_type=self._param.query_type, perpage=self._param.top_n)
|
||||
res = []
|
||||
if self.check_if_canceled("WenCai processing"):
|
||||
return
|
||||
|
||||
|
||||
Reference in New Issue
Block a user