Spaces:
Sleeping
Sleeping
amaye15
commited on
Commit
·
c4f488d
1
Parent(s):
6f4f307
Debug - Circular import
Browse files- src/api/dependency.py +0 -13
- src/api/services/huggingface_service.py +1 -1
- src/main.py +12 -1
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.
|
| 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 |
-
|
|
|
|
| 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():
|