amaye15 commited on
Commit
c4f488d
·
1 Parent(s): 6f4f307

Debug - Circular import

Browse files
src/api/dependency.py DELETED
@@ -1,13 +0,0 @@
1
- import os
2
- from src.api.services.embedding_service import EmbeddingService
3
- from src.api.services.huggingface_service import HuggingFaceService
4
-
5
-
6
- # Dependency to get EmbeddingService
7
- def get_embedding_service() -> EmbeddingService:
8
- return EmbeddingService(openai_api_key=os.getenv("OPENAI_API_KEY"))
9
-
10
-
11
- # Dependency to get HuggingFaceService
12
- def get_huggingface_service() -> HuggingFaceService:
13
- return HuggingFaceService()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
src/api/services/huggingface_service.py CHANGED
@@ -3,7 +3,7 @@ from huggingface_hub import HfApi, HfFolder
3
  import logging
4
  from typing import Optional, Dict, List
5
  import pandas as pd
6
- from src.api.dependency import get_embedding_service, get_huggingface_service
7
  from src.api.exceptions import (
8
  DatasetNotFoundError,
9
  DatasetPushError,
 
3
  import logging
4
  from typing import Optional, Dict, List
5
  import pandas as pd
6
+ from src.main import get_embedding_service
7
  from src.api.exceptions import (
8
  DatasetNotFoundError,
9
  DatasetPushError,
src/main.py CHANGED
@@ -203,7 +203,8 @@ from src.api.database import get_db, Database, QueryExecutionError, HealthCheckE
203
  from src.api.services.embedding_service import EmbeddingService
204
  from src.api.services.huggingface_service import HuggingFaceService
205
  from src.api.exceptions import DatasetNotFoundError, DatasetPushError, OpenAIError
206
- from src.api.dependency import get_embedding_service, get_huggingface_service
 
207
  import pandas as pd
208
  import logging
209
  from dotenv import load_dotenv
@@ -233,6 +234,16 @@ app = FastAPI(
233
  app.add_middleware(GZipMiddleware, minimum_size=1000)
234
 
235
 
 
 
 
 
 
 
 
 
 
 
236
  # Root endpoint redirects to /docs
237
  @app.get("/")
238
  async def root():
 
203
  from src.api.services.embedding_service import EmbeddingService
204
  from src.api.services.huggingface_service import HuggingFaceService
205
  from src.api.exceptions import DatasetNotFoundError, DatasetPushError, OpenAIError
206
+
207
+ # from src.api.dependency import get_embedding_service, get_huggingface_service
208
  import pandas as pd
209
  import logging
210
  from dotenv import load_dotenv
 
234
  app.add_middleware(GZipMiddleware, minimum_size=1000)
235
 
236
 
237
+ # Dependency to get EmbeddingService
238
+ def get_embedding_service() -> EmbeddingService:
239
+ return EmbeddingService(openai_api_key=os.getenv("OPENAI_API_KEY"))
240
+
241
+
242
+ # Dependency to get HuggingFaceService
243
+ def get_huggingface_service() -> HuggingFaceService:
244
+ return HuggingFaceService()
245
+
246
+
247
  # Root endpoint redirects to /docs
248
  @app.get("/")
249
  async def root():