diff --git a/docs/docs/integrations/text_embedding/solar.ipynb b/docs/docs/integrations/text_embedding/solar.ipynb new file mode 100644 index 0000000000..a2e2443bcb --- /dev/null +++ b/docs/docs/integrations/text_embedding/solar.ipynb @@ -0,0 +1,2257 @@ +{ + "cells": [ + { + "cell_type": "markdown", + "id": "0f1199c1-f885-4290-b5e7-d1defd49abe1", + "metadata": {}, + "source": [ + "# Soalr\n", + "\n", + "[Solar](https://console.upstage.ai/services/embedding) offers an embeddings service.\n", + "\n", + "This example goes over how to use LangChain to interact with Solar Inference for text embedding." + ] + }, + { + "cell_type": "code", + "execution_count": 9, + "id": "595c52be-ee54-4a67-83e0-066b6980d240", + "metadata": { + "ExecuteTime": { + "end_time": "2023-05-24T15:13:15.397075Z", + "start_time": "2023-05-24T15:13:15.387540Z" + }, + "execution": { + "iopub.execute_input": "2024-03-29T15:39:46.059500Z", + "iopub.status.busy": "2024-03-29T15:39:46.058840Z", + "iopub.status.idle": "2024-03-29T15:39:46.066609Z", + "shell.execute_reply": "2024-03-29T15:39:46.063869Z", + "shell.execute_reply.started": "2024-03-29T15:39:46.059467Z" + } + }, + "outputs": [], + "source": [ + "import os\n", + "\n", + "os.environ[\"SOLAR_API_KEY\"] = \"\"" + ] + }, + { + "cell_type": "code", + "execution_count": 2, + "id": "d25dc22d-b656-46c6-a42d-eace958590cd", + "metadata": { + "ExecuteTime": { + "end_time": "2023-05-24T15:13:17.176956Z", + "start_time": "2023-05-24T15:13:15.399076Z" + }, + "execution": { + "iopub.execute_input": "2024-03-29T15:39:19.252281Z", + "iopub.status.busy": "2024-03-29T15:39:19.252101Z", + "iopub.status.idle": "2024-03-29T15:39:19.339106Z", + "shell.execute_reply": "2024-03-29T15:39:19.338614Z", + "shell.execute_reply.started": "2024-03-29T15:39:19.252260Z" + } + }, + "outputs": [], + "source": [ + "from langchain_community.embeddings import SolarEmbeddings" + ] + }, + { + "cell_type": "code", + "execution_count": 3, + "id": "8397b91f-a1f9-4be6-a699-fedaada7c37a", + "metadata": { + "ExecuteTime": { + "end_time": "2023-05-24T15:13:17.193751Z", + "start_time": "2023-05-24T15:13:17.182053Z" + }, + "execution": { + "iopub.execute_input": "2024-03-29T15:39:19.901573Z", + "iopub.status.busy": "2024-03-29T15:39:19.900935Z", + "iopub.status.idle": "2024-03-29T15:39:19.906540Z", + "shell.execute_reply": "2024-03-29T15:39:19.905345Z", + "shell.execute_reply.started": "2024-03-29T15:39:19.901529Z" + } + }, + "outputs": [], + "source": [ + "embeddings = SolarEmbeddings()" + ] + }, + { + "cell_type": "code", + "execution_count": 4, + "id": "abcf98b7-424c-4691-a1cd-862c3d53be11", + "metadata": { + "ExecuteTime": { + "end_time": "2023-05-24T15:13:17.844903Z", + "start_time": "2023-05-24T15:13:17.198751Z" + }, + "execution": { + "iopub.execute_input": "2024-03-29T15:39:20.434581Z", + "iopub.status.busy": "2024-03-29T15:39:20.433117Z", + "iopub.status.idle": "2024-03-29T15:39:22.178650Z", + "shell.execute_reply": "2024-03-29T15:39:22.176058Z", + "shell.execute_reply.started": "2024-03-29T15:39:20.434501Z" + }, + "scrolled": true + }, + "outputs": [], + "source": [ + "query_text = \"This is a test query.\"\n", + "query_result = embeddings.embed_query(query_text)" + ] + }, + { + "cell_type": "code", + "execution_count": 5, + "id": "e68b5cc1-8c6b-40bc-8103-ba40e2e06a29", + "metadata": { + "collapsed": true, + "execution": { + "iopub.execute_input": "2024-03-29T15:39:22.182986Z", + "iopub.status.busy": "2024-03-29T15:39:22.182334Z", + "iopub.status.idle": "2024-03-29T15:39:22.207603Z", + "shell.execute_reply": "2024-03-29T15:39:22.206733Z", + "shell.execute_reply.started": "2024-03-29T15:39:22.182936Z" + }, + "jupyter": { + "outputs_hidden": true + }, + "scrolled": true + }, + "outputs": [ + { + "data": { + "text/plain": [ + "[-0.009612835943698883,\n", + " 0.005192634183913469,\n", + " -0.0007243562722578645,\n", + " -0.02104002982378006,\n", + " -0.004770803730934858,\n", + " -0.024557538330554962,\n", + " -0.03355177119374275,\n", + " 0.002088239649310708,\n", + " 0.005196372978389263,\n", + " -0.025660645216703415,\n", + " -0.00485575944185257,\n", + " -0.015621133148670197,\n", + " 0.014192958362400532,\n", + " -0.011372988112270832,\n", + " 0.02780674397945404,\n", + " 0.0032780447509139776,\n", + " -0.015384051948785782,\n", + " 0.014557680115103722,\n", + " -0.002221834147349,\n", + " -0.004098917823284864,\n", + " 0.019031716510653496,\n", + " 0.0012823417782783508,\n", + " 0.00443899305537343,\n", + " 0.010559789836406708,\n", + " 0.0029694491531699896,\n", + " 0.006230773404240608,\n", + " -0.006915881764143705,\n", + " 0.007640184834599495,\n", + " 0.002265951596200466,\n", + " -0.00772814080119133,\n", + " 0.009235503152012825,\n", + " 0.006972184870392084,\n", + " -0.01011792290955782,\n", + " -0.01449803076684475,\n", + " 0.0034380410797894,\n", + " 0.017988374456763268,\n", + " -0.001981367589905858,\n", + " 0.019687853753566742,\n", + " 0.00599881773814559,\n", + " -0.033464811742305756,\n", + " -0.005420745350420475,\n", + " 0.026795821264386177,\n", + " -0.02160714939236641,\n", + " -0.013100927695631981,\n", + " 0.008083999156951904,\n", + " 0.014485755935311317,\n", + " -0.0009732113685458899,\n", + " -0.012884712778031826,\n", + " 0.025087689980864525,\n", + " -0.03585042431950569,\n", + " 0.04038093611598015,\n", + " -0.0028256087098270655,\n", + " -0.0011333064176142216,\n", + " 0.12208127230405807,\n", + " 0.01880730129778385,\n", + " 0.01855185627937317,\n", + " -0.0038447133265435696,\n", + " 0.014112002216279507,\n", + " 0.0018906412879005075,\n", + " 0.010727775283157825,\n", + " 0.007657645735889673,\n", + " -0.010718741454184055,\n", + " 0.0009449812932871282,\n", + " 0.00786784291267395,\n", + " -0.004893230274319649,\n", + " -0.0017297398298978806,\n", + " -0.014865854755043983,\n", + " -0.0161128006875515,\n", + " -0.02509428933262825,\n", + " -0.011645237915217876,\n", + " -0.02115057036280632,\n", + " 0.027240969240665436,\n", + " -0.03013959713280201,\n", + " 0.020873211324214935,\n", + " -0.023083331063389778,\n", + " -0.014274565503001213,\n", + " 0.018262118101119995,\n", + " 0.03854333609342575,\n", + " -0.011523822322487831,\n", + " 0.0016228322638198733,\n", + " 0.018532132729887962,\n", + " -0.021359337493777275,\n", + " -0.0073746913112699986,\n", + " -0.00306093436665833,\n", + " -0.0224248506128788,\n", + " 0.022097807377576828,\n", + " 0.009984304197132587,\n", + " -0.009028983302414417,\n", + " 0.014073910191655159,\n", + " -0.01653546467423439,\n", + " 0.005302212201058865,\n", + " -0.0038526973221451044,\n", + " -0.002201930619776249,\n", + " -0.010180548764765263,\n", + " -0.014118028804659843,\n", + " 0.0020483224652707577,\n", + " -0.004559666849672794,\n", + " -0.025423899292945862,\n", + " -0.030747853219509125,\n", + " 0.0042665028013288975,\n", + " 0.01872953400015831,\n", + " -0.015436792746186256,\n", + " -0.0012506360653787851,\n", + " 0.002480799565091729,\n", + " 0.012969587929546833,\n", + " -0.0030781375244259834,\n", + " -0.003880476113408804,\n", + " 0.005123113747686148,\n", + " 0.01838541217148304,\n", + " -0.012043023481965065,\n", + " 0.005955793894827366,\n", + " 0.005877435207366943,\n", + " -0.021440008655190468,\n", + " 0.007327971048653126,\n", + " 0.005668329074978828,\n", + " -0.020333116874098778,\n", + " -0.010220373049378395,\n", + " -0.025336747989058495,\n", + " 0.009634329937398434,\n", + " -0.022424353286623955,\n", + " 0.0036242357455193996,\n", + " 0.019212981685996056,\n", + " 0.0008957164827734232,\n", + " -0.0027208265382796526,\n", + " 0.0007398341549560428,\n", + " -0.014906578697264194,\n", + " 0.0026832111179828644,\n", + " 0.008843235671520233,\n", + " -0.009975744411349297,\n", + " -0.017594290897250175,\n", + " -0.007390517275780439,\n", + " -0.0018038008129224181,\n", + " 0.025810424238443375,\n", + " -0.03663061559200287,\n", + " -0.016672957688570023,\n", + " 0.009701783768832684,\n", + " -0.015615193173289299,\n", + " -0.0017102754209190607,\n", + " 0.018835289403796196,\n", + " -0.014688814990222454,\n", + " -0.02063092403113842,\n", + " 0.020857617259025574,\n", + " -0.007686559576541185,\n", + " 0.01151837594807148,\n", + " 0.0033596211578696966,\n", + " -0.014537064358592033,\n", + " 0.0036612350959330797,\n", + " -0.013696428388357162,\n", + " -0.011392973363399506,\n", + " -0.001989757176488638,\n", + " -0.020189374685287476,\n", + " -0.024850046262145042,\n", + " 0.00836894754320383,\n", + " -0.020748576149344444,\n", + " -0.004396480042487383,\n", + " 0.03407088667154312,\n", + " 0.021833691745996475,\n", + " 0.02130814827978611,\n", + " 0.006232410203665495,\n", + " 0.0039503793232142925,\n", + " -0.012550112791359425,\n", + " -0.021708764135837555,\n", + " -0.004276007879525423,\n", + " 0.02033841423690319,\n", + " -0.003566763596609235,\n", + " 0.021997885778546333,\n", + " -0.01681455411016941,\n", + " -0.018676014617085457,\n", + " 0.01742757298052311,\n", + " -0.00598341366276145,\n", + " 0.009576573967933655,\n", + " -0.027214830741286278,\n", + " -0.011387384496629238,\n", + " -0.003966265358030796,\n", + " 0.013394222594797611,\n", + " 0.00260531110689044,\n", + " -0.0018310232553631067,\n", + " -0.004507850389927626,\n", + " -0.02329740673303604,\n", + " -0.0011288138339295983,\n", + " -0.029134375974535942,\n", + " 0.009268014691770077,\n", + " -0.0029798042960464954,\n", + " -0.0181003175675869,\n", + " 0.010883892886340618,\n", + " 0.003947356250137091,\n", + " 0.012287858873605728,\n", + " 0.012322994880378246,\n", + " 0.01976163126528263,\n", + " -0.009208086878061295,\n", + " 0.02332279458642006,\n", + " -0.024003351107239723,\n", + " -0.01067762915045023,\n", + " -0.02086666040122509,\n", + " 0.012497895397245884,\n", + " -0.018715588375926018,\n", + " -0.01373564638197422,\n", + " 0.01511511579155922,\n", + " -0.004894122015684843,\n", + " 0.0102867865934968,\n", + " 0.01963503472507,\n", + " 0.010318577289581299,\n", + " -0.025310182943940163,\n", + " 0.009286437183618546,\n", + " -0.014914891682565212,\n", + " -0.022296326234936714,\n", + " 0.0092850960791111,\n", + " -0.003506426466628909,\n", + " -0.009920112788677216,\n", + " 0.0064842249266803265,\n", + " -0.006868164520710707,\n", + " 0.010974838398396969,\n", + " 0.0196993350982666,\n", + " -0.015959274023771286,\n", + " -0.01983925700187683,\n", + " -0.0032453376334160566,\n", + " -0.007468512747436762,\n", + " 0.014899743720889091,\n", + " -0.031494736671447754,\n", + " -0.003950838930904865,\n", + " -0.002206148114055395,\n", + " -0.020258402451872826,\n", + " 0.007158157415688038,\n", + " 0.004851853474974632,\n", + " 0.009486673399806023,\n", + " 0.027494588866829872,\n", + " -0.007189360447227955,\n", + " 0.008461890742182732,\n", + " -0.0004358790465630591,\n", + " 0.0076111904345452785,\n", + " -0.0007249601767398417,\n", + " 0.030181696638464928,\n", + " -0.0005211788229644299,\n", + " 0.018220754340291023,\n", + " 0.005095703527331352,\n", + " 0.004471085965633392,\n", + " -0.009794448502361774,\n", + " -0.0030862493440508842,\n", + " 0.025696849450469017,\n", + " -0.004642312414944172,\n", + " 0.004473725333809853,\n", + " 0.0010625463910400867,\n", + " -0.003617122070863843,\n", + " -0.015058541670441628,\n", + " -0.012431029230356216,\n", + " -0.0048724086955189705,\n", + " -0.0003166526439599693,\n", + " -0.009018509648740292,\n", + " 0.012120921164751053,\n", + " -0.006830958649516106,\n", + " 0.0032028749119490385,\n", + " -0.033147528767585754,\n", + " 0.010441552847623825,\n", + " -0.015877237543463707,\n", + " 0.024799810722470284,\n", + " -0.016226306557655334,\n", + " -0.005698256194591522,\n", + " 0.02627044916152954,\n", + " -0.0183611661195755,\n", + " -0.010974086821079254,\n", + " 0.0007014335715211928,\n", + " 0.028307344764471054,\n", + " -0.0016226363368332386,\n", + " -0.009277352131903172,\n", + " -2.833910366462078e-05,\n", + " -0.0024536976125091314,\n", + " 0.0029181847348809242,\n", + " 0.0004575004568323493,\n", + " -0.001210278132930398,\n", + " -0.02081933431327343,\n", + " -0.005646225530654192,\n", + " 0.013928511179983616,\n", + " -0.007426239550113678,\n", + " -0.00886646006256342,\n", + " -0.0024645142257213593,\n", + " 0.026097506284713745,\n", + " 0.00356660527177155,\n", + " 0.011681274510920048,\n", + " 0.02047765627503395,\n", + " -0.0023202800657600164,\n", + " -0.014933145605027676,\n", + " -0.0023335106670856476,\n", + " -0.015454763546586037,\n", + " -0.006096171215176582,\n", + " 0.003595830872654915,\n", + " 0.010202085599303246,\n", + " 0.0098204854875803,\n", + " 0.028708523139357567,\n", + " 0.004216618370264769,\n", + " -0.0037561950739473104,\n", + " -0.010510984808206558,\n", + " 0.025745287537574768,\n", + " -0.01602184772491455,\n", + " -0.008643347769975662,\n", + " 0.03565279394388199,\n", + " -0.00391846289858222,\n", + " 0.0067151449620723724,\n", + " 0.006582110188901424,\n", + " 0.011514297686517239,\n", + " -0.006655575707554817,\n", + " -0.02781674824655056,\n", + " 0.021441366523504257,\n", + " 0.0023280216846615076,\n", + " -0.006080655846744776,\n", + " -0.0015929073560982943,\n", + " 0.0012868221383541822,\n", + " 0.0185436699539423,\n", + " 0.004185759928077459,\n", + " 0.01332230307161808,\n", + " 0.021739855408668518,\n", + " -0.0018754908815026283,\n", + " -0.00912096630781889,\n", + " -0.019020525738596916,\n", + " -0.010130912065505981,\n", + " 0.0012980804312974215,\n", + " -0.02112886682152748,\n", + " 0.01379894558340311,\n", + " -0.019787294790148735,\n", + " -0.017875097692012787,\n", + " -0.016615208238363266,\n", + " 0.013888785615563393,\n", + " -0.006550563499331474,\n", + " 0.012047415599226952,\n", + " -0.005544085055589676,\n", + " 0.00046558587928302586,\n", + " -0.03268171101808548,\n", + " -0.01981344074010849,\n", + " 0.01326016429811716,\n", + " -0.0039311726577579975,\n", + " -0.016159934923052788,\n", + " 0.007912284694612026,\n", + " 0.017360089346766472,\n", + " -0.00917900912463665,\n", + " -0.015222931280732155,\n", + " 0.01181393675506115,\n", + " -0.0036780585069209337,\n", + " 0.0008283006027340889,\n", + " -0.03979955613613129,\n", + " 0.005076248664408922,\n", + " -0.00943879596889019,\n", + " 0.0049046906642615795,\n", + " 0.0503443107008934,\n", + " 0.007274238392710686,\n", + " -0.024708405137062073,\n", + " 0.03180333226919174,\n", + " -0.024111684411764145,\n", + " 0.014323772862553596,\n", + " -0.009170287288725376,\n", + " 0.0014948569005355239,\n", + " 0.014668592251837254,\n", + " 0.009418771602213383,\n", + " 0.024519264698028564,\n", + " -0.0028209055308252573,\n", + " -0.011101690120995045,\n", + " 0.008618107996881008,\n", + " -0.008609072305262089,\n", + " -0.002511126920580864,\n", + " -0.012777554802596569,\n", + " 0.02389429695904255,\n", + " -0.00396793894469738,\n", + " -0.00549497501924634,\n", + " 0.009450403042137623,\n", + " 0.007008947432041168,\n", + " 0.02295737899839878,\n", + " -0.03609155863523483,\n", + " -0.005497496109455824,\n", + " -0.011227840557694435,\n", + " 0.005365726538002491,\n", + " 0.0178525410592556,\n", + " -0.009112250059843063,\n", + " 0.023930715397000313,\n", + " -0.019720004871487617,\n", + " -0.0016240220284089446,\n", + " -0.008786618709564209,\n", + " -0.0031093028374016285,\n", + " 0.0060302517376840115,\n", + " -0.01411086693406105,\n", + " 0.009259095415472984,\n", + " 0.026444211602211,\n", + " -0.012551960535347462,\n", + " 0.0007369715603999794,\n", + " 0.029658250510692596,\n", + " 0.012057793326675892,\n", + " 0.007323889993131161,\n", + " 0.006429325323551893,\n", + " -0.04151007905602455,\n", + " -0.013682323507964611,\n", + " 0.015200085006654263,\n", + " 0.005704157520085573,\n", + " -0.0024766852147877216,\n", + " 0.009210777468979359,\n", + " 0.019238585606217384,\n", + " 0.01412604283541441,\n", + " 0.008283768780529499,\n", + " -0.012045786716043949,\n", + " -0.019151538610458374,\n", + " -0.008146371692419052,\n", + " -0.0003192038566339761,\n", + " -0.013413612730801105,\n", + " -0.005666160024702549,\n", + " 0.02234991453588009,\n", + " -0.017263537272810936,\n", + " -0.0004641618288587779,\n", + " 0.011303545907139778,\n", + " -0.007437041960656643,\n", + " 0.009322786703705788,\n", + " -0.011791775934398174,\n", + " -0.029371924698352814,\n", + " 0.011847944930195808,\n", + " -0.012746450491249561,\n", + " 0.0006847915938124061,\n", + " 0.007335502654314041,\n", + " 0.002275111386552453,\n", + " -0.02154112234711647,\n", + " 0.02269313670694828,\n", + " 0.022147901356220245,\n", + " 0.004866878502070904,\n", + " -0.018933145329356194,\n", + " 0.005171678960323334,\n", + " 0.0040926444344222546,\n", + " 0.014178114011883736,\n", + " -0.00392212113365531,\n", + " 0.008461268618702888,\n", + " -0.007197211030870676,\n", + " 0.031356364488601685,\n", + " 0.006938708946108818,\n", + " 0.004082654602825642,\n", + " 0.01137610711157322,\n", + " 0.0035184617154300213,\n", + " 0.02558928169310093,\n", + " -0.002163136610761285,\n", + " 0.018033865839242935,\n", + " -0.004614625591784716,\n", + " 0.00650483462959528,\n", + " -0.008514108136296272,\n", + " -0.028073208406567574,\n", + " 0.013087261468172073,\n", + " -0.007777229882776737,\n", + " 0.013863285072147846,\n", + " 2.2165347672853386e-06,\n", + " 0.02258075214922428,\n", + " 0.02859872579574585,\n", + " 0.009593948721885681,\n", + " 0.005797175690531731,\n", + " 0.013160321861505508,\n", + " 0.010131489485502243,\n", + " 0.007077783811837435,\n", + " 0.019415685907006264,\n", + " 0.02302323468029499,\n", + " -0.02055949531495571,\n", + " -0.014745713211596012,\n", + " -0.012071777135133743,\n", + " 0.005507923197001219,\n", + " -0.006524238269776106,\n", + " -0.010228286497294903,\n", + " 0.01565360277891159,\n", + " 0.009641028009355068,\n", + " -0.01614265702664852,\n", + " 0.000580347201321274,\n", + " 0.0005364116514101624,\n", + " -0.023418348282575607,\n", + " -0.02351146936416626,\n", + " 0.0065367743372917175,\n", + " 0.011478199623525143,\n", + " 0.0029822385404258966,\n", + " 0.016715558245778084,\n", + " 0.016576580703258514,\n", + " -0.0019007038790732622,\n", + " -0.01610071025788784,\n", + " 0.03106078691780567,\n", + " 0.008976636454463005,\n", + " -0.015568850561976433,\n", + " 0.01831594854593277,\n", + " 0.021407460793852806,\n", + " 0.012749534100294113,\n", + " 0.022004077211022377,\n", + " 0.01768640987575054,\n", + " -0.0031022171024233103,\n", + " 0.003041802207008004,\n", + " 0.005421467125415802,\n", + " -0.013152116909623146,\n", + " 0.014155357144773006,\n", + " -0.0011325232917442918,\n", + " -0.0008713186252862215,\n", + " 0.014029284939169884,\n", + " 0.005906077567487955,\n", + " 0.008186781778931618,\n", + " -0.006220459938049316,\n", + " 0.015955379232764244,\n", + " -0.017231818288564682,\n", + " 0.01309678889811039,\n", + " 0.01892782561480999,\n", + " 0.0074506704695522785,\n", + " 0.00252294703386724,\n", + " 0.021975934505462646,\n", + " -0.008126703090965748,\n", + " 0.029144490137696266,\n", + " -0.01697709411382675,\n", + " 0.005407759919762611,\n", + " 0.007914980873465538,\n", + " 0.016133509576320648,\n", + " -0.002494237618520856,\n", + " 0.020019978284835815,\n", + " -0.005621489603072405,\n", + " 0.020884649828076363,\n", + " -0.022830966860055923,\n", + " 0.003192953998222947,\n", + " 0.009623222053050995,\n", + " -0.016929129138588905,\n", + " 0.008178411982953548,\n", + " -0.006764373742043972,\n", + " 0.011279193684458733,\n", + " -0.013274733908474445,\n", + " 0.0067980908788740635,\n", + " -0.021725470200181007,\n", + " -0.009492350742220879,\n", + " -0.013368367217481136,\n", + " -0.0005820324295200408,\n", + " 0.010973022319376469,\n", + " -0.016382437199354172,\n", + " -0.013291421346366405,\n", + " 0.01631794311106205,\n", + " 0.026708001270890236,\n", + " -0.01604301854968071,\n", + " 0.029547305777668953,\n", + " -0.012892454862594604,\n", + " -0.018933599814772606,\n", + " -0.0046638804487884045,\n", + " 0.0424632728099823,\n", + " 0.005532404873520136,\n", + " -0.00618926202878356,\n", + " 0.01928447186946869,\n", + " -0.016525447368621826,\n", + " 0.005132186226546764,\n", + " 0.017395589500665665,\n", + " 0.010804228484630585,\n", + " -0.01774679683148861,\n", + " -0.03498842939734459,\n", + " -0.009244519285857677,\n", + " 0.002269187942147255,\n", + " -0.017580782994627953,\n", + " 0.03342902287840843,\n", + " 0.019035592675209045,\n", + " -0.010364466346800327,\n", + " -0.0010365818161517382,\n", + " -0.008475861512124538,\n", + " -0.024768078699707985,\n", + " 0.007811828516423702,\n", + " 0.0007224922883324325,\n", + " 0.0053406283259391785,\n", + " 0.015365003608167171,\n", + " 0.014544358476996422,\n", + " 0.006721693091094494,\n", + " -0.0053669000044465065,\n", + " -0.0061641717329621315,\n", + " 0.0167725570499897,\n", + " -0.012045960873365402,\n", + " -0.017861204221844673,\n", + " -0.002282701665535569,\n", + " -0.01277306117117405,\n", + " -0.026085669174790382,\n", + " 0.02142571657896042,\n", + " 0.01169880572706461,\n", + " 0.00661891745403409,\n", + " -0.008942786604166031,\n", + " -0.0005775789613835514,\n", + " 0.017732907086610794,\n", + " 1.2999666068935767e-05,\n", + " 0.01615849696099758,\n", + " 0.03065437451004982,\n", + " -0.00019303745648358017,\n", + " 0.024879885837435722,\n", + " 0.009697318077087402,\n", + " 0.003906070487573743,\n", + " -0.001108623924665153,\n", + " 0.010587952099740505,\n", + " -0.015321311540901661,\n", + " 0.014482120983302593,\n", + " -0.014630504883825779,\n", + " 0.008109631016850471,\n", + " 0.013947028666734695,\n", + " 0.020127564668655396,\n", + " -0.02681734412908554,\n", + " -0.001262568635866046,\n", + " -0.02351762354373932,\n", + " -0.0034904133062809706,\n", + " -0.025115966796875,\n", + " 0.00041233477531932294,\n", + " -0.03210841864347458,\n", + " -0.014403645880520344,\n", + " 0.01508869044482708,\n", + " -0.01426045224070549,\n", + " 0.017466282472014427,\n", + " 0.005857695359736681,\n", + " -0.0013472529826685786,\n", + " -0.002424640581011772,\n", + " -0.0014821934746578336,\n", + " -0.017711561173200607,\n", + " 0.020194660872220993,\n", + " 0.007711687125265598,\n", + " -0.006724135018885136,\n", + " -0.01219252496957779,\n", + " -0.002240788424387574,\n", + " -0.017092730849981308,\n", + " -0.013157549314200878,\n", + " -0.004683325998485088,\n", + " -0.006799815222620964,\n", + " 0.0013616927899420261,\n", + " 0.003650276456028223,\n", + " 0.004854041151702404,\n", + " 0.014137422665953636,\n", + " 0.015527388080954552,\n", + " -0.03160852566361427,\n", + " 0.0007112329476512969,\n", + " -0.002946733497083187,\n", + " -0.021824302151799202,\n", + " 0.006391474977135658,\n", + " -0.03130871802568436,\n", + " 0.002444390906020999,\n", + " -0.02205747179687023,\n", + " -0.0009384482982568443,\n", + " 0.0037650992162525654,\n", + " -0.005415714345872402,\n", + " 0.0182612556964159,\n", + " -0.006117376498878002,\n", + " -0.01413779053837061,\n", + " -0.014110713265836239,\n", + " -0.0016754124080762267,\n", + " -0.0027341260574758053,\n", + " -0.017401142045855522,\n", + " -0.014090651646256447,\n", + " -0.006296559236943722,\n", + " 0.011119811795651913,\n", + " -0.013338878750801086,\n", + " 0.022201355546712875,\n", + " -0.008421794511377811,\n", + " -0.024969641119241714,\n", + " 0.016300074756145477,\n", + " 0.00221728952601552,\n", + " -0.025288395583629608,\n", + " -0.024768929928541183,\n", + " -0.005367298610508442,\n", + " -0.011850270442664623,\n", + " -7.055165769997984e-05,\n", + " -0.02498014271259308,\n", + " 0.002521191257983446,\n", + " -0.0005549240158870816,\n", + " -0.002553754486143589,\n", + " 0.01495042908936739,\n", + " -0.0168534517288208,\n", + " 0.01468364056199789,\n", + " -0.0002745247620623559,\n", + " -0.0012332743499428034,\n", + " 0.02281203493475914,\n", + " -0.0019585280679166317,\n", + " 0.0025182447861880064,\n", + " 0.007781229913234711,\n", + " -0.009566482156515121,\n", + " -0.013032464310526848,\n", + " -0.03374152258038521,\n", + " -0.007732870988547802,\n", + " -0.005964191630482674,\n", + " -0.027642998844385147,\n", + " -0.002493371721357107,\n", + " 0.013606597669422626,\n", + " 0.0027858021203428507,\n", + " -0.004969800356775522,\n", + " -0.008887036703526974,\n", + " 0.017043963074684143,\n", + " -0.01029882486909628,\n", + " -0.00596567802131176,\n", + " -0.0030601369217038155,\n", + " -0.0038627428002655506,\n", + " 0.004196135327219963,\n", + " -0.02537938579916954,\n", + " -0.011517830193042755,\n", + " 0.003922145813703537,\n", + " 0.024173494428396225,\n", + " 0.007839345373213291,\n", + " 0.018174149096012115,\n", + " 0.01833866909146309,\n", + " 0.007239053025841713,\n", + " 0.006254516541957855,\n", + " 0.017041588202118874,\n", + " 0.05501232296228409,\n", + " 0.006659498438239098,\n", + " -0.03173157200217247,\n", + " 0.011870153248310089,\n", + " -0.044423483312129974,\n", + " 0.00765900406986475,\n", + " 0.003303903853520751,\n", + " -0.00989844836294651,\n", + " -0.00102717406116426,\n", + " 0.010751670226454735,\n", + " -0.01436996553093195,\n", + " 0.0007458398467861116,\n", + " -0.02406933903694153,\n", + " 0.013927231542766094,\n", + " -0.0023855960462242365,\n", + " -0.018460353836417198,\n", + " -0.013494566082954407,\n", + " -0.024894949048757553,\n", + " 0.0027491513174027205,\n", + " 0.01960483193397522,\n", + " 0.0020772041752934456,\n", + " 0.02088438905775547,\n", + " -0.007962409406900406,\n", + " 0.01874588616192341,\n", + " -0.0119165675714612,\n", + " 0.006801045034080744,\n", + " 0.005523370113223791,\n", + " 0.005721281748265028,\n", + " 8.281860937131569e-05,\n", + " 0.022861666977405548,\n", + " 0.031650010496377945,\n", + " 0.011051682755351067,\n", + " 0.014575383625924587,\n", + " -0.008896112442016602,\n", + " -0.0064266943372786045,\n", + " -0.008789743296802044,\n", + " -0.005537368822842836,\n", + " -0.029184775426983833,\n", + " -0.012891268357634544,\n", + " 0.008750290609896183,\n", + " -0.013342045247554779,\n", + " -0.018940439447760582,\n", + " -0.010383781976997852,\n", + " 0.009893164038658142,\n", + " 0.00484957080334425,\n", + " -0.003208030480891466,\n", + " 0.002685114974156022,\n", + " 0.02932116575539112,\n", + " -0.005980887915939093,\n", + " -0.02094399183988571,\n", + " 0.0011950458865612745,\n", + " -0.0013160411035642028,\n", + " -0.015973364934325218,\n", + " 0.006585970055311918,\n", + " -0.013596748933196068,\n", + " -0.014491614885628223,\n", + " -0.002483466174453497,\n", + " -0.015564654022455215,\n", + " -0.004617113154381514,\n", + " 0.005632814951241016,\n", + " 0.013269959948956966,\n", + " -0.0102331368252635,\n", + " -0.01374089252203703,\n", + " 0.010636764578521252,\n", + " -0.00011052726040361449,\n", + " -0.020722508430480957,\n", + " -0.00012687862908933312,\n", + " -0.00044137012446299195,\n", + " -0.002424860605970025,\n", + " 0.031966038048267365,\n", + " -0.02460266463458538,\n", + " 0.0014620558358728886,\n", + " -0.005570637993514538,\n", + " -0.017171526327729225,\n", + " -0.004151195287704468,\n", + " -0.00979167316108942,\n", + " 0.013350186869502068,\n", + " -0.03380487486720085,\n", + " 0.004512457642704248,\n", + " -0.030104100704193115,\n", + " 0.00020586112805176526,\n", + " -0.004360636696219444,\n", + " 0.024787265807390213,\n", + " -0.021622182801365852,\n", + " -0.013142443262040615,\n", + " -0.008689089678227901,\n", + " -0.019221695140004158,\n", + " 0.015511195175349712,\n", + " 0.004761400632560253,\n", + " -0.018051810562610626,\n", + " 0.0030495638493448496,\n", + " 0.013037407770752907,\n", + " 0.018515795469284058,\n", + " 0.030628709122538567,\n", + " -0.008378121070563793,\n", + " 0.005477331578731537,\n", + " 0.030206406489014626,\n", + " -0.018550679087638855,\n", + " -0.005074893124401569,\n", + " 0.018194109201431274,\n", + " -0.022404147312045097,\n", + " 0.005452401004731655,\n", + " -0.0061740027740597725,\n", + " 0.007163482252508402,\n", + " -0.007498984690755606,\n", + " 0.0013850930845364928,\n", + " 0.019100110977888107,\n", + " -0.00539770070463419,\n", + " -0.02813248336315155,\n", + " 0.021426543593406677,\n", + " -0.0020243236795067787,\n", + " -0.012561444193124771,\n", + " 0.005466975271701813,\n", + " -0.0004141190438531339,\n", + " 0.008710913360118866,\n", + " -0.01259232871234417,\n", + " 0.02724912390112877,\n", + " 0.014795316383242607,\n", + " 0.0017043438274413347,\n", + " 0.03569337725639343,\n", + " 0.009455371648073196,\n", + " -0.008252507075667381,\n", + " 0.034219030290842056,\n", + " -0.003471348201856017,\n", + " -0.005572606343775988,\n", + " 0.002426962135359645,\n", + " 0.006176020484417677,\n", + " -0.02644067071378231,\n", + " -0.0015432301443070173,\n", + " 0.01251029409468174,\n", + " 0.006000349763780832,\n", + " 0.012471841648221016,\n", + " -0.001398047199472785,\n", + " -0.013531356118619442,\n", + " -0.01039454061537981,\n", + " -0.004671303555369377,\n", + " 0.00626105023548007,\n", + " -0.0019008438102900982,\n", + " 0.020720865577459335,\n", + " 0.012591890059411526,\n", + " -0.0053941598162055016,\n", + " -0.025267941877245903,\n", + " 0.005296881310641766,\n", + " 0.0342840850353241,\n", + " -0.01581035926938057,\n", + " 0.004621365573257208,\n", + " 0.0030632903799414635,\n", + " 0.007074137218296528,\n", + " -0.005330575164407492,\n", + " -0.0030899883713573217,\n", + " 0.016070717945694923,\n", + " -0.045663513243198395,\n", + " -0.0010349617805331945,\n", + " -0.007994215004146099,\n", + " -0.017588473856449127,\n", + " -0.014046519063413143,\n", + " -0.0028416865970939398,\n", + " -0.00362231838516891,\n", + " -0.0026648773346096277,\n", + " 0.006982769817113876,\n", + " 0.006077419500797987,\n", + " -0.012517980299890041,\n", + " 0.016320543363690376,\n", + " 0.006708477158099413,\n", + " -0.02435096725821495,\n", + " 0.020286191254854202,\n", + " -0.001916136359795928,\n", + " -0.020461106672883034,\n", + " 0.03223827853798866,\n", + " -0.008052353747189045,\n", + " 0.03137693554162979,\n", + " 0.0007936311303637922,\n", + " 0.026611249893903732,\n", + " -0.013749106787145138,\n", + " -0.005045521073043346,\n", + " 0.01802709884941578,\n", + " 0.004193250089883804,\n", + " -0.0074610221199691296,\n", + " 0.012689094990491867,\n", + " -0.001128576579503715,\n", + " -0.008252380415797234,\n", + " -0.008191979490220547,\n", + " -0.008434522897005081,\n", + " -0.02567083016037941,\n", + " -0.006246744189411402,\n", + " -0.024753373116254807,\n", + " 0.005886504426598549,\n", + " -0.0030029790941625834,\n", + " 0.011522923596203327,\n", + " 0.0011658172588795424,\n", + " 0.00444172415882349,\n", + " 0.03330754488706589,\n", + " -0.028662286698818207,\n", + " -0.0243659857660532,\n", + " -0.016821498051285744,\n", + " 0.018770718947052956,\n", + " 0.01755281165242195,\n", + " 0.015005288645625114,\n", + " -0.0038322769105434418,\n", + " 0.016096081584692,\n", + " 0.005756937898695469,\n", + " 0.004192751832306385,\n", + " 0.01487874798476696,\n", + " -0.018225383013486862,\n", + " 0.00040869449730962515,\n", + " -0.009901725687086582,\n", + " 0.011486656963825226,\n", + " 0.022721173241734505,\n", + " 0.008551487699151039,\n", + " -0.006110388319939375,\n", + " 0.027253510430455208,\n", + " 0.025853939354419708,\n", + " -0.011822552420198917,\n", + " 0.011195230297744274,\n", + " 0.023045159876346588,\n", + " 0.0054076313972473145,\n", + " -0.0376087948679924,\n", + " -0.012947173789143562,\n", + " -0.01948842778801918,\n", + " -0.006805140990763903,\n", + " -0.016297485679388046,\n", + " 0.01277123112231493,\n", + " 0.005486239679157734,\n", + " 0.013064263388514519,\n", + " -0.01799067109823227,\n", + " -0.000999069889076054,\n", + " 0.0032741266768425703,\n", + " -0.004913169424980879,\n", + " 0.010930745862424374,\n", + " -0.0022265056613832712,\n", + " -0.007856646552681923,\n", + " 0.024474594742059708,\n", + " -0.01740814931690693,\n", + " -0.0058359322138130665,\n", + " -0.0076317558996379375,\n", + " 0.02482902817428112,\n", + " -0.0038906049448996782,\n", + " 0.009218372404575348,\n", + " 0.011257494799792767,\n", + " 0.02811446040868759,\n", + " 0.01012449711561203,\n", + " -0.009031664580106735,\n", + " -0.010511829517781734,\n", + " 0.03654777631163597,\n", + " 0.0030149882659316063,\n", + " 0.022236613556742668,\n", + " -0.011791135184466839,\n", + " -7.580777310067788e-05,\n", + " 0.00784097146242857,\n", + " -0.0025190457236021757,\n", + " -0.0004561890091281384,\n", + " -0.01860455982387066,\n", + " 0.0008333594887517393,\n", + " -0.002219945890828967,\n", + " 0.02410193160176277,\n", + " -0.006336560007184744,\n", + " 0.013507379218935966,\n", + " 0.01625504530966282,\n", + " -0.005512222182005644,\n", + " 0.017335521057248116,\n", + " 0.001445610774680972,\n", + " -0.014676893129944801,\n", + " -0.01950543373823166,\n", + " 0.027771327644586563,\n", + " 0.010210845619440079,\n", + " -0.003559559816494584,\n", + " 0.0018264109967276454,\n", + " 0.0008935378864407539,\n", + " 0.0026427831035107374,\n", + " 0.01573711261153221,\n", + " 0.0014196783304214478,\n", + " 0.014842817559838295,\n", + " -0.0027134984266012907,\n", + " 0.0011339110787957907,\n", + " -0.002446472179144621,\n", + " -0.03947463259100914,\n", + " -0.012350163422524929,\n", + " -0.0068352906964719296,\n", + " 0.016724968329072,\n", + " 0.02971581369638443,\n", + " -0.0023575620725750923,\n", + " -0.0028808927163481712,\n", + " 0.0055499328300356865,\n", + " -0.024555519223213196,\n", + " 0.008399837650358677,\n", + " -0.013832250609993935,\n", + " -0.010051798075437546,\n", + " 0.0062475660815835,\n", + " 0.010128488764166832,\n", + " -0.03516209498047829,\n", + " 0.016856608912348747,\n", + " -0.01280664186924696,\n", + " -0.008145435713231564,\n", + " -0.013778863474726677,\n", + " -0.007605956867337227,\n", + " -0.0023700245656073093,\n", + " -0.02099779061973095,\n", + " -0.00743044214323163,\n", + " -0.02712254971265793,\n", + " 0.029353691264986992,\n", + " 0.005820101127028465,\n", + " 0.012708257883787155,\n", + " -0.004160662181675434,\n", + " -0.02543794736266136,\n", + " 0.002900070045143366,\n", + " 0.007988318800926208,\n", + " -0.007849618792533875,\n", + " 0.00019223698473069817,\n", + " -0.0029571824707090855,\n", + " 0.0017812871374189854,\n", + " -0.0067518725991249084,\n", + " -0.010918932035565376,\n", + " -0.0021185216028243303,\n", + " -0.01898864097893238,\n", + " -0.014883413910865784,\n", + " -0.024012362584471703,\n", + " ...]" + ] + }, + "execution_count": 5, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "query_result" + ] + }, + { + "cell_type": "code", + "execution_count": 6, + "id": "98897454-b280-4ee1-bbb9-2c6c15342f87", + "metadata": { + "ExecuteTime": { + "end_time": "2023-05-24T15:13:18.605339Z", + "start_time": "2023-05-24T15:13:17.845906Z" + }, + "execution": { + "iopub.execute_input": "2024-03-29T15:39:28.164009Z", + "iopub.status.busy": "2024-03-29T15:39:28.161759Z", + "iopub.status.idle": "2024-03-29T15:39:30.217232Z", + "shell.execute_reply": "2024-03-29T15:39:30.215348Z", + "shell.execute_reply.started": "2024-03-29T15:39:28.163876Z" + }, + "scrolled": true + }, + "outputs": [], + "source": [ + "document_text = \"This is a test document.\"\n", + "document_result = embeddings.embed_documents([document_text])" + ] + }, + { + "cell_type": "code", + "execution_count": 7, + "id": "21e03cca-bdb2-49e4-95d7-105cf6a5874d", + "metadata": { + "collapsed": true, + "execution": { + "iopub.execute_input": "2024-03-29T15:39:32.330215Z", + "iopub.status.busy": "2024-03-29T15:39:32.328926Z", + "iopub.status.idle": "2024-03-29T15:39:32.356001Z", + "shell.execute_reply": "2024-03-29T15:39:32.355284Z", + "shell.execute_reply.started": "2024-03-29T15:39:32.330135Z" + }, + "jupyter": { + "outputs_hidden": true + } + }, + "outputs": [ + { + "data": { + "text/plain": [ + "[[-0.019484492018818855,\n", + " 0.0004918322083540261,\n", + " -0.007027746178209782,\n", + " -0.012673289515078068,\n", + " -0.005353343673050404,\n", + " -0.03189416974782944,\n", + " -0.027227548882365227,\n", + " 0.0009138379828073084,\n", + " -0.0017150233034044504,\n", + " -0.028936535120010376,\n", + " -0.003939046058803797,\n", + " -0.026341330260038376,\n", + " 0.008856299333274364,\n", + " -0.013755874708294868,\n", + " 0.016992073506116867,\n", + " -0.0032008232083171606,\n", + " -0.008546354249119759,\n", + " 0.018413474783301353,\n", + " -0.004322977736592293,\n", + " -0.0033296330366283655,\n", + " 0.014928839169442654,\n", + " 0.00902748666703701,\n", + " 0.0033201989717781544,\n", + " 0.01944599114358425,\n", + " -0.004280788823962212,\n", + " 0.026516154408454895,\n", + " -0.0004448844993021339,\n", + " 0.005788407754153013,\n", + " 0.004848824813961983,\n", + " -0.010850796476006508,\n", + " 0.0074156359769403934,\n", + " 0.0028794733807444572,\n", + " -0.005040694493800402,\n", + " -0.009943140670657158,\n", + " -0.0014771600253880024,\n", + " 0.02702862024307251,\n", + " 0.012307064607739449,\n", + " 0.031931404024362564,\n", + " -0.0077228182926774025,\n", + " -0.027956398203969002,\n", + " 0.017846351489424706,\n", + " 0.01735025644302368,\n", + " -0.024033349007368088,\n", + " -0.009735107421875,\n", + " 0.01633143052458763,\n", + " 0.010355479083955288,\n", + " -0.0019731861539185047,\n", + " -0.0035277868155390024,\n", + " 0.020027706399559975,\n", + " -0.04373868554830551,\n", + " 0.0354132205247879,\n", + " -0.001807031687349081,\n", + " -0.002918412210419774,\n", + " 0.09777230769395828,\n", + " 0.015062221325933933,\n", + " 0.004985701758414507,\n", + " -0.018884792923927307,\n", + " 0.010831230320036411,\n", + " -0.008481908589601517,\n", + " -0.003799594473093748,\n", + " 0.0004316098056733608,\n", + " -0.019112855195999146,\n", + " 0.014202145859599113,\n", + " 0.011331802234053612,\n", + " -0.007499997038394213,\n", + " -0.007928249426186085,\n", + " -0.017887219786643982,\n", + " -0.03139348700642586,\n", + " -0.01899610459804535,\n", + " 0.0013776234118267894,\n", + " -0.008819176815450191,\n", + " 0.03704814240336418,\n", + " -0.022388434037566185,\n", + " 0.0156440120190382,\n", + " -0.017101433128118515,\n", + " -0.013089085929095745,\n", + " 0.02561251074075699,\n", + " 0.029660305008292198,\n", + " -0.007624164689332247,\n", + " -0.006829928606748581,\n", + " 0.026884594932198524,\n", + " -0.027975428849458694,\n", + " -0.010908747091889381,\n", + " 0.007614677771925926,\n", + " -0.0005592447123490274,\n", + " 0.032569821923971176,\n", + " 0.009540022350847721,\n", + " -0.0028657703660428524,\n", + " 0.01721801981329918,\n", + " -0.010194444097578526,\n", + " -0.01614883914589882,\n", + " -0.009784751571714878,\n", + " 0.002176648238673806,\n", + " -0.019132796674966812,\n", + " -0.01863129623234272,\n", + " 0.006362563464790583,\n", + " -0.003694645594805479,\n", + " -0.024288378655910492,\n", + " -0.022363897413015366,\n", + " 0.000744891818612814,\n", + " 0.006259715650230646,\n", + " -0.019178815186023712,\n", + " 0.011478138156235218,\n", + " -1.5152631931414362e-05,\n", + " 0.017119858413934708,\n", + " 0.0019262970890849829,\n", + " -0.007264178246259689,\n", + " 0.0020848813001066446,\n", + " 0.012564039789140224,\n", + " -0.015715299174189568,\n", + " 0.0036766608245670795,\n", + " 0.007963340729475021,\n", + " -0.03583437204360962,\n", + " 0.0167242344468832,\n", + " -0.004585846792906523,\n", + " -0.02034004032611847,\n", + " -0.008786039426922798,\n", + " -0.019419966265559196,\n", + " 0.003955639433115721,\n", + " -0.02039165608584881,\n", + " 0.007168842479586601,\n", + " 0.024760562926530838,\n", + " -0.010934860445559025,\n", + " 0.003110958728939295,\n", + " -0.0054563055746257305,\n", + " -0.01438088808208704,\n", + " -0.0013200901448726654,\n", + " 0.010165776126086712,\n", + " -0.002869517309591174,\n", + " -0.006820392794907093,\n", + " -0.006658782716840506,\n", + " 0.004162106662988663,\n", + " 0.020661450922489166,\n", + " -0.02874227613210678,\n", + " -0.014118155464529991,\n", + " 0.003224856685847044,\n", + " -0.014369030483067036,\n", + " 0.004771883133798838,\n", + " 0.012497876770794392,\n", + " -0.018075305968523026,\n", + " -0.028585655614733696,\n", + " 0.015290608629584312,\n", + " -0.00422133831307292,\n", + " 0.0003679264336824417,\n", + " -0.004252501763403416,\n", + " -0.011410473845899105,\n", + " 0.002173950197175145,\n", + " -0.01132588367909193,\n", + " -0.017615757882595062,\n", + " -0.005467323586344719,\n", + " -0.022641275078058243,\n", + " -0.030672792345285416,\n", + " 0.020841708406805992,\n", + " -0.004163825884461403,\n", + " -0.003081672824919224,\n", + " 0.04334355145692825,\n", + " 0.016485434025526047,\n", + " 0.02830098755657673,\n", + " 0.014162690378725529,\n", + " 0.003305956721305847,\n", + " -0.01558461133390665,\n", + " -0.028950272127985954,\n", + " 0.0017209401121363044,\n", + " 0.016394009813666344,\n", + " -0.014193333685398102,\n", + " 0.032301925122737885,\n", + " -0.008469345979392529,\n", + " -0.018771948292851448,\n", + " 0.007705388590693474,\n", + " 0.0048446026630699635,\n", + " 0.00827891007065773,\n", + " -0.024297840893268585,\n", + " -0.015459121204912663,\n", + " -0.004894130397588015,\n", + " 0.019174423068761826,\n", + " 0.003726472845301032,\n", + " -0.0069329096004366875,\n", + " -0.005054902285337448,\n", + " -0.01115730032324791,\n", + " 0.0011553125223144889,\n", + " -0.013363232836127281,\n", + " 0.012810817919671535,\n", + " 0.0065435804426670074,\n", + " -0.019356241449713707,\n", + " 0.0038377989549189806,\n", + " -0.0059433975256979465,\n", + " 0.01719961129128933,\n", + " 0.01027001440525055,\n", + " 0.023838665336370468,\n", + " -0.017898323014378548,\n", + " 0.0275045745074749,\n", + " -0.01586216874420643,\n", + " -0.017809314653277397,\n", + " -0.01702960953116417,\n", + " -0.00023454823531210423,\n", + " -0.023614460602402687,\n", + " -0.02317613735795021,\n", + " 0.004228908568620682,\n", + " -0.010320615954697132,\n", + " 0.012252600863575935,\n", + " 0.01613335683941841,\n", + " 0.00556036876514554,\n", + " -0.024423038586974144,\n", + " -0.00248654349707067,\n", + " -0.0052187684923410416,\n", + " -0.02748170867562294,\n", + " 0.000613022071775049,\n", + " -0.010094189085066319,\n", + " -0.0061216638423502445,\n", + " 0.01032200176268816,\n", + " 0.005635530222207308,\n", + " 0.01639268361032009,\n", + " 0.020736921578645706,\n", + " -0.016877925023436546,\n", + " -0.021583687514066696,\n", + " -0.000881461426615715,\n", + " -0.000917142431717366,\n", + " 0.025361627340316772,\n", + " -0.017409449443221092,\n", + " -0.0007481586071662605,\n", + " -0.006518878508359194,\n", + " -0.014359765686094761,\n", + " 0.009346549399197102,\n", + " 0.0006721566896885633,\n", + " 0.002496484899893403,\n", + " 0.012045742943882942,\n", + " 0.0023702955804765224,\n", + " 0.009324215352535248,\n", + " -0.00405908515676856,\n", + " 0.010660269297659397,\n", + " 0.00604375870898366,\n", + " 0.02218792587518692,\n", + " -0.0003027633356396109,\n", + " 0.023658229038119316,\n", + " 0.0015295293414965272,\n", + " -0.009180267341434956,\n", + " -0.013470915146172047,\n", + " -0.00011685601202771068,\n", + " 0.019391989335417747,\n", + " -0.0016365452902391553,\n", + " 0.016382677480578423,\n", + " -0.0025949093978852034,\n", + " -0.01129817683249712,\n", + " -0.028478750959038734,\n", + " -0.011386929079890251,\n", + " 0.0024167357478290796,\n", + " -0.015677297487854958,\n", + " 0.0006413079099729657,\n", + " 0.008419468067586422,\n", + " 0.002269485266879201,\n", + " -0.010327519848942757,\n", + " -0.04196741059422493,\n", + " -0.0024877903051674366,\n", + " -0.009378228336572647,\n", + " 0.01839737594127655,\n", + " -0.01404246874153614,\n", + " -0.0018654247978702188,\n", + " 0.01985299400985241,\n", + " -0.01309738215059042,\n", + " -0.012849090620875359,\n", + " -0.018644336611032486,\n", + " 0.01661038212478161,\n", + " -0.018413694575428963,\n", + " -0.012359190732240677,\n", + " -0.002676716772839427,\n", + " -0.004197251051664352,\n", + " 0.0035521811805665493,\n", + " 0.007935849018394947,\n", + " 0.010034419596195221,\n", + " -0.025826072320342064,\n", + " -0.005588399711996317,\n", + " 0.0067875268869102,\n", + " -0.007897238247096539,\n", + " -0.0012126719811931252,\n", + " -0.007319039199501276,\n", + " 0.013140472583472729,\n", + " -0.013658048585057259,\n", + " 0.016172612085938454,\n", + " 0.031625062227249146,\n", + " -0.0027903085574507713,\n", + " -0.009913383983075619,\n", + " -0.011814743280410767,\n", + " -0.013551912270486355,\n", + " -0.00040318811079487205,\n", + " -0.004645766690373421,\n", + " 0.018931986764073372,\n", + " -0.006715825293213129,\n", + " 0.0345010980963707,\n", + " 0.009808865375816822,\n", + " 0.00031219382071867585,\n", + " -0.021361790597438812,\n", + " 0.029589565470814705,\n", + " -0.019545778632164,\n", + " -0.006839600391685963,\n", + " 0.03414703160524368,\n", + " 0.003162563545629382,\n", + " -0.01362021267414093,\n", + " 0.011285877786576748,\n", + " 0.0028935351874679327,\n", + " -0.005350036080926657,\n", + " -0.02735786698758602,\n", + " 0.02172314189374447,\n", + " 0.005949749611318111,\n", + " -0.0007144561968743801,\n", + " -0.013414089567959309,\n", + " -0.007161424029618502,\n", + " 0.024019431322813034,\n", + " 0.004262072965502739,\n", + " 0.002016711048781872,\n", + " 0.0222645066678524,\n", + " -0.012368962168693542,\n", + " -0.008090445771813393,\n", + " -0.007152413949370384,\n", + " 0.004305841866880655,\n", + " -0.0049229636788368225,\n", + " -0.01076631247997284,\n", + " 0.01656140387058258,\n", + " -0.03583301976323128,\n", + " -0.01484199520200491,\n", + " -0.018741128966212273,\n", + " -0.002573228208348155,\n", + " -0.004580455832183361,\n", + " -0.003019571304321289,\n", + " -0.010984795168042183,\n", + " 0.002048774156719446,\n", + " -0.025104226544499397,\n", + " -0.02455284260213375,\n", + " 7.540378283010796e-05,\n", + " -0.012761498801410198,\n", + " -0.013445761054754257,\n", + " 0.0035847313702106476,\n", + " 0.0231394711881876,\n", + " -0.02027887850999832,\n", + " -0.013337776996195316,\n", + " 0.00901948381215334,\n", + " -0.003112646285444498,\n", + " 0.01194683089852333,\n", + " -0.03696063160896301,\n", + " 0.014971568249166012,\n", + " -0.016337668523192406,\n", + " 0.015908148139715195,\n", + " 0.04104166850447655,\n", + " 0.004572720266878605,\n", + " -0.021547675132751465,\n", + " 0.03474141284823418,\n", + " -0.017567714676260948,\n", + " 0.014558297581970692,\n", + " -0.0008156535332091153,\n", + " 0.003627184545621276,\n", + " 0.021257365122437477,\n", + " 0.01536672841757536,\n", + " 0.016293726861476898,\n", + " 0.0008670052629895508,\n", + " -0.00728483684360981,\n", + " 0.01691974140703678,\n", + " -0.014672094956040382,\n", + " -0.0008179476717486978,\n", + " -0.018543900921940804,\n", + " 0.0226394385099411,\n", + " -0.0002712066634558141,\n", + " 0.00036770993028767407,\n", + " 0.00850330013781786,\n", + " 0.006761811673641205,\n", + " 0.031168123707175255,\n", + " -0.03146185725927353,\n", + " -0.001735692610964179,\n", + " -0.013010626658797264,\n", + " 0.00505995936691761,\n", + " 0.019633151590824127,\n", + " 0.0012399450642988086,\n", + " 0.029671084135770798,\n", + " -0.02056892216205597,\n", + " 0.0035886557307094336,\n", + " -0.002683571306988597,\n", + " 0.0002559150743763894,\n", + " 0.008231519721448421,\n", + " -0.01546843908727169,\n", + " 0.015084458515048027,\n", + " 0.0261235274374485,\n", + " 0.010675269179046154,\n", + " 0.00859019160270691,\n", + " 0.01880238577723503,\n", + " 0.012341131456196308,\n", + " 0.00215032952837646,\n", + " 0.010820840485394001,\n", + " -0.037973176687955856,\n", + " -0.015073548071086407,\n", + " 0.005285357125103474,\n", + " -0.0039015556685626507,\n", + " -0.012085077352821827,\n", + " 0.008736337535083294,\n", + " 0.003232941497117281,\n", + " 0.0007238306570798159,\n", + " 0.007120898459106684,\n", + " 0.004377692937850952,\n", + " -0.012878673151135445,\n", + " -0.004737012088298798,\n", + " 0.0016103372909128666,\n", + " -0.014453768730163574,\n", + " -0.0030761680100113153,\n", + " 0.024939827620983124,\n", + " -0.009631255641579628,\n", + " 0.0015462863957509398,\n", + " 0.018152868375182152,\n", + " 0.002558876993134618,\n", + " 0.013886932283639908,\n", + " -0.010613802820444107,\n", + " -0.011718024499714375,\n", + " 0.01970844343304634,\n", + " -0.025368008762598038,\n", + " 0.004451524466276169,\n", + " 0.0026539869140833616,\n", + " -0.00317376758903265,\n", + " -0.004587087314575911,\n", + " 0.02286575548350811,\n", + " 0.026008864864706993,\n", + " 0.013202764093875885,\n", + " -0.016171438619494438,\n", + " -0.009343815967440605,\n", + " 0.002988232532516122,\n", + " 0.015619875863194466,\n", + " 0.0038960971869528294,\n", + " 0.0048093171790242195,\n", + " 0.011655006557703018,\n", + " 0.03504527732729912,\n", + " -0.0006444973987527192,\n", + " 0.014385323040187359,\n", + " 0.011684667319059372,\n", + " 0.0051994482055306435,\n", + " 0.006360795348882675,\n", + " -0.005261885933578014,\n", + " 0.01097958255559206,\n", + " -0.0075597199611365795,\n", + " 0.001088718301616609,\n", + " -0.008491522632539272,\n", + " -0.022506099194288254,\n", + " 0.002214604988694191,\n", + " 0.0016500533092767,\n", + " 0.002922724699601531,\n", + " -0.015052741393446922,\n", + " -0.005067442078143358,\n", + " 0.026262778788805008,\n", + " 0.002882997505366802,\n", + " 0.008469714783132076,\n", + " 0.0009098969167098403,\n", + " 0.0007244800799526274,\n", + " 0.011361891403794289,\n", + " 0.008085162378847599,\n", + " 0.01785528101027012,\n", + " -0.021736353635787964,\n", + " -0.014902740716934204,\n", + " -0.02387191355228424,\n", + " 0.01154129859060049,\n", + " -0.008052042685449123,\n", + " -0.01643543504178524,\n", + " 0.016863014549016953,\n", + " -0.0014375959290191531,\n", + " -0.010861627757549286,\n", + " -0.005060057621449232,\n", + " 0.004441055003553629,\n", + " -0.02616089954972267,\n", + " -0.017412282526493073,\n", + " 0.005458134692162275,\n", + " 0.012355134822428226,\n", + " 0.003947863355278969,\n", + " 0.016718722879886627,\n", + " 0.0049648103304207325,\n", + " 0.006712459027767181,\n", + " -0.01303650438785553,\n", + " 0.024115873500704765,\n", + " -0.00809017475694418,\n", + " -0.027580678462982178,\n", + " 0.014839811250567436,\n", + " 0.0116657679900527,\n", + " 0.006128309294581413,\n", + " 0.03048730455338955,\n", + " 0.0058337547816336155,\n", + " 0.006805578246712685,\n", + " -0.0014874201733618975,\n", + " 0.001879621879197657,\n", + " -0.015665048733353615,\n", + " 0.017865389585494995,\n", + " 0.011625503189861774,\n", + " 0.009321278892457485,\n", + " 0.013675824739038944,\n", + " 0.01227673888206482,\n", + " 0.0006669477443210781,\n", + " -0.0032042409293353558,\n", + " 0.010426733642816544,\n", + " 0.0017667359206825495,\n", + " 0.0029695217963308096,\n", + " 0.013515078462660313,\n", + " 0.00724818417802453,\n", + " -0.009386356920003891,\n", + " 0.01737366057932377,\n", + " -0.006175730377435684,\n", + " 0.025559378787875175,\n", + " -0.013050810433924198,\n", + " -0.014836403541266918,\n", + " 0.013735868968069553,\n", + " 0.029224025085568428,\n", + " -0.0019481983035802841,\n", + " 0.018222419545054436,\n", + " -0.007173576392233372,\n", + " 0.012109430506825447,\n", + " -0.019521046429872513,\n", + " 0.009070102125406265,\n", + " 0.008546192198991776,\n", + " 0.007099777925759554,\n", + " 0.011943133547902107,\n", + " -0.02416291832923889,\n", + " 0.007409253157675266,\n", + " -0.015731152147054672,\n", + " 0.005225952249020338,\n", + " -0.01997862383723259,\n", + " -0.021982494741678238,\n", + " -0.02488778717815876,\n", + " 0.0017780216876417398,\n", + " -0.0012331722537055612,\n", + " -0.006630309857428074,\n", + " -0.015080750919878483,\n", + " 0.007971370592713356,\n", + " 0.018193203955888748,\n", + " -0.01859109289944172,\n", + " 0.01914096623659134,\n", + " -0.020169110968708992,\n", + " -0.02489267662167549,\n", + " -0.02323361672461033,\n", + " 0.04145375266671181,\n", + " 0.028890211135149002,\n", + " -0.007760887034237385,\n", + " 0.0045552244409918785,\n", + " -0.0176457017660141,\n", + " -0.008273054845631123,\n", + " 0.012306966818869114,\n", + " -0.0031461024191230536,\n", + " -0.020325353369116783,\n", + " -0.0398121140897274,\n", + " -0.013626369647681713,\n", + " -0.007093450985848904,\n", + " -0.017960568889975548,\n", + " 0.0556635856628418,\n", + " 0.02151196263730526,\n", + " -0.006550669204443693,\n", + " -0.004232341423630714,\n", + " -0.01489347219467163,\n", + " -0.021089769899845123,\n", + " 0.0007471065619029105,\n", + " 0.005566490814089775,\n", + " 0.014780324883759022,\n", + " 0.004473445471376181,\n", + " 0.02594108320772648,\n", + " -0.008353671059012413,\n", + " -0.012298411689698696,\n", + " -0.027804264798760414,\n", + " 0.008500847034156322,\n", + " -0.01670648157596588,\n", + " -0.030227677896618843,\n", + " -0.0008617430576123297,\n", + " -0.012609113939106464,\n", + " -0.026223087683320045,\n", + " 0.011928856372833252,\n", + " 0.013128691352903843,\n", + " 0.015468685887753963,\n", + " -0.009659596718847752,\n", + " -0.005760476924479008,\n", + " 0.017638003453612328,\n", + " -0.007418491877615452,\n", + " 0.00456077279523015,\n", + " 0.024832524359226227,\n", + " -0.003971753176301718,\n", + " 0.024014055728912354,\n", + " 0.0029347536619752645,\n", + " 0.009343280456960201,\n", + " -0.007382581476122141,\n", + " 0.02028382383286953,\n", + " -0.01377318985760212,\n", + " 0.00569793488830328,\n", + " -0.009646281599998474,\n", + " 0.004583550151437521,\n", + " 0.02593171037733555,\n", + " 0.010284800082445145,\n", + " -0.02534230425953865,\n", + " 0.016492048278450966,\n", + " -0.01944207213819027,\n", + " 0.012236645445227623,\n", + " -0.018289977684617043,\n", + " -0.011027022264897823,\n", + " -0.03984448313713074,\n", + " -0.01360741350799799,\n", + " 0.014925851486623287,\n", + " -0.024778995662927628,\n", + " 0.0075136348605155945,\n", + " 7.207586895674467e-05,\n", + " -0.0034446946810930967,\n", + " 0.014232967980206013,\n", + " 0.004762297961860895,\n", + " -0.020427986979484558,\n", + " 0.016299230977892876,\n", + " 0.007874958217144012,\n", + " -0.0037723788991570473,\n", + " -0.020174451172351837,\n", + " 0.0064780935645103455,\n", + " -0.01707850955426693,\n", + " -0.008320528082549572,\n", + " -0.014858445152640343,\n", + " -0.0104805463925004,\n", + " -0.00347711774520576,\n", + " -0.003243209794163704,\n", + " 0.008600924164056778,\n", + " 0.019620854407548904,\n", + " 0.010859405621886253,\n", + " -0.03035123646259308,\n", + " 0.0031244850251823664,\n", + " -0.0008457346120849252,\n", + " -0.030203018337488174,\n", + " 0.005136424675583839,\n", + " -0.029637040570378304,\n", + " 0.004290843848139048,\n", + " -0.020740751177072525,\n", + " 0.0008698026067577302,\n", + " 0.01733979769051075,\n", + " -0.0017592560034245253,\n", + " 0.005069995764642954,\n", + " -0.008046209812164307,\n", + " -0.014235840179026127,\n", + " -0.0037953874561935663,\n", + " -6.226154800970107e-05,\n", + " 0.012463097460567951,\n", + " -0.0012896147090941668,\n", + " -0.012952055782079697,\n", + " 0.00035749879316426814,\n", + " 0.002543324837461114,\n", + " 0.000518229731824249,\n", + " 0.024755332618951797,\n", + " -0.012228927575051785,\n", + " -0.023000486195087433,\n", + " 0.021329350769519806,\n", + " 0.015798911452293396,\n", + " -0.016479918733239174,\n", + " -0.020029818639159203,\n", + " -0.01717989146709442,\n", + " -0.004491395782679319,\n", + " -0.0003751168551389128,\n", + " -0.022424226626753807,\n", + " 0.0035433790180832148,\n", + " -0.013971994630992413,\n", + " -0.002235779073089361,\n", + " 0.012958453968167305,\n", + " -0.01934337057173252,\n", + " 0.01162923313677311,\n", + " 0.0017600803403183818,\n", + " 0.001735839992761612,\n", + " 0.02399849146604538,\n", + " -0.013805736787617207,\n", + " -0.0017815890023484826,\n", + " 0.0096052261069417,\n", + " -0.002516506239771843,\n", + " -0.010889054276049137,\n", + " -0.038546815514564514,\n", + " -0.0009700870723463595,\n", + " 0.003600931726396084,\n", + " -0.012653791345655918,\n", + " -0.015539748594164848,\n", + " 0.0036487646866589785,\n", + " -0.011216487735509872,\n", + " 0.0043421583250164986,\n", + " -0.006353787612169981,\n", + " 0.016105052083730698,\n", + " -0.006433302536606789,\n", + " -0.009744004346430302,\n", + " 0.0037180231884121895,\n", + " -0.01781967096030712,\n", + " 0.0012477737618610263,\n", + " -0.029512789100408554,\n", + " -0.011096007190644741,\n", + " 0.010373931378126144,\n", + " 0.015442590229213238,\n", + " 0.006841790396720171,\n", + " 0.012226310558617115,\n", + " 0.02514396794140339,\n", + " 6.883557216497138e-05,\n", + " -0.0019605269189924,\n", + " 0.005450403783470392,\n", + " 0.05505552142858505,\n", + " -0.0008810920407995582,\n", + " -0.025708142668008804,\n", + " 0.0008815747569315135,\n", + " -0.06268516927957535,\n", + " 0.002696027047932148,\n", + " 0.006442879792302847,\n", + " 0.004262510221451521,\n", + " 0.008320296183228493,\n", + " 0.012818093411624432,\n", + " -0.006261391565203667,\n", + " -0.0016345081385225058,\n", + " -0.014989924617111683,\n", + " 0.011508957482874393,\n", + " -0.015395257622003555,\n", + " -0.0002456325455568731,\n", + " 0.0028725401498377323,\n", + " -0.022297225892543793,\n", + " 0.012327374890446663,\n", + " 0.010972017422318459,\n", + " 0.006332955323159695,\n", + " 0.014015263877809048,\n", + " -0.010212399065494537,\n", + " 0.024118591099977493,\n", + " -0.014639408327639103,\n", + " 0.009966536425054073,\n", + " 0.004061818588525057,\n", + " 0.002801054622977972,\n", + " -0.002328819828107953,\n", + " 0.022628651931881905,\n", + " 0.03169957548379898,\n", + " -0.005670144222676754,\n", + " 0.014185333624482155,\n", + " -0.00693044438958168,\n", + " -0.0018200587946921587,\n", + " -0.010325311683118343,\n", + " -0.0049256859347224236,\n", + " -0.02498791180551052,\n", + " -0.01577681303024292,\n", + " -0.0033557023853063583,\n", + " -0.008299502544105053,\n", + " -0.00450667692348361,\n", + " -0.011009606532752514,\n", + " 0.01727048121392727,\n", + " 0.004911783616989851,\n", + " -0.017111871391534805,\n", + " -0.0019733328372240067,\n", + " 0.014826241880655289,\n", + " 0.0017785666277632117,\n", + " 0.0052349017933011055,\n", + " 0.0073284609243273735,\n", + " -0.018747160211205482,\n", + " -0.024404797703027725,\n", + " 0.009125935845077038,\n", + " -0.00042940620915032923,\n", + " -0.010243147611618042,\n", + " 0.0018020515562966466,\n", + " -0.013518726453185081,\n", + " 0.0012687112903222442,\n", + " 0.008444477804005146,\n", + " 0.016314662992954254,\n", + " -0.021775074303150177,\n", + " -0.017303291708230972,\n", + " 0.001829018467105925,\n", + " -0.0019452639389783144,\n", + " -0.022065294906497,\n", + " 0.008146111853420734,\n", + " 0.012680048123002052,\n", + " -0.010362723842263222,\n", + " 0.029195884242653847,\n", + " -0.011800278909504414,\n", + " 0.0045953961089253426,\n", + " -0.0025577708147466183,\n", + " -0.01839444600045681,\n", + " 0.007579263299703598,\n", + " -0.010845270939171314,\n", + " 0.0101514533162117,\n", + " -0.03438518941402435,\n", + " 0.004026987124234438,\n", + " -0.0043350569903850555,\n", + " -0.0015670316061004996,\n", + " -0.013465072959661484,\n", + " 0.014462114311754704,\n", + " -0.013360978104174137,\n", + " -0.0072088055312633514,\n", + " -0.009346218779683113,\n", + " -0.01592816226184368,\n", + " 0.020320124924182892,\n", + " -0.010124020278453827,\n", + " -0.009361792355775833,\n", + " 0.005349436774849892,\n", + " 0.007697821594774723,\n", + " 0.02099333517253399,\n", + " 0.03613070026040077,\n", + " 0.004412619397044182,\n", + " -0.0007328703650273383,\n", + " 0.026337556540966034,\n", + " -0.007886849343776703,\n", + " 0.0010734288953244686,\n", + " 0.02038503810763359,\n", + " -0.021293507888913155,\n", + " 0.0005149429198354483,\n", + " -0.010475543327629566,\n", + " -0.006535436026751995,\n", + " -0.009200300090014935,\n", + " 0.0029004113748669624,\n", + " 0.013081453740596771,\n", + " -0.0035991701297461987,\n", + " -0.008680792525410652,\n", + " 0.008129253052175045,\n", + " -0.0077785924077034,\n", + " -0.00902999471873045,\n", + " 0.00724017946049571,\n", + " -0.0012517786817625165,\n", + " 0.013853000476956367,\n", + " -0.015145980753004551,\n", + " 0.027656378224492073,\n", + " 0.013293327763676643,\n", + " -0.0061129010282456875,\n", + " 0.030545543879270554,\n", + " 0.023482991382479668,\n", + " -0.009798603132367134,\n", + " 0.027960622683167458,\n", + " -0.0126644903793931,\n", + " -0.00012814425281248987,\n", + " 0.006706354208290577,\n", + " 0.0018757573561742902,\n", + " -0.029307106509804726,\n", + " 0.004845940973609686,\n", + " 0.008660756051540375,\n", + " 0.011811697855591774,\n", + " 0.01259523257613182,\n", + " 0.00584376510232687,\n", + " -0.009611032903194427,\n", + " -0.006454362999647856,\n", + " -0.008835878223180771,\n", + " 0.013815462589263916,\n", + " -0.0005935532390139997,\n", + " 0.011585534550249577,\n", + " 0.00804165843874216,\n", + " -0.0046113841235637665,\n", + " -0.022198613733053207,\n", + " -0.0011589800706133246,\n", + " 0.011985939927399158,\n", + " -0.0070546455681324005,\n", + " -0.0011772031430155039,\n", + " 0.005077525973320007,\n", + " 0.004629608243703842,\n", + " -0.00513886334374547,\n", + " 0.010327215306460857,\n", + " 0.023579830303788185,\n", + " -0.03293757513165474,\n", + " -0.009293223731219769,\n", + " -0.010876808315515518,\n", + " -0.027919895946979523,\n", + " 0.002014430705457926,\n", + " 0.0015256097540259361,\n", + " -0.0007074680761434138,\n", + " -0.009122752584517002,\n", + " 0.008312408812344074,\n", + " 0.01027339231222868,\n", + " -0.02813871204853058,\n", + " 0.007871834561228752,\n", + " 0.001521389465779066,\n", + " -0.011350546963512897,\n", + " 0.021417556330561638,\n", + " 0.0006441604346036911,\n", + " -0.02114005759358406,\n", + " 0.038964953273534775,\n", + " -0.0042233336716890335,\n", + " 0.027741871774196625,\n", + " -0.00549342343583703,\n", + " 0.023450210690498352,\n", + " -0.013218838721513748,\n", + " -0.008897709660232067,\n", + " 0.0169205442070961,\n", + " 0.004693590570241213,\n", + " 0.004693206399679184,\n", + " 0.027811110019683838,\n", + " 0.009191364981234074,\n", + " -0.013211927376687527,\n", + " -0.0007477460894733667,\n", + " -0.008817661553621292,\n", + " -0.03000003471970558,\n", + " -0.013140132650732994,\n", + " -0.030061693862080574,\n", + " 0.015250189229846,\n", + " -0.014456876553595066,\n", + " 0.01388415414839983,\n", + " 0.0044051725417375565,\n", + " 0.019094303250312805,\n", + " 0.030994007363915443,\n", + " -0.035488810390233994,\n", + " -0.019251754507422447,\n", + " -0.02982616238296032,\n", + " 0.014683743007481098,\n", + " 0.030743330717086792,\n", + " 0.021809089928865433,\n", + " -0.004061093553900719,\n", + " 0.008110971190035343,\n", + " 0.00030069550848565996,\n", + " 0.007436910178512335,\n", + " 0.017309658229351044,\n", + " -0.01872468926012516,\n", + " -0.0038973200134932995,\n", + " -0.011617379263043404,\n", + " 0.0028235134668648243,\n", + " 0.010349615477025509,\n", + " 0.018053589388728142,\n", + " -0.01204252801835537,\n", + " 0.007784688845276833,\n", + " 0.04340056702494621,\n", + " -0.0224344152957201,\n", + " -0.003077515633776784,\n", + " -0.0005072857020422816,\n", + " -0.0025440549943596125,\n", + " -0.03158242627978325,\n", + " -0.004591826349496841,\n", + " -0.015459216199815273,\n", + " 0.0016550722066313028,\n", + " -0.021909017115831375,\n", + " 0.00791469868272543,\n", + " 0.017703266814351082,\n", + " 0.014343260787427425,\n", + " -0.009737424552440643,\n", + " -0.003000229364261031,\n", + " 0.004739667288959026,\n", + " -0.012545120902359486,\n", + " 0.018552439287304878,\n", + " 0.011897699907422066,\n", + " -0.0030499869026243687,\n", + " 0.019290996715426445,\n", + " -0.010966756381094456,\n", + " -0.0069915358908474445,\n", + " -0.013163027353584766,\n", + " 0.021801728755235672,\n", + " 0.0011354534653946757,\n", + " -0.005458917003124952,\n", + " 0.026549678295850754,\n", + " 0.020782314240932465,\n", + " 0.0176919586956501,\n", + " -0.009557580575346947,\n", + " -0.007981647737324238,\n", + " 0.03168530389666557,\n", + " -0.002494144020602107,\n", + " 0.01719747669994831,\n", + " -0.013710014522075653,\n", + " -0.003989398945122957,\n", + " 0.011352983303368092,\n", + " -0.003987086936831474,\n", + " 0.005175672937184572,\n", + " -0.010003799572587013,\n", + " 0.004276175983250141,\n", + " 0.008259350433945656,\n", + " 0.016041047871112823,\n", + " -0.002010929863899946,\n", + " 0.007027979474514723,\n", + " 0.012356432154774666,\n", + " -0.013807359151542187,\n", + " 0.018796386197209358,\n", + " 0.002758659655228257,\n", + " -0.013705180026590824,\n", + " -0.0011855674674734473,\n", + " 0.030971845611929893,\n", + " 0.009778724983334541,\n", + " -0.011201448738574982,\n", + " 0.010989927686750889,\n", + " 0.0008666506037116051,\n", + " 0.017514413222670555,\n", + " 0.017922034487128258,\n", + " 0.008039798587560654,\n", + " 0.018007325008511543,\n", + " -0.000454249995527789,\n", + " 0.0043387943878769875,\n", + " 0.014981968328356743,\n", + " -0.031026123091578484,\n", + " -0.009392671287059784,\n", + " -0.016183026134967804,\n", + " 0.016184339299798012,\n", + " 0.02907208539545536,\n", + " -0.008433868177235126,\n", + " 0.005499284714460373,\n", + " 0.013863838277757168,\n", + " -0.021100474521517754,\n", + " -0.008125292137265205,\n", + " -0.007032178808003664,\n", + " -0.010406806133687496,\n", + " 0.00202157418243587,\n", + " -0.002188085112720728,\n", + " -0.03734145313501358,\n", + " 0.024517972022294998,\n", + " -0.008487368002533913,\n", + " -0.000533925776835531,\n", + " -0.019055671989917755,\n", + " -0.010654153302311897,\n", + " 0.005966866388916969,\n", + " -0.01976938173174858,\n", + " -0.010791301727294922,\n", + " -0.025069167837500572,\n", + " 0.032491523772478104,\n", + " -0.0010522839147597551,\n", + " 0.02935481071472168,\n", + " 0.001831167726777494,\n", + " -0.006750455126166344,\n", + " 0.006963513791561127,\n", + " -0.01235498022288084,\n", + " -0.00947477575391531,\n", + " 0.005211047828197479,\n", + " -0.00418825214728713,\n", + " 0.00045644232886843383,\n", + " -0.0051966155879199505,\n", + " -0.008230665698647499,\n", + " -0.000525494571775198,\n", + " -0.021747473627328873,\n", + " -0.025246966630220413,\n", + " -0.0023829247802495956,\n", + " ...]]" + ] + }, + "execution_count": 7, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "document_result" + ] + }, + { + "cell_type": "code", + "execution_count": 8, + "id": "9b7ba678-6c02-46b3-8c4d-169fe4c93ea1", + "metadata": { + "ExecuteTime": { + "end_time": "2023-05-24T15:13:18.620432Z", + "start_time": "2023-05-24T15:13:18.608335Z" + }, + "execution": { + "iopub.execute_input": "2024-03-29T15:39:38.446478Z", + "iopub.status.busy": "2024-03-29T15:39:38.445110Z", + "iopub.status.idle": "2024-03-29T15:39:38.521371Z", + "shell.execute_reply": "2024-03-29T15:39:38.520658Z", + "shell.execute_reply.started": "2024-03-29T15:39:38.446388Z" + } + }, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Cosine similarity between document and query: 0.8685132879722154\n" + ] + } + ], + "source": [ + "import numpy as np\n", + "\n", + "query_numpy = np.array(query_result)\n", + "document_numpy = np.array(document_result[0])\n", + "similarity = np.dot(query_numpy, document_numpy) / (\n", + " np.linalg.norm(query_numpy) * np.linalg.norm(document_numpy)\n", + ")\n", + "print(f\"Cosine similarity between document and query: {similarity}\")" + ] + } + ], + "metadata": { + "kernelspec": { + "display_name": "Python 3 (ipykernel)", + "language": "python", + "name": "python3" + }, + "language_info": { + "codemirror_mode": { + "name": "ipython", + "version": 3 + }, + "file_extension": ".py", + "mimetype": "text/x-python", + "name": "python", + "nbconvert_exporter": "python", + "pygments_lexer": "ipython3", + "version": "3.9.0" + } + }, + "nbformat": 4, + "nbformat_minor": 5 +} diff --git a/libs/community/langchain_community/embeddings/__init__.py b/libs/community/langchain_community/embeddings/__init__.py index 929301a9ce..866374c446 100644 --- a/libs/community/langchain_community/embeddings/__init__.py +++ b/libs/community/langchain_community/embeddings/__init__.py @@ -77,6 +77,7 @@ _module_lookup = { "SelfHostedHuggingFaceEmbeddings": "langchain_community.embeddings.self_hosted_hugging_face", # noqa: E501 "SelfHostedHuggingFaceInstructEmbeddings": "langchain_community.embeddings.self_hosted_hugging_face", # noqa: E501 "SentenceTransformerEmbeddings": "langchain_community.embeddings.sentence_transformer", # noqa: E501 + "SolarEmbeddings": "langchain_community.embeddings.solar", "SpacyEmbeddings": "langchain_community.embeddings.spacy_embeddings", "SparkLLMTextEmbeddings": "langchain_community.embeddings.sparkllm", "TensorflowHubEmbeddings": "langchain_community.embeddings.tensorflow_hub", diff --git a/libs/community/langchain_community/embeddings/solar.py b/libs/community/langchain_community/embeddings/solar.py new file mode 100644 index 0000000000..30979c952c --- /dev/null +++ b/libs/community/langchain_community/embeddings/solar.py @@ -0,0 +1,139 @@ +from __future__ import annotations + +import logging +from typing import Any, Callable, Dict, List, Optional + +import requests +from langchain_core.embeddings import Embeddings +from langchain_core.pydantic_v1 import BaseModel, Extra, SecretStr, root_validator +from langchain_core.utils import convert_to_secret_str, get_from_dict_or_env +from tenacity import ( + before_sleep_log, + retry, + stop_after_attempt, + wait_exponential, +) + +logger = logging.getLogger(__name__) + + +def _create_retry_decorator() -> Callable[[Any], Any]: + """Returns a tenacity retry decorator.""" + + multiplier = 1 + min_seconds = 1 + max_seconds = 4 + max_retries = 6 + + return retry( + reraise=True, + stop=stop_after_attempt(max_retries), + wait=wait_exponential(multiplier=multiplier, min=min_seconds, max=max_seconds), + before_sleep=before_sleep_log(logger, logging.WARNING), + ) + + +def embed_with_retry(embeddings: SolarEmbeddings, *args: Any, **kwargs: Any) -> Any: + """Use tenacity to retry the completion call.""" + retry_decorator = _create_retry_decorator() + + @retry_decorator + def _embed_with_retry(*args: Any, **kwargs: Any) -> Any: + return embeddings.embed(*args, **kwargs) + + return _embed_with_retry(*args, **kwargs) + + +class SolarEmbeddings(BaseModel, Embeddings): + """Solar's embedding service. + + To use, you should have the environment variable``SOLAR_API_KEY`` set + with your API token, or pass it as a named parameter to the constructor. + + Example: + .. code-block:: python + + from langchain_community.embeddings import SolarEmbeddings + embeddings = SolarEmbeddings() + + query_text = "This is a test query." + query_result = embeddings.embed_query(query_text) + + document_text = "This is a test document." + document_result = embeddings.embed_documents([document_text]) + + """ + + endpoint_url: str = "https://api.upstage.ai/v1/solar/embeddings" + """Endpoint URL to use.""" + model: str = "solar-1-mini-embedding-query" + """Embeddings model name to use.""" + solar_api_key: Optional[SecretStr] = None + """API Key for Solar API.""" + + class Config: + """Configuration for this pydantic object.""" + + extra = Extra.forbid + + @root_validator() + def validate_environment(cls, values: Dict) -> Dict: + """Validate api key exists in environment.""" + solar_api_key = convert_to_secret_str( + get_from_dict_or_env(values, "solar_api_key", "SOLAR_API_KEY") + ) + values["solar_api_key"] = solar_api_key + return values + + def embed( + self, + text: str, + ) -> List[List[float]]: + payload = { + "model": self.model, + "input": text, + } + + # HTTP headers for authorization + headers = { + "Authorization": f"Bearer {self.solar_api_key.get_secret_value()}", # type: ignore[union-attr] + "Content-Type": "application/json", + } + + # send request + response = requests.post(self.endpoint_url, headers=headers, json=payload) + parsed_response = response.json() + + # check for errors + if len(parsed_response["data"]) == 0: + raise ValueError( + f"Solar API returned an error: {parsed_response['base_resp']}" + ) + + embedding = parsed_response["data"][0]["embedding"] + + return embedding + + def embed_documents(self, texts: List[str]) -> List[List[float]]: + """Embed documents using a Solar embedding endpoint. + + Args: + texts: The list of texts to embed. + + Returns: + List of embeddings, one for each text. + """ + embeddings = [embed_with_retry(self, text=text) for text in texts] + return embeddings + + def embed_query(self, text: str) -> List[float]: + """Embed a query using a Solar embedding endpoint. + + Args: + text: The text to embed. + + Returns: + Embeddings for the text. + """ + embedding = embed_with_retry(self, text=text) + return embedding diff --git a/libs/community/tests/unit_tests/embeddings/test_imports.py b/libs/community/tests/unit_tests/embeddings/test_imports.py index 56b7f247d6..4d1c821385 100644 --- a/libs/community/tests/unit_tests/embeddings/test_imports.py +++ b/libs/community/tests/unit_tests/embeddings/test_imports.py @@ -71,6 +71,7 @@ EXPECTED_ALL = [ "YandexGPTEmbeddings", "OpenVINOEmbeddings", "OpenVINOBgeEmbeddings", + "SolarEmbeddings", ]