From 0367dc2b4411a76cd06ab694ce31ccee8d418e75 Mon Sep 17 00:00:00 2001 From: Yuanwen Tian Date: Tue, 24 Sep 2024 22:09:18 +0800 Subject: [PATCH] fix: add func name to florencev2 fine tune --- vision_agent/tools/tools.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index fca3819c..4b1e166e 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -1181,7 +1181,8 @@ def florence2_phrase_grounding( fine_tuning=FineTuning(job_id=UUID(fine_tune_id)), ) data = data_obj.model_dump(by_alias=True) - detections = send_inference_request(data, "tools", v2=False) + metadata = {"function_name": "florence2_phrase_grounding"} + detections = send_inference_request(data, "tools", v2=False, metadata=metadata) else: data = { "image": image_b64,