Outlook connector - add "X-Connector-Id" (#10)

Co-authored-by: EugeneP <epavletsov@itera.ws>
cohere-ai · Dec 5, 2023 · 4f70a06 · 4f70a06
1 parent 5aa06b0
commit 4f70a06
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/outlook/provider/__init__.py b/outlook/provider/__init__.py
@@ -26,4 +26,6 @@ def create_app() -> connexion.FlaskApp:
     flask_app = app.app
     config_prefix = os.path.split(os.getcwd())[1].upper()
     flask_app.config.from_prefixed_env(config_prefix)
+    flask_app.config["APP_ID"] = config_prefix
+
     return flask_app
diff --git a/outlook/provider/app.py b/outlook/provider/app.py
@@ -14,7 +14,7 @@ def search(body):
     except UpstreamProviderError as error:
         logger.error(f"Upstream provider error: {error.message}")
         abort(502, error.message)
-    return {"results": data}
+    return {"results": data}, 200, {"X-Connector-Id": app.config.get("APP_ID")}
 
 
 def apikey_auth(token):