Spaces:
Running
Running
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():
|