Add support for hosted file search and vector store IDs

HostedFileSearchTool => Collection Search
HostedVectorStoreContent => CollectionIds

Author: kzu

src/Extensions.Grok/GrokChatClient.cs

@@ -252,6 +252,18 @@ GetCompletionsRequest MapToRequest(IEnumerable<ChatMessage> messages, ChatOption
                 {
                     request.Tools.Add(new Tool { CodeExecution = new CodeExecution { } });
                 }
+                else if (tool is HostedFileSearchTool fileSearch)
+                {
+                    var toolProto = new CollectionsSearch();
+
+                    if (fileSearch.Inputs?.OfType<HostedVectorStoreContent>() is { } vectorStores)
+                        toolProto.CollectionIds.AddRange(vectorStores.Select(x => x.VectorStoreId).Distinct());
+                    
+                    if (fileSearch.MaximumResultCount is { } maxResults)
+                        toolProto.Limit = maxResults;
+                    
+                    request.Tools.Add(new Tool { CollectionsSearch = toolProto });
+                }
             }
         }
 

src/Tests/GrokTests.cs

@@ -266,4 +266,30 @@ public async Task GrokInvokesHostedCodeExecution()
             response.Messages.SelectMany(x => x.Contents).OfType<HostedToolCallContent>(),
             x => x.ToolCall.Type == Devlooped.Grok.ToolCallType.CodeExecutionTool);
     }
+
+    [SecretsFact("XAI_API_KEY")]
+    public async Task GrokInvokesHostedCollectionSearch()
+    {
+        var messages = new Chat()
+        {
+            { "user", "¿Cuál es el monto exacto del rango de la multa por inasistencia injustificada a la audiencia señalada por el juez en el proceso sucesorio, según lo establecido en el Artículo 691 del Código Procesal Civil y Comercial de la Nación (Ley 17.454)?" },
+        };
+
+        var grok = new GrokClient(Configuration["XAI_API_KEY"]!).AsIChatClient("grok-4-fast");
+
+        var options = new ChatOptions
+        {
+            Tools = [new HostedFileSearchTool {
+                Inputs = [new HostedVectorStoreContent("collection_91559d9b-a55d-42fe-b2ad-ecf8904d9049")]
+            }]
+        };
+
+        var response = await grok.GetResponseAsync(messages, options);
+        var text = response.Text;
+
+        Assert.Contains("11,74", text);
+        Assert.Contains(
+            response.Messages.SelectMany(x => x.Contents).OfType<HostedToolCallContent>(),
+            x => x.ToolCall.Type == Devlooped.Grok.ToolCallType.CollectionsSearchTool);
+    }
 }