Skip to content

Commit 61e29d8

Browse files
committed
fix(generate_taxonomy_embeddings): add UUIDHistory and originUUID fields to the output COM-606
1 parent 0149c5b commit 61e29d8

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

backend/scripts/embeddings/generate_taxonomy_embeddings.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,22 +1,20 @@
11
#!/usr/bin/env python3
2-
import time
3-
4-
import vertexai
52
import argparse
63
import asyncio
4+
import logging.config
5+
import time
76
from datetime import datetime
87

9-
from app.vector_search.embeddings_model import EmbeddingService
10-
from app.vector_search.vector_search_dependencies import get_embeddings_service
11-
from pydantic import BaseModel
12-
13-
from tqdm import tqdm
14-
import logging.config
15-
from dotenv import load_dotenv
8+
import vertexai
169
from bson.objectid import ObjectId
10+
from dotenv import load_dotenv
1711
from motor.motor_asyncio import AsyncIOMotorClient
12+
from pydantic import BaseModel
13+
from tqdm import tqdm
1814

1915
from _base_data_settings import EmbeddingsScriptSettings, CompassEmbeddingsCollections, PlatformCollections
16+
from app.vector_search.embeddings_model import EmbeddingService
17+
from app.vector_search.vector_search_dependencies import get_embeddings_service
2018
from common_libs.logging.log_utilities import setup_logging_config
2119
from common_libs.time_utilities import get_now, datetime_to_mongo_date
2220
from scripts.embeddings._common import EmbeddingContext, generate_indexes, redact_credentials_from_uri
@@ -128,6 +126,8 @@ async def generate_and_save_embeddings(*,
128126
"altLabels": document["altLabels"],
129127
"description": document["description"],
130128
"updatedAt": datetime.now(),
129+
"UUIDHistory": document["UUIDHistory"],
130+
"originUUID": document["UUIDHistory"][-1],
131131
ctx.id_field_name: document["_id"]
132132
}
133133

0 commit comments

Comments
 (0)