Merge pull request #139 from weaviate/fix/headers-missing-in-grpc

Add support for custom headers in WeaviateGrpcClient

Author: dirkkul

src/Weaviate.Client/WeaviateClient.cs

@@ -229,7 +229,12 @@ public WeaviateClient(
         }
 
         RestClient = new WeaviateRestClient(Configuration.RestUri, httpClient);
-        GrpcClient = new WeaviateGrpcClient(Configuration.GrpcUri, wcdHost, _tokenService);
+        GrpcClient = new WeaviateGrpcClient(
+            Configuration.GrpcUri,
+            wcdHost,
+            _tokenService,
+            Configuration.Headers
+        );
 
         Nodes = new NodesClient(RestClient);
         Collections = new CollectionsClient(this);

src/Weaviate.Client/gRPC/Client.cs

@@ -38,6 +38,7 @@ public WeaviateGrpcClient(
         Uri grpcUri,
         string? wcdHost,
         ITokenService? tokenService,
+        Dictionary<string, string>? headers = null,
         ILogger<WeaviateGrpcClient>? logger = null
     )
     {
@@ -112,6 +113,19 @@ public WeaviateGrpcClient(
         {
             _defaultHeaders = new Metadata { { "X-Weaviate-Cluster-URL", wcdHost } };
         }
+        if (headers != null)
+        {
+            if (_defaultHeaders == null)
+            {
+                _defaultHeaders = new Metadata();
+            }
+
+            foreach (var header in headers)
+            {
+                _defaultHeaders.Add(header.Key, header.Value);
+            }
+        }
+
         _grpcClient = new V1.Weaviate.WeaviateClient(_channel);
     }