feat: pass headers to endpoint.

Author: Khurdhula-Harshavardhan

jigsawstack/__init__.py

@@ -68,32 +68,48 @@ def __init__(
 
         self.web = Web(api_key=api_key, base_url=base_url + "/v1", headers=headers)
 
-        self.sentiment = Sentiment(api_key=api_key, base_url=base_url + "/v1", headers=headers).analyze
+        self.sentiment = Sentiment(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).analyze
 
         self.validate = Validate(api_key=api_key, base_url=base_url + "/v1", headers=headers)
-        self.summary = Summary(api_key=api_key, base_url=base_url + "/v1", headers=headers).summarize
+        self.summary = Summary(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).summarize
 
         self.vision = Vision(api_key=api_key, base_url=base_url + "/v1", headers=headers)
 
-        self.prediction = Prediction(api_key=api_key, base_url=base_url + "/v1", headers=headers).predict
+        self.prediction = Prediction(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).predict
 
-        self.text_to_sql = SQL(api_key=api_key, base_url=base_url + "/v1", headers=headers).text_to_sql
+        self.text_to_sql = SQL(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).text_to_sql
 
         self.store = Store(api_key=api_key, base_url=base_url + "/v1", headers=headers)
 
         self.translate = Translate(api_key=api_key, base_url=base_url + "/v1", headers=headers)
 
-        self.embedding = Embedding(api_key=api_key, base_url=base_url + "/v1", headers=headers).execute
+        self.embedding = Embedding(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).execute
 
-        self.embedding_v2 = EmbeddingV2(api_key=api_key, base_url=base_url + "/v2", headers=headers).execute
+        self.embedding_v2 = EmbeddingV2(
+            api_key=api_key, base_url=base_url + "/v2", headers=headers
+        ).execute
 
         self.image_generation = ImageGeneration(
             api_key=api_key, base_url=base_url + "/v1", headers=headers
         ).image_generation
 
-        self.classification = Classification(api_key=api_key, base_url=base_url + "/v1", headers=headers).classify
+        self.classification = Classification(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).classify
 
-        self.prompt_engine = PromptEngine(api_key=api_key, base_url=base_url + "/v1", headers=headers)
+        self.prompt_engine = PromptEngine(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        )
 
 
 class AsyncJigsawStack:
@@ -149,19 +165,29 @@ def __init__(
 
         self.store = AsyncStore(api_key=api_key, base_url=base_url + "/v1", headers=headers)
 
-        self.summary = AsyncSummary(api_key=api_key, base_url=base_url + "/v1", headers=headers).summarize
+        self.summary = AsyncSummary(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).summarize
 
         self.prediction = AsyncPrediction(api_key=api_key, base_url=base_url + "/v1").predict
 
-        self.text_to_sql = AsyncSQL(api_key=api_key, base_url=base_url + "/v1", headers=headers).text_to_sql
+        self.text_to_sql = AsyncSQL(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).text_to_sql
 
-        self.sentiment = AsyncSentiment(api_key=api_key, base_url=base_url + "/v1", headers=headers).analyze
+        self.sentiment = AsyncSentiment(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).analyze
 
         self.translate = AsyncTranslate(api_key=api_key, base_url=base_url + "/v1", headers=headers)
 
-        self.embedding = AsyncEmbedding(api_key=api_key, base_url=base_url + "/v1", headers=headers).execute
+        self.embedding = AsyncEmbedding(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        ).execute
 
-        self.embedding_v2 = AsyncEmbeddingV2(api_key=api_key, base_url=base_url + "/v2", headers=headers).execute
+        self.embedding_v2 = AsyncEmbeddingV2(
+            api_key=api_key, base_url=base_url + "/v2", headers=headers
+        ).execute
 
         self.image_generation = AsyncImageGeneration(
             api_key=api_key, base_url=base_url + "/v1", headers=headers
@@ -171,7 +197,9 @@ def __init__(
             api_key=api_key, base_url=base_url + "/v1", headers=headers
         ).classify
 
-        self.prompt_engine = AsyncPromptEngine(api_key=api_key, base_url=base_url + "/v1", headers=headers)
+        self.prompt_engine = AsyncPromptEngine(
+            api_key=api_key, base_url=base_url + "/v1", headers=headers
+        )
 
 
 # Create a global instance of the Web class