Merge pull request #55 from mainframecomputer/sayak/mfe-442-add-model-providers-to-bt-logging

feat: add model providers to Braintrust logging

Author: sayakmaity

packages/python/src/mainframe_orchestra/llm.py

@@ -627,9 +627,9 @@ async def send_openrouter_request(
         spinner.start()
 
         try:
-            client = AsyncOpenAI(
+            client = wrap_openai(AsyncOpenAI(
                 base_url="https://openrouter.ai/api/v1", api_key=config.OPENROUTER_API_KEY
-            )
+            ))
             if not client.api_key:
                 raise ValueError("OpenRouter API key not found in environment variables.")
 
@@ -952,7 +952,10 @@ async def send_groq_request(
 
         try:
             api_key = config.validate_api_key("GROQ_API_KEY")
-            client = Groq(api_key=api_key)
+            client = wrap_openai(OpenAI(
+                base_url="https://api.groq.com/openai/v1",
+                api_key=api_key
+            ))
             if not client.api_key:
                 raise ValueError("Groq API key not found in environment variables.")
 
@@ -1072,7 +1075,7 @@ async def send_together_request(
 
         try:
             api_key = config.validate_api_key("TOGETHERAI_API_KEY")
-            client = OpenAI(api_key=api_key, base_url="https://api.together.xyz/v1")
+            client = wrap_openai(OpenAI(api_key=api_key, base_url="https://api.together.xyz/v1"))
 
             # Process images if present
             if image_data:
@@ -1426,10 +1429,10 @@ async def send_deepseek_request(
                 raise ValueError("DeepSeek API key not found in environment variables.")
 
             # Create an AsyncOpenAI client
-            client = AsyncOpenAI(
+            client = wrap_openai(AsyncOpenAI(
                 api_key=api_key,
                 base_url="https://api.deepseek.com/v1",
-            )
+            ))
 
             # Warn if image data was provided
             if image_data: