תיאור מפורט של הפונקציות של Vector Assist

בוחרים גרסת תיעוד:

כדי לפשט את עומסי העבודה של וקטורים ב-AlloyDB Omni, אפשר להיעזר בפונקציות המפורטות הבאות של סיוע וקטורי.

vector_assist.define_spec

הפונקציה מגדירה מפרט וקטורי חדש, או מפרט וקטורי, ומחזירה המלצות שנדרשות כדי להגדיל את נפח העבודה של הווקטור. הפונקציה הזו מיישמת מפרטים שהמשתמש מספק כדי ליצור את ההמלצות.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.define_spec(
    spec_id TEXT DEFAULT NULL,
    table_name TEXT,
    schema_name TEXT DEFAULT NULL,
    vector_column_name TEXT DEFAULT NULL,
    text_column_name TEXT DEFAULT NULL,
    vector_index_type TEXT DEFAULT NULL, -- Allowed: 'hnsw', 'ivfflat'
    embeddings_available BOOL DEFAULT NULL,
    embedding_model TEXT DEFAULT NULL,
    num_vectors INT DEFAULT NULL,
    dimensionality INT DEFAULT NULL,
    distance_func TEXT DEFAULT NULL, -- Allowed: 'cosine', 'ip', 'l2', 'l1'
    quantization TEXT DEFAULT NULL, -- Allowed: 'none', 'halfvec', 'bit'
    prefilter_column_names TEXT[] DEFAULT NULL,
    memory_budget_kb INT DEFAULT NULL, -- In KB
    target_recall FLOAT DEFAULT NULL,
    target_top_k INT DEFAULT NULL,
    tune_vector_index BOOL DEFAULT FALSE, -- Set to true if target_recall is set
)
RETURNS SETOF vector_assist.RECOMMENDATIONS;

פרמטרים

פרמטר חובה? תיאור
spec_id אופציונלי, ייווצר אוטומטית אם לא תספקו אותו מזהה ייחודי של מפרט הווקטור שאתם מגדירים.
table_name חובה שם הטבלה שבה יש להשתמש בהגדרת עומס העבודה של הווקטורים.
schema_name אופציונלי הסכימה שמכילה את הטבלה שתשמש לעומס העבודה הווקטורי. הסכימה מוסקת באופן אוטומטי על סמך נתיב החיפוש של הסכימה.
vector_column_name הצגת התיאור שם העמודה שמכילה את הטמעות הווקטורים.

העמודה הזו נדרשת אם משתמשים ביותר מעמודת וקטור אחת ולא מצהירים על text_column_name. אם משתמשים רק בעמודת וקטור אחת, הפרמטר הזה הוא אופציונלי.
text_column_name הצגת התיאור שם העמודה של עמודת הטקסט. אם עומס העבודה של הווקטור דורש יצירת הטמעה מעמודת טקסט, הפרמטר הזה נדרש.

חובה לכלול את העמודה הזו אם הפרמטר vector_column_name הוא NULL ואין עמודות וקטוריות אחרות.

הפרמטר הזה משתמש בתוסף google_ml_integration כדי ליצור הטמעות מעמודת הטקסט שצוינה.
vector_index_type אופציונלי סוג אינדקס הווקטורים. ערכים קבילים:
  • (ברירת מחדל) HNSW
  • IVFFlat
embeddings_available אופציונלי ערך בוליאני שבודק אם קיימים הטמעות וקטוריות בטבלה שנבחרה. השדה הזה מוגדר לערך FALSE אם לא מזוהות עמודות וקטוריות והפרמטר text_column_name מוגדר.
embedding_model הצגת התיאור מזהה המודל ששימש ליצירת ההטמעות. השדה הזה הוא אופציונלי, אלא אם הפרמטר embeddings_available הוא FALSE או NULL. ערך ברירת המחדל הוא text-embedding-005.
num_vectors אופציונלי מספר הווקטורים בטבלה.
dismensionality אופציונלי הממד של הווקטורים הנתונים.
distance_func אופציונלי פונקציית המרחק שתשמש להשוואה בין וקטורים. ערכים קבילים:
  • ‫cosine: ברירת המחדל כש-quantization לא מוגדר ל-bit
  • ip
  • l2
  • l1
  • jaccard
  • ‫hamming: ברירת המחדל כש-quantization מוגדר ל-bit
quantization אופציונלי הקוונטיזציה של אינדקסים של וקטורים. ערכים קבילים:
  • ‫none: ברירת מחדל אם dimensionality קטן מ-2000
  • ‫halfvec: ברירת המחדל כש-dimensionality הוא בין 2000 לבין 4000
  • bit: ברירת המחדל כש-dimensionality גדול מ-4,000
prefilter_column_names אופציונלי הפונקציה ממליצה על אינדקס לעמודה שצוינה לצורך סינון מראש. הפעולה הזו עשויה לשפר את הביצועים של שאילתות עם מסנן סלקטיביות גבוהה בעמודה.
memory_budget_kb אופציונלי הזיכרון המקסימלי (ב-KB) ליצירת אינדקס. ערך ברירת המחדל הוא 60% מזיכרון המופע.
target_recall אופציונלי ההחזרה של התוצאות בשאילתות וקטוריות רגילות.
target_top_k אופציונלי מספר הערכים התואמים המובילים שיש לאחזר לחישוב ההחזרה. הערך הזה משמש אם מציינים את הפרמטר target_recall.
tune_vector_index אופציונלי מכוונן את מדדי הווקטור בעומס העבודה. אם האפשרות הזו מופעלת, התכונה 'סיוע וקטורי' מכוונת אוטומטית את אינדקס הווקטור ואת שאילתת החיפוש בהתאם ליעדי ההחזרה של הטירגוט, באמצעות מפרט הווקטור, ומחשבת את הפרמטרים האופטימליים של מסד הנתונים והאינדקס. ערכים קבילים:
  • (ברירת מחדל)FALSE: שינוי רק של פרמטר החיפוש (ef_search). נעשה שימוש בערכי הפרמטר של משך זמן של תהליך build של האינדקס שמוגדר כברירת מחדל (m,‏ ef_construction).
  • ‫TRUE: כוונון של פרמטרים שקשורים לחיפוש ולזמן בניית האינדקס.

הערך שמוחזר

הפונקציה מחזירה טבלה של המלצות, שכוללת את המזהה של מפרט הווקטור שנוצר לאחרונה.

vector_assist.list_specs

מחזירה את מפרט הווקטור הנוכחי עבור טבלה ועמודה נתונות.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.list_specs(
  table_name      TEXT,
  column_name     TEXT DEFAULT NULL
)
RETURNS SETOF vector_assist.VECTOR_SPECS

פרמטרים

פרמטר חובה? תיאור
table_name חובה שם הטבלה שלגביה מאחזרים את מפרט הווקטור.
column_name אופציונלי שם העמודה שלפיה רוצים לסנן את הטבלה לפני החזרת המפרט. אם מציינים את שם העמודה, המערכת מחזירה מפרטים שמכילים שמות של עמודות טקסט או וקטור שתואמים לשם העמודה הזה.

הערך שמוחזר

מחזירה טבלה של מפרטים עבור הטבלה הנתונה, ואפשר גם לציין שם עמודה.

vector_assist.modify_spec

משנה מפרט וקטורי קיים. כשמשנים את המפרט הווקטורי, ההמלצות המשויכות נוצרות מחדש על סמך המפרטים החדשים.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.modify_spec(
  spec_id                TEXT,
  table_name             TEXT DEFAULT NULL,
  schema_name            TEXT DEFAULT NULL,
  vector_column_name     TEXT DEFAULT NULL,
  text_column_name       TEXT DEFAULT NULL,
  vector_index_type      TEXT DEFAULT NULL,
  embeddings_available   BOOL DEFAULT NULL,
  num_vectors            INT DEFAULT NULL,
  dimensionality         INT DEFAULT NULL,
  embedding_model        TEXT DEFAULT NULL,
  prefilter_column_names TEXT[] DEFAULT NULL,
  distance_func          TEXT DEFAULT NULL,
  quantization           TEXT DEFAULT NULL,
  memory_budget_kb       INT DEFAULT NULL,
  target_recall          FLOAT DEFAULT NULL,
  target_top_k           INT DEFAULT NULL,
  tune_vector_index      BOOL DEFAULT NULL
)
RETURNS SETOF vector_assist.RECOMMENDATIONS

פרמטרים

פרמטר חובה? תיאור
spec_id חובה מזהה ייחודי של מפרט הווקטור שאתם מגדירים.
table_name חובה שם הטבלה שבה יש להשתמש בהגדרת עומס העבודה של הווקטורים.
schema_name אופציונלי הסכימה שמכילה את הטבלה שתשמש לעומס העבודה הווקטורי. הסכימה מוסקת באופן אוטומטי על סמך נתיב החיפוש של הסכימה.
vector_column_name הצגת התיאור שם העמודה שמכילה את הטמעות הווקטורים.

העמודה הזו נדרשת אם משתמשים ביותר מעמודת וקטור אחת ולא מצהירים על text_column_name. אם משתמשים רק בעמודת וקטור אחת, הפרמטר הזה הוא אופציונלי.
text_column_name הצגת התיאור שם העמודה של עמודת הטקסט. אם עומס העבודה של הווקטור דורש יצירת הטמעה מעמודת טקסט, הפרמטר הזה נדרש.

חובה לכלול את העמודה הזו אם הפרמטר vector_column_name הוא NULL ואין עמודות וקטוריות אחרות.

הפרמטר הזה משתמש בתוסף google_ml_integration כדי ליצור הטמעות מעמודת הטקסט שצוינה.
vector_index_type אופציונלי סוג אינדקס הווקטורים. ערכים קבילים:
  • HNSW
  • IVFFlat
embeddings_available אופציונלי ערך בוליאני שבודק אם קיימים הטמעות וקטוריות בטבלה שנבחרה. השדה הזה מוגדר לערך FALSE אם לא מזוהות עמודות וקטוריות והפרמטר text_column_name מוגדר.
embedding_model הצגת התיאור מזהה המודל ששימש ליצירת ההטמעות. השדה הזה הוא אופציונלי, אלא אם הפרמטר embeddings_available הוא FALSE או NULL. ערך ברירת המחדל הוא text-embedding-005.
num_vectors אופציונלי מספר הווקטורים בטבלה.
dismensionality אופציונלי הממד של הווקטורים הנתונים.
distance_func אופציונלי פונקציית המרחק שתשמש להשוואת וקטורים. הערכים הקבילים כוללים cosine,‏ ip,‏ l2 או l1.
quantization אופציונלי הקוונטיזציה של אינדקסים של וקטורים. ערכים קבילים כוללים את none,‏ halfvec או bit.
prefilter_column_names אופציונלי מזהה את העמודות שזמינות לסינון מראש בשאילתות של חיפוש וקטורי.
memory_budget_kb אופציונלי הזיכרון המקסימלי (ב-KB) ליצירת אינדקס.
target_recall אופציונלי ההחזרה של התוצאות בשאילתות וקטוריות רגילות.
target_top_k אופציונלי מספר הערכים התואמים המובילים שיש לאחזר לחישוב ההחזרה. הערך הזה משמש אם מציינים את הפרמטר target_recall.
tune_vector_index אופציונלי מכוונן את מדדי הווקטור בעומס העבודה. אם האפשרות הזו מופעלת, התכונה 'סיוע וקטורי' מכוונת אוטומטית את אינדקס הווקטור או את שאילתת החיפוש באמצעות מפרט הווקטור, ומחשבת את הפרמטרים האופטימליים של מסד הנתונים והאינדקס. הערכים האפשריים הם TRUE או FALSE.

הערך שמוחזר

הפונקציה מחזירה טבלה מעודכנת של המלצות.

vector_assist.get_recommendations

הפונקציה מחזירה המלצות שמשויכות לווקטור ספציפי spec_id.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.get_recommendations (
    spec_id TEXT
)
RETURNS SETOF vector_assist.RECOMMENDATIONS;

פרמטרים

פרמטר חובה? תיאור
spec_id חובה מקבלים המלצות לגבי הווקטור שצוין spec_id.

הערך שמוחזר

מחזירה טבלה של המלצות שצריך להחיל כדי להגדיר את עומס העבודה של הווקטור המבוקש. כל שורה היא המלצה אחת שמכילה שאילתת SQL שניתן להפעיל והסבר מפורט למה כדאי ליישם את ההמלצה. אפשר ליישם את ההמלצה באמצעות הפונקציה apply_recommendation().

טבלת ההמלצות כוללת את העמודות הבאות:

שם העמודה סוג תיאור
recommendation_id טקסט מזהה ההמלצה. ההמלצה מיושמת באמצעות המזהה הזה.
vector_spec_id טקסט המזהה של מפרט הווקטור שמשויך להמלצה.
table_name טקסט שם הטבלה שמשויכת להמלצה.
schema_name טקסט שם הסכימה שמכילה את הטבלה הנתונה.
query טקסט שאילתת SQL שניתן להריץ כדי להשיג את התוצאה של ההמלצה.
recommendation טקסט ההסבר להמלצה, כולל התוצאה הצפויה אחרי יישום ההמלצה. העמודה הזו מכילה גם את הסיבה וההסבר להמלצה. אפשר להשתמש בהקשר הזה כדי לשנות את השאילתה לפי הצורך וליצור מחדש המלצות.
applied בוליאני האם ההמלצה יושמה.
modified בוליאני האם המשתמש שינה את ההמלצה.
created_at חותמת זמן עם אזור זמן השעה שבה ההמלצה נוצרה.

vector_assist.apply_recommendation

מיישמת המלצה אחת באמצעות recommendation_id.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.apply_recommendation (
    recommendation_id TEXT
)
RETURNS BOOLEAN;

פרמטרים

פרמטר חובה? תיאור
recommenation_id חובה המזהה של המלצת הווקטור שרוצים להחיל.

הערך שמוחזר

הפונקציה מחזירה ערך בוליאני שמאשר שהמלצה יושמה בהצלחה, ומעדכנת את העמודה vector_assist.RECOMMENDATIONS בטבלה applied לערך TRUE.

אם העזרה בווקטור לא מצליחה להחיל את ההמלצה, תוצג הודעת שגיאה.

vector_assist.apply_spec

מיישמת את כל ההמלצות שמשויכות לווקטור spec_id או לשם טבלה ספציפיים. הפונקציה הזו עוברת על כל ההמלצות ומריצה את הפונקציה vector_assist.apply_recommendation על כל המלצה.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.apply_spec(
    spec_id TEXT DEFAULT NULL,
    table_name TEXT DEFAULT NULL,
    schema_name TEXT DEFAULT NULL,
    column_name TEXT DEFAULT NULL -- text_column_name or vector_column_name
)
RETURNS BOOLEAN;

פרמטרים

כדי להפעיל את הפונקציה, צריך לציין את spec_id או את table_name.

פרמטר חובה? תיאור
spec_id אופציונלי הפונקציה מחילה את כל ההמלצות על הווקטור הנתון spec_id.
table_name אופציונלי החלת כל ההמלצות על הטבלה הנתונה.
schema_name אופציונלי הסכימה שמכילה את הטבלה שצוינה.
column_name אופציונלי מזהה את text_column_name או את vector_column_name שבהם צריך להשתמש אם הטבלה מכילה כמה מפרטי וקטורים.

הערך שמוחזר

הפונקציה מחזירה ערך בוליאני שמאשר שכל ההמלצות יושמו בהצלחה, ומגדירה את הערך בעמודה vector_assist.RECOMMENDATIONS בטבלה applied ל-TRUE עבור כל ההמלצות.

אם העזרה בווקטור לא מצליחה להחיל את ההמלצה, תוצג הודעת שגיאה.

vector_assist.modify_recommendation

משנה את השאילתה להמלצה ספציפית ומגדיר את השדה modified בטבלה vector_assist.RECOMMENDATIONS להמלצה המשויכת לערך TRUE.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.modify_recommendation(
    recommendation_id TEXT,
    modified_query TEXT
)
RETURNS BOOLEAN;

פרמטרים

פרמטר חובה? תיאור
recommendation_id חובה המזהה של ההמלצה שרוצים לשנות.
modified_query חובה שאילתת SQL שמשתמש שינה בשביל ההמלצה.

הערך שמוחזר

הפונקציה מחזירה ערך בוליאני שמאשר אם המלצה עודכנה בהצלחה ומגדירה את העמודה vector_assist.RECOMMENDATIONS בטבלה modified לערך TRUE.

אם העזרה בווקטור לא מצליחה לעדכן את ההמלצה, תוצג הודעת שגיאה.

vector_assist.generate_query

יצירת שאילתת חיפוש וקטורית שעברה אופטימיזציה על סמך מפרט הווקטור ופרמטרי החיפוש שהוגדרו. אם רלוונטי, הפונקציה הזו עשויה לשנות את הערך של ef_search כדי לעמוד בדרישות של target_recall.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.generate_query(
    spec_id TEXT DEFAULT NULL,
    table_name TEXT DEFAULT NULL,
    schema_name TEXT DEFAULT NULL,
    column_name TEXT DEFAULT NULL, -- text_column_name or vector_column_name of the spec
    search_text TEXT DEFAULT NULL,
    search_vector VECTOR DEFAULT NULL, -- e.g. '[1.0,2.0,3.0]'::vector
    output_column_names TEXT[] DEFAULT NULL, -- Columns to retrieve, NULL for SELECT *
    top_k INT DEFAULT NULL,
    filter_expressions TEXT[] DEFAULT NULL,
    target_recall FLOAT DEFAULT NULL, -- Inherited from spec, can be overridden
    iterative_index_search BOOL DEFAULT NULL
)
RETURNS TEXT; -- Returns the text representing the search query

פרמטרים

כדי להפעיל את הפונקציה, צריך לציין את spec_id או את table_name.

פרמטר חובה? תיאור
spec_id אופציונלי מציינים את spec_id שעבורו רוצים ליצור את השאילתה.
table_name אופציונלי מציין את שם הטבלה ליצירת שאילתת החיפוש אם לא צוין spec_id. אם טבלה מכילה כמה מפרטים, הפרמטר מחזיר שגיאה.
column_name אופציונלי מציינת את שם העמודה ליצירת שאילתת החיפוש אם לא צוינה ההגדרה spec_id. אם העמודה מכילה כמה מפרטים, הפרמטר מחזיר שגיאה. הערך של column_name יכול להיות text_column_name או vector_column_name.
output_column_name אופציונלי שמות העמודות לאחזור בשאילתת הפלט. כדי לאחזר את כל העמודות, מגדירים את הפרמטר הזה לערך NULL.
search_text אופציונלי טקסט החיפוש ליצירת הטמעות. ההטמעה נוצרת באמצעות embedding_model מהמפרט. אפשר להגדיר רק את הפרמטר search_text או את הפרמטר search_vector.
search_vector אופציונלי הווקטור של שאילתת החיפוש. אפשר להגדיר רק את הפרמטר search_text או את הפרמטר search_vector
top_k אופציונלי מספר השכנים הקרובים שיש להחזיר.
filter_expressions אופציונלי ביטויי הסינון להחלה בשאילתת החיפוש. אפשר להחיל את המסננים על שדות בטבלה, כמו מחיר < 100.
iterative_index_search אופציונלי ההגדרה הזו קובעת אם לבצע חיפוש אינדקס איטרטיבי עבור שאילתות עם מסננים.
target_recall אופציונלי ההחזרה של התוצאות הרלוונטיות רק לשאילתת החיפוש. אם ההגדרה הזו מוגדרת, הפונקציה עשויה להחזיר המלצה לערך ef_search אם היא מוצאת ערך טוב יותר.

הערך שמוחזר

הפונקציה מחזירה טקסט שמכיל את שאילתת החיפוש הווקטורי עבור הפרמטרים שצוינו. אם מציינים את הערך target_recall, יכול להיות שגם המלצה לערך ef_search תיכלל אם יימצא ערך טוב יותר. אפשר להריץ את השאילתה הזו או לשמור אותה להרצה מאוחר יותר.

vector_assist.evaluate_query_recall

הפונקציה מחשבת את ערך ה-recall של אינדקס הווקטורים עבור שאילתה נתונה. הפונקציה מקבלת כקלט מחרוזת JSON עם הגדרות של מסד הנתונים והאינדקס.

תפקיד

CREATE OR REPLACE FUNCTION vector_assist.evaluate_query_recall(
  ann_query text,
  vector_search_configurations json DEFAULT NULL,
  recall float,
  ann_execution_time float,
  ground_truth_execution_time float,
  index_type text)
RETURNS RECORD

פרמטרים

פרמטר חובה? תיאור
ann_query חובה שאילתת ה-SQL שצריך להעריך את ההחזרה שלה. השאילתה הזו צריכה להיות שאילתת pgvector ANN.
vector_search_configurations אופציונלי ההגדרות שאפשר להגדיר לשאילתת ANN. הפרמטר הזה חייב להיות בפורמט JSON.

הערך שמוחזר

הפונקציה מחזירה טבלה שמעריכה את הזיכרון של אינדקס הווקטור לשאילתה הנתונה.

הטבלה מכילה את העמודות הבאות:

שם העמודה סוג תיאור
recall Float אחוז הווקטורים שמוחזרים מהאינדקס, שהם השכנים הקרובים האמיתיים. המדד הזה משמש לכימות איכות החיפוש. הפונקציה הזו מאפשרת לכם לשנות את הפרמטרים כדי לקבל את התוצאות הצפויות של שאילתת הווקטור.
ann_execution_time Float הזמן שלוקח להריץ שאילתת וקטור באמצעות סריקות אינדקס.
ground_truth_execution_time Float הזמן שלוקח להריץ את השאילתה באמצעות סריקה רציפה.
index_type טקסט סוג אינדקס הווקטורים, למשל HNSW.

דוגמה

בדוגמה הבאה מחושב ה-recall של אינדקס HNSW, ומוגדר ערך חדש לפרמטר ההגדרה HNSW.ef_search:

SELECT * FROM vector_assist.evaluate_query_recall(
  ann_query => 'SELECT id FROM items ORDER BY embedding <=> ''[1,2,3]'' LIMIT 10',
  vector_search_configurations => '{"hnsw.ef_search": 30}'
);