fix: add func name to florencev2 fine tune

landing-ai · Sep 24, 2024 · 0367dc2 · 0367dc2
1 parent 03d0e5d
commit 0367dc2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,