amaye15 commited on
Commit
5c144a1
·
1 Parent(s): 0611c31

Debug - create embeddings endpoint

Browse files
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
- "embeddings" # Column to store embeddings (default: "embeddings")
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