Fix: add download stats to kb logs. (#11112)

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-11-10 13:28:07 +08:00
committed by GitHub
parent bf382e5c4d
commit d207291217
12 changed files with 50 additions and 15 deletions

View File

@ -13,10 +13,10 @@
# See the License for the specific language governing permissions and
# limitations under the License.
#
from agent.component.base import ComponentBase, ComponentParamBase
from agent.component.message import MessageParam, Message
class UserFillUpParam(ComponentParamBase):
class UserFillUpParam(MessageParam):
def __init__(self):
super().__init__()
@ -27,10 +27,13 @@ class UserFillUpParam(ComponentParamBase):
return True
class UserFillUp(ComponentBase):
class UserFillUp(Message):
component_name = "UserFillUp"
def _invoke(self, **kwargs):
if self._param.enable_tips:
tips, kwargs = self.get_kwargs(self._param.tips)
self.set_output("tips", tips)
for k, v in kwargs.get("inputs", {}).items():
self.set_output(k, v)