From fa80365933d445418e775fdd6baa01328c41a987 Mon Sep 17 00:00:00 2001 From: Alexander_Kabui Date: Wed, 3 Jan 2024 17:52:53 +0300 Subject: fetch auth token from current app --- gn3/llms/client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'gn3/llms/client.py') diff --git a/gn3/llms/client.py b/gn3/llms/client.py index f205247..6bcbf31 100644 --- a/gn3/llms/client.py +++ b/gn3/llms/client.py @@ -5,13 +5,13 @@ import os import datetime import time import requests +from flask import current_app from requests import Session from urllib.parse import urljoin from requests.packages.urllib3.util.retry import Retry from requests import HTTPError from requests import Session from requests.adapters import HTTPAdapter -from gn3.settings import FAHAMU_AUTH_TOKEN basedir = os.path.join(os.path.dirname(__file__)) @@ -59,7 +59,7 @@ class GeneNetworkQAClient(Session): def __init__(self, account, api_key, version="v3", timeout=5, total_retries=5, backoff_factor=30): super().__init__() - self.headers.update({"Authorization": "Bearer " + FAHAMU_AUTH_TOKEN}) + self.headers.update({"Authorization": "Bearer " + current_app.config.get("FAHAMU_AUTH_TOKEN","")}) self.answer_url = f"{self.BASE_URL}/answers" self.feedback_url = f"{self.BASE_URL}/feedback" -- cgit v1.2.3