^v(jSrSSKJr SSKJrJr SSKrSSKJ r SSK J r "SS\5r "S S 5r SS \S-S \S-S \ 4Sjjrg)z0Embedding generation service for table metadata.)Enum)AnyListN)logger) get_settingsc \rSrSrSrSrSrSrg)EmbeddingProvider zAvailable embedding providers.localopenaiN)__name__ __module__ __qualname____firstlineno____doc__LOCALOPENAI__static_attributes__r 1/home/mande/repo/quber/src/quber/db/embeddings.pyr r s( E Frr c \rSrSrSrSS\\-S\S\S-4SjjrS\S\4SjrS\4S jr SS \ \S \ S \ S S4Sjjr S \ \S \ S \ S S4SjrS \ \S S4SjrS\S\S S4SjrSrg)EmbeddingServicezDGenerate embeddings for table metadata using local or OpenAI models.Nproviderdevice model_namec[U[5(a [U5nXlX lSUlSUlSUlSUlU[R:XaURU=(d SU5 O=U[R:XaURU=(d S5 O[SU35e[R"SUR S35 g) z Initialize the embedding service. Args: provider: Embedding provider ('local' or 'openai') device: Device for local model ('cuda' or 'cpu') model_name: Optional custom model name (defaults per provider) NrzBAAI/bge-large-en-v1.5ztext-embedding-3-smallzUnsupported provider: z Initialized z embedding service) isinstancestrr rrmodelclientr embedding_dimrinit_local_modelrinit_openai_client ValueErrorrinfovalue)selfrrrs r__init__EmbeddingService.__init__s h $ $(2H    !"# (.. .  ! !*"H0H& Q *11 1  # #J$J2J K5hZ@A A l8>>"22DEFrcSSKJn [R"SUSU35 U"U5UlUS:XaSSKnURR5(aRUR RU5 [R"SUS URRS5S 35 OT[R"S 5 UR RS 5 S Ul OUR RS 5 XlUR R5Ul[R"S UR 35 g![an[S5UeSnAff=f)z-Initialize local sentence-transformers model.r)SentenceTransformerzTsentence-transformers not installed. Install with: pip install sentence-transformersNzLoading local model: z on cudazModel loaded on z (GPU: )zCUDA not available, using CPUcpuzModel embedding dimension: )sentence_transformersr. ImportErrorrr(r"torchr/ is_availabletosuccessget_device_namewarningrr get_sentence_embedding_dimensionr$)r*rrr.errr4s rr%!EmbeddingService.init_local_model7s  A  +J? e$# JJMM% $!ZZHHJ 1$2D2D1EFG1 f  sE E$ EE$c<SSKJn [5RR nU(d [ S5eU"US9UlXlSU;aSOS Ul [R"S US URS 35 g![an[S5UeSnAff=f) zInitialize OpenAI client.r)OpenAIz6openai not installed. Install with: pip install openaiNz+OPENAI_API_KEY environment variable not set)api_keyz3-smallii zUsing OpenAI model: z (z dims)) r r>r3rllmopenai_api_keyr'r#rr$rr()r*rr>r;r?s rr&#EmbeddingService.init_openai_clientUs a %.$$33JK KW- $%.*%USU3nURU/5nUS$)z Generate embedding for table metadata. Combines title and description into a single text for embedding. Args: title: Table title (llm_title) description: Table description (llm_description) Returns: Embedding vector as NumPy array z r)rL)r*rfrgtextrUs rembed_table_metadata%EmbeddingService.embed_table_metadatas.[M*ZZ' !}r)r#rr$r"rr)r r/N) F)rrrrrr!r r+r%r&rintboolrLrJrKrjrr rrrrsN-4!% G))GG$J GBH3HH<USU$MR,#Y,,/,EI, ,,cDUk($s)(0F("#CDZrrrrrFcx[5RnUc URnUc URn[ XS9$)a) Factory function to create an embedding service based on environment config. Args: provider: Override provider (defaults to EMBEDDING_PROVIDER env var or 'local') device: Override device (defaults to EMBEDDING_DEVICE env var or 'cuda') Returns: Configured EmbeddingService instance )rr)rrUrrr)rrembedding_settingss rget_embedding_servicerqs>&22%.. ~#** X ==r)NN)renumrtypingrrnumpyrHlogururquber.settingsrr rr!rqr rrrwsS6'TTn>C$J>sTz>Ue>r