fix: NexusRESTClient circular import

app/celery.py

@@ -5,7 +5,7 @@
 from app.indexer.indexer_file_manager import IndexerFileManager
 from app.downloaders.s3 import S3FileDownloader
 
-from app.handlers.content_bases import NexusRESTClient
+from app.handlers.nexus import NexusRESTClient
 
 celery = Celery(__name__)
 celery.conf.broker_url = os.environ.get(

app/handlers/content_bases.py

@@ -9,8 +9,6 @@
 from typing import Annotated
 from app.handlers.authorizations import token_verification
 
-import requests, os
-
 
 class ContentBaseIndexRequest(BaseModel):
  file: str
@@ -76,23 +74,3 @@ def search(self, request: ContentBaseSearchRequest, Authorization: Annotated[str
  filter=request.filter
  )
  return ContentBaseSearchResponse(response=response)
-
-
-class NexusRESTClient:
- token = os.environ.get("NEXUS_AI_TOKEN")
- base_url = os.environ.get("NEXUS_AI_URL")
-
- def __init__(self) -> None:
- self.headers = {
- 'Authorization': self.token,
- 'Content-Type': "application/json"
- }
-
- def index_succedded(self, task_succeded: bool, nexus_task_uuid: str) -> None:
- endpoint = f'{self.base_url}/api/v1/content-base-file'
- data = { 
- "status": int(task_succeded),
- "task_uuid": nexus_task_uuid,
- }
- response = requests.patch(url=endpoint, data=data, headers=self.headers)
- response.raise_for_status()

app/handlers/nexus.py

@@ -0,0 +1,21 @@
+import requests, os
+
+
+class NexusRESTClient:
+ token = os.environ.get("NEXUS_AI_TOKEN")
+ base_url = os.environ.get("NEXUS_AI_URL")
+
+ def __init__(self) -> None:
+ self.headers = {
+ 'Authorization': self.token,
+ 'Content-Type': "application/json"
+ }
+
+ def index_succedded(self, task_succeded: bool, nexus_task_uuid: str) -> None:
+ endpoint = f'{self.base_url}/api/v1/content-base-file'
+ data = { 
+ "status": int(task_succeded),
+ "task_uuid": nexus_task_uuid,
+ }
+ response = requests.patch(url=endpoint, data=data, headers=self.headers)
+ response.raise_for_status()