Spaces:
Running
Running
amaye15
commited on
Commit
·
5c144a1
1
Parent(s):
0611c31
Debug - create embeddings endpoint
Browse files- src/api/models/embedding_models.py +1 -1
- src/main.py +3 -0
src/api/models/embedding_models.py
CHANGED
@@ -46,5 +46,5 @@ class DeleteEmbeddingRequest(BaseModel):
|
|
46 |
class EmbedRequest(BaseModel):
|
47 |
texts: List[str] # List of strings to generate embeddings for
|
48 |
output_column: str = (
|
49 |
-
"
|
50 |
)
|
|
|
46 |
class EmbedRequest(BaseModel):
|
47 |
texts: List[str] # List of strings to generate embeddings for
|
48 |
output_column: str = (
|
49 |
+
"embedding" # Column to store embeddings (default: "embeddings")
|
50 |
)
|
src/main.py
CHANGED
@@ -366,6 +366,9 @@ async def create_embedding(
|
|
366 |
# Step 1: Query the database
|
367 |
logger.info("Fetching data from the database...")
|
368 |
result = await db.fetch(request.query)
|
|
|
|
|
|
|
369 |
dataset = Dataset.from_dict(result)
|
370 |
|
371 |
# Step 2: Generate embeddings
|
|
|
366 |
# Step 1: Query the database
|
367 |
logger.info("Fetching data from the database...")
|
368 |
result = await db.fetch(request.query)
|
369 |
+
|
370 |
+
logger.info(f"{result}")
|
371 |
+
|
372 |
dataset = Dataset.from_dict(result)
|
373 |
|
374 |
# Step 2: Generate embeddings
|