Adding safety adapter for Together

2025-06-28 02:53:30 +00:00 · 2024-09-20 09:35:01 -07:00 · 2024-09-20 09:35:01 -07:00 · b85d675c6f
commit b85d675c6f
parent 0d2eb3bd25
8 changed files with 188 additions and 23 deletions
--- a/llama_stack/apis/safety/client.py
+++ b/llama_stack/apis/safety/client.py
@ -49,7 +49,14 @@ class SafetyClient(Safety):
                    shield_type=shield_type,
                    messages=[encodable_dict(m) for m in messages],
                ),
-                headers={"Content-Type": "application/json"},
+                headers={
+                    "Content-Type": "application/json",
+                    "X-LlamaStack-ProviderData": json.dumps(
+                        {
+                            "together_api_key": "1882f9a484fc7c6ce3e4dc90272d5db52346c93838daab3d704803181f396b22"
+                        }
+                    ),
+                },
                timeout=20,
            )