diff --git a/coverage.xml b/coverage.xml index 1a05608c73365623edca53f76e019449b1e39698..14b27c40f0c8e94dbd46232dd28b021f8ff9da1b 100644 --- a/coverage.xml +++ b/coverage.xml @@ -1,12 +1,12 @@ <?xml version="1.0" ?> -<coverage version="7.4.1" timestamp="1708577611401" lines-valid="1704" lines-covered="1346" line-rate="0.7899" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0"> - <!-- Generated by coverage.py: https://coverage.readthedocs.io/en/7.4.1 --> +<coverage version="7.4.3" timestamp="1709360905659" lines-valid="2024" lines-covered="1605" line-rate="0.793" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0"> + <!-- Generated by coverage.py: https://coverage.readthedocs.io/en/7.4.3 --> <!-- Based on https://raw.githubusercontent.com/cobertura/web/master/htdocs/xml/coverage-04.dtd --> <sources> - <source>/home/dustinmorris/semantic-router/semantic_router</source> + <source>/Users/jamesbriggs/Documents/projects/aurelio-labs/semantic-router/semantic_router</source> </sources> <packages> - <package name="." line-rate="0.9" branch-rate="0" complexity="0"> + <package name="." line-rate="0.9007" branch-rate="0" complexity="0"> <classes> <class name="__init__.py" filename="__init__.py" complexity="0" line-rate="1" branch-rate="0"> <methods/> @@ -147,38 +147,40 @@ <line number="17" hits="1"/> <line number="18" hits="1"/> <line number="21" hits="1"/> - <line number="23" hits="1"/> <line number="24" hits="1"/> <line number="25" hits="1"/> - <line number="27" hits="1"/> - <line number="28" hits="0"/> + <line number="26" hits="1"/> + <line number="28" hits="1"/> <line number="29" hits="0"/> <line number="30" hits="0"/> <line number="31" hits="0"/> - <line number="34" hits="0"/> + <line number="32" hits="0"/> <line number="35" hits="0"/> - <line number="37" hits="1"/> + <line number="36" hits="0"/> <line number="38" hits="1"/> - <line number="39" hits="0"/> - <line number="42" hits="0"/> - <line number="44" hits="1"/> - <line number="45" hits="0"/> + <line number="39" hits="1"/> + <line number="40" hits="0"/> + <line number="43" hits="0"/> + <line number="45" hits="1"/> <line number="46" hits="0"/> <line number="47" hits="0"/> - <line number="50" hits="1"/> - <line number="56" hits="1"/> - <line number="58" hits="1"/> - <line number="64" hits="1"/> + <line number="48" hits="0"/> + <line number="51" hits="1"/> + <line number="57" hits="1"/> + <line number="59" hits="1"/> <line number="65" hits="1"/> + <line number="66" hits="1"/> + <line number="67" hits="1"/> + <line number="68" hits="1"/> <line number="69" hits="1"/> - <line number="70" hits="1"/> - <line number="71" hits="1"/> - <line number="72" hits="0"/> - <line number="75" hits="1"/> + <line number="70" hits="0"/> + <line number="73" hits="1"/> + <line number="76" hits="1"/> + <line number="77" hits="1"/> <line number="78" hits="1"/> <line number="79" hits="1"/> - <line number="80" hits="1"/> <line number="81" hits="1"/> + <line number="82" hits="1"/> <line number="83" hits="1"/> <line number="84" hits="1"/> <line number="85" hits="1"/> @@ -186,226 +188,224 @@ <line number="87" hits="1"/> <line number="88" hits="1"/> <line number="89" hits="1"/> - <line number="90" hits="1"/> <line number="91" hits="1"/> - <line number="93" hits="1"/> - <line number="97" hits="1"/> + <line number="95" hits="1"/> + <line number="96" hits="1"/> <line number="98" hits="1"/> + <line number="99" hits="1"/> <line number="100" hits="1"/> <line number="101" hits="1"/> - <line number="102" hits="1"/> <line number="103" hits="1"/> - <line number="105" hits="1"/> - <line number="106" hits="1"/> + <line number="104" hits="1"/> + <line number="108" hits="1"/> <line number="110" hits="1"/> - <line number="112" hits="1"/> + <line number="111" hits="1"/> <line number="113" hits="1"/> - <line number="115" hits="1"/> - <line number="116" hits="1"/> - <line number="121" hits="1"/> + <line number="114" hits="1"/> + <line number="119" hits="1"/> + <line number="120" hits="1"/> <line number="122" hits="1"/> - <line number="124" hits="1"/> - <line number="128" hits="1"/> - <line number="129" hits="1"/> + <line number="126" hits="1"/> + <line number="127" hits="1"/> + <line number="133" hits="1"/> <line number="135" hits="1"/> - <line number="137" hits="1"/> - <line number="138" hits="1"/> - <line number="141" hits="1"/> - <line number="142" hits="1"/> - <line number="146" hits="1"/> - <line number="149" hits="1"/> - <line number="150" hits="0"/> + <line number="136" hits="1"/> + <line number="139" hits="1"/> + <line number="140" hits="1"/> + <line number="144" hits="1"/> + <line number="147" hits="1"/> + <line number="148" hits="0"/> + <line number="150" hits="1"/> + <line number="151" hits="1"/> <line number="152" hits="1"/> <line number="153" hits="1"/> <line number="154" hits="1"/> - <line number="155" hits="1"/> <line number="156" hits="1"/> + <line number="157" hits="1"/> <line number="158" hits="1"/> - <line number="159" hits="1"/> <line number="160" hits="1"/> + <line number="161" hits="1"/> <line number="162" hits="1"/> <line number="163" hits="1"/> <line number="164" hits="1"/> <line number="165" hits="1"/> - <line number="166" hits="1"/> <line number="167" hits="1"/> - <line number="169" hits="1"/> - <line number="170" hits="1"/> - <line number="171" hits="0"/> - <line number="173" hits="1"/> - <line number="174" hits="1"/> + <line number="168" hits="1"/> + <line number="169" hits="0"/> + <line number="171" hits="1"/> + <line number="172" hits="1"/> + <line number="175" hits="1"/> + <line number="176" hits="1"/> <line number="177" hits="1"/> <line number="178" hits="1"/> - <line number="179" hits="1"/> <line number="180" hits="1"/> - <line number="182" hits="1"/> + <line number="187" hits="1"/> + <line number="188" hits="1"/> <line number="189" hits="1"/> <line number="190" hits="1"/> - <line number="191" hits="1"/> - <line number="192" hits="1"/> + <line number="194" hits="1"/> <line number="196" hits="1"/> + <line number="197" hits="1"/> <line number="198" hits="1"/> <line number="199" hits="1"/> - <line number="200" hits="1"/> <line number="201" hits="1"/> + <line number="202" hits="1"/> <line number="203" hits="1"/> - <line number="204" hits="1"/> <line number="205" hits="1"/> <line number="207" hits="1"/> <line number="209" hits="1"/> + <line number="210" hits="1"/> <line number="211" hits="1"/> - <line number="212" hits="1"/> - <line number="213" hits="1"/> - <line number="214" hits="0"/> - <line number="217" hits="0"/> - <line number="218" hits="1"/> - <line number="220" hits="1"/> + <line number="212" hits="0"/> + <line number="216" hits="0"/> + <line number="217" hits="1"/> + <line number="219" hits="1"/> + <line number="225" hits="1"/> <line number="226" hits="1"/> <line number="227" hits="1"/> <line number="228" hits="1"/> - <line number="229" hits="1"/> - <line number="231" hits="1"/> - <line number="233" hits="1"/> - <line number="235" hits="1"/> + <line number="230" hits="1"/> + <line number="232" hits="1"/> + <line number="234" hits="1"/> + <line number="236" hits="1"/> <line number="237" hits="1"/> - <line number="238" hits="1"/> - <line number="239" hits="0"/> - <line number="240" hits="1"/> + <line number="238" hits="0"/> + <line number="239" hits="1"/> + <line number="244" hits="1"/> <line number="245" hits="1"/> <line number="246" hits="1"/> <line number="247" hits="1"/> - <line number="248" hits="1"/> - <line number="251" hits="1"/> + <line number="250" hits="1"/> + <line number="251" hits="0"/> <line number="252" hits="0"/> - <line number="253" hits="0"/> + <line number="258" hits="0"/> <line number="259" hits="0"/> - <line number="260" hits="0"/> - <line number="262" hits="0"/> - <line number="263" hits="1"/> - <line number="266" hits="1"/> - <line number="268" hits="1"/> - <line number="269" hits="0"/> + <line number="261" hits="0"/> + <line number="262" hits="1"/> + <line number="265" hits="1"/> + <line number="267" hits="1"/> + <line number="268" hits="0"/> + <line number="274" hits="1"/> <line number="275" hits="1"/> <line number="276" hits="1"/> <line number="277" hits="1"/> <line number="278" hits="1"/> - <line number="279" hits="1"/> + <line number="280" hits="1"/> <line number="281" hits="1"/> <line number="282" hits="1"/> <line number="283" hits="1"/> <line number="284" hits="1"/> - <line number="285" hits="1"/> + <line number="286" hits="1"/> <line number="287" hits="1"/> <line number="288" hits="1"/> <line number="289" hits="1"/> - <line number="290" hits="1"/> + <line number="291" hits="1"/> <line number="292" hits="1"/> - <line number="293" hits="1"/> - <line number="295" hits="1"/> + <line number="294" hits="1"/> + <line number="296" hits="1"/> <line number="297" hits="1"/> - <line number="298" hits="1"/> - <line number="301" hits="1"/> - <line number="306" hits="1"/> + <line number="300" hits="1"/> + <line number="305" hits="1"/> + <line number="307" hits="1"/> <line number="308" hits="1"/> - <line number="309" hits="1"/> - <line number="311" hits="1"/> - <line number="312" hits="0"/> - <line number="314" hits="1"/> + <line number="310" hits="1"/> + <line number="311" hits="0"/> + <line number="313" hits="1"/> + <line number="319" hits="1"/> <line number="320" hits="1"/> <line number="321" hits="1"/> <line number="322" hits="1"/> - <line number="323" hits="1"/> + <line number="324" hits="1"/> <line number="325" hits="1"/> - <line number="326" hits="1"/> - <line number="328" hits="1"/> - <line number="330" hits="0"/> - <line number="345" hits="1"/> - <line number="347" hits="1"/> - <line number="350" hits="1"/> - <line number="352" hits="1"/> - <line number="354" hits="1"/> - <line number="360" hits="1"/> + <line number="327" hits="1"/> + <line number="329" hits="0"/> + <line number="344" hits="1"/> + <line number="346" hits="1"/> + <line number="349" hits="1"/> + <line number="351" hits="1"/> + <line number="353" hits="1"/> + <line number="359" hits="1"/> + <line number="362" hits="1"/> <line number="363" hits="1"/> <line number="364" hits="1"/> - <line number="365" hits="1"/> - <line number="367" hits="1"/> + <line number="366" hits="1"/> + <line number="369" hits="1"/> <line number="370" hits="1"/> - <line number="371" hits="1"/> + <line number="372" hits="1"/> <line number="373" hits="1"/> <line number="374" hits="1"/> <line number="375" hits="1"/> <line number="376" hits="1"/> <line number="377" hits="1"/> <line number="378" hits="1"/> - <line number="379" hits="1"/> - <line number="381" hits="1"/> + <line number="380" hits="1"/> + <line number="383" hits="1"/> <line number="384" hits="1"/> - <line number="385" hits="1"/> + <line number="387" hits="1"/> <line number="388" hits="1"/> - <line number="389" hits="1"/> + <line number="390" hits="0"/> <line number="391" hits="0"/> - <line number="392" hits="0"/> + <line number="393" hits="1"/> <line number="394" hits="1"/> <line number="395" hits="1"/> - <line number="396" hits="1"/> - <line number="398" hits="1"/> - <line number="400" hits="1"/> + <line number="397" hits="1"/> + <line number="399" hits="1"/> + <line number="403" hits="1"/> <line number="404" hits="1"/> <line number="405" hits="1"/> - <line number="406" hits="1"/> + <line number="409" hits="1"/> <line number="410" hits="1"/> - <line number="411" hits="1"/> + <line number="416" hits="1"/> <line number="417" hits="1"/> <line number="418" hits="1"/> - <line number="419" hits="1"/> + <line number="420" hits="1"/> <line number="421" hits="1"/> <line number="422" hits="1"/> - <line number="423" hits="1"/> - <line number="425" hits="1"/> - <line number="427" hits="1"/> - <line number="431" hits="1"/> - <line number="433" hits="1"/> + <line number="424" hits="1"/> + <line number="426" hits="1"/> + <line number="430" hits="1"/> + <line number="432" hits="1"/> + <line number="440" hits="1"/> <line number="441" hits="1"/> <line number="442" hits="1"/> <line number="443" hits="1"/> - <line number="444" hits="1"/> + <line number="445" hits="1"/> <line number="446" hits="1"/> - <line number="447" hits="1"/> + <line number="448" hits="1"/> <line number="449" hits="1"/> - <line number="450" hits="1"/> - <line number="452" hits="1"/> - <line number="457" hits="1"/> - <line number="459" hits="1"/> - <line number="461" hits="1"/> + <line number="451" hits="1"/> + <line number="456" hits="1"/> + <line number="458" hits="1"/> + <line number="460" hits="1"/> + <line number="461" hits="0"/> <line number="462" hits="0"/> - <line number="463" hits="0"/> - <line number="465" hits="1"/> - <line number="467" hits="1"/> + <line number="464" hits="1"/> + <line number="466" hits="1"/> + <line number="470" hits="1"/> <line number="471" hits="1"/> <line number="472" hits="1"/> <line number="473" hits="1"/> - <line number="474" hits="1"/> + <line number="475" hits="1"/> <line number="476" hits="1"/> - <line number="477" hits="1"/> - <line number="479" hits="1"/> + <line number="478" hits="1"/> + <line number="482" hits="1"/> <line number="483" hits="1"/> <line number="484" hits="1"/> <line number="485" hits="1"/> <line number="486" hits="1"/> <line number="487" hits="1"/> <line number="488" hits="1"/> - <line number="489" hits="1"/> + <line number="490" hits="1"/> <line number="491" hits="1"/> - <line number="492" hits="1"/> - <line number="495" hits="1"/> + <line number="494" hits="1"/> + <line number="500" hits="1"/> <line number="501" hits="1"/> <line number="502" hits="1"/> - <line number="503" hits="1"/> + <line number="504" hits="1"/> <line number="505" hits="1"/> <line number="506" hits="1"/> - <line number="507" hits="1"/> - <line number="515" hits="1"/> - <line number="519" hits="1"/> + <line number="514" hits="1"/> + <line number="518" hits="1"/> </lines> </class> <class name="linear.py" filename="linear.py" complexity="0" line-rate="1" branch-rate="0"> @@ -426,96 +426,95 @@ <line number="30" hits="1"/> </lines> </class> - <class name="route.py" filename="route.py" complexity="0" line-rate="0.9048" branch-rate="0"> + <class name="route.py" filename="route.py" complexity="0" line-rate="0.9036" branch-rate="0"> <methods/> <lines> <line number="1" hits="1"/> <line number="2" hits="1"/> <line number="3" hits="1"/> <line number="5" hits="1"/> - <line number="6" hits="1"/> + <line number="7" hits="1"/> <line number="8" hits="1"/> <line number="9" hits="1"/> <line number="10" hits="1"/> - <line number="11" hits="1"/> + <line number="13" hits="1"/> <line number="14" hits="1"/> <line number="15" hits="1"/> <line number="16" hits="1"/> - <line number="17" hits="1"/> + <line number="18" hits="1"/> <line number="19" hits="1"/> <line number="20" hits="1"/> <line number="21" hits="1"/> <line number="22" hits="1"/> - <line number="23" hits="1"/> + <line number="25" hits="1"/> <line number="26" hits="1"/> - <line number="27" hits="1"/> + <line number="28" hits="1"/> <line number="29" hits="1"/> <line number="30" hits="1"/> - <line number="31" hits="1"/> - <line number="34" hits="1"/> + <line number="33" hits="1"/> + <line number="35" hits="1"/> <line number="36" hits="1"/> <line number="37" hits="1"/> <line number="38" hits="1"/> - <line number="39" hits="1"/> + <line number="41" hits="1"/> <line number="42" hits="1"/> <line number="43" hits="1"/> <line number="44" hits="1"/> <line number="45" hits="1"/> <line number="46" hits="1"/> <line number="47" hits="1"/> - <line number="48" hits="1"/> + <line number="49" hits="1"/> <line number="50" hits="1"/> - <line number="51" hits="1"/> + <line number="52" hits="1"/> <line number="53" hits="1"/> <line number="54" hits="1"/> <line number="55" hits="1"/> - <line number="56" hits="1"/> + <line number="59" hits="0"/> <line number="60" hits="0"/> - <line number="61" hits="0"/> - <line number="66" hits="0"/> - <line number="69" hits="0"/> + <line number="65" hits="0"/> + <line number="68" hits="0"/> + <line number="71" hits="1"/> <line number="72" hits="1"/> - <line number="73" hits="1"/> + <line number="77" hits="1"/> <line number="78" hits="1"/> <line number="79" hits="1"/> - <line number="80" hits="1"/> - <line number="81" hits="0"/> - <line number="86" hits="1"/> + <line number="80" hits="0"/> + <line number="85" hits="1"/> + <line number="87" hits="1"/> <line number="88" hits="1"/> <line number="89" hits="1"/> - <line number="90" hits="1"/> + <line number="91" hits="1"/> <line number="92" hits="1"/> - <line number="93" hits="1"/> + <line number="96" hits="1"/> <line number="97" hits="1"/> <line number="98" hits="1"/> <line number="99" hits="1"/> - <line number="100" hits="1"/> + <line number="101" hits="1"/> <line number="102" hits="1"/> - <line number="103" hits="1"/> + <line number="104" hits="1"/> <line number="105" hits="1"/> - <line number="106" hits="1"/> + <line number="107" hits="1"/> <line number="108" hits="1"/> <line number="109" hits="1"/> - <line number="110" hits="1"/> - <line number="112" hits="0"/> + <line number="111" hits="0"/> + <line number="113" hits="1"/> <line number="114" hits="1"/> <line number="115" hits="1"/> - <line number="116" hits="1"/> - <line number="118" hits="1"/> + <line number="117" hits="1"/> + <line number="142" hits="1"/> <line number="143" hits="1"/> <line number="144" hits="1"/> - <line number="145" hits="1"/> - <line number="146" hits="0"/> - <line number="148" hits="1"/> - <line number="150" hits="1"/> + <line number="145" hits="0"/> + <line number="147" hits="1"/> + <line number="149" hits="1"/> + <line number="151" hits="1"/> <line number="152" hits="1"/> <line number="153" hits="1"/> <line number="154" hits="1"/> - <line number="155" hits="1"/> - <line number="156" hits="0"/> + <line number="155" hits="0"/> </lines> </class> - <class name="schema.py" filename="schema.py" complexity="0" line-rate="0.8929" branch-rate="0"> + <class name="schema.py" filename="schema.py" complexity="0" line-rate="0.9016" branch-rate="0"> <methods/> <lines> <line number="1" hits="1"/> @@ -565,7 +564,7 @@ <line number="66" hits="1"/> <line number="67" hits="1"/> <line number="69" hits="1"/> - <line number="70" hits="0"/> + <line number="70" hits="1"/> <line number="72" hits="1"/> <line number="73" hits="1"/> <line number="75" hits="1"/> @@ -574,6 +573,11 @@ <line number="80" hits="1"/> <line number="81" hits="1"/> <line number="82" hits="1"/> + <line number="83" hits="1"/> + <line number="84" hits="1"/> + <line number="86" hits="1"/> + <line number="87" hits="1"/> + <line number="88" hits="0"/> </lines> </class> <class name="text.py" filename="text.py" complexity="0" line-rate="0.7778" branch-rate="0"> @@ -655,7 +659,7 @@ </class> </classes> </package> - <package name="encoders" line-rate="0.834" branch-rate="0" complexity="0"> + <package name="encoders" line-rate="0.9574" branch-rate="0" complexity="0"> <classes> <class name="__init__.py" filename="encoders/__init__.py" complexity="0" line-rate="1" branch-rate="0"> <methods/> @@ -670,7 +674,8 @@ <line number="8" hits="1"/> <line number="9" hits="1"/> <line number="10" hits="1"/> - <line number="12" hits="1"/> + <line number="11" hits="1"/> + <line number="13" hits="1"/> </lines> </class> <class name="base.py" filename="encoders/base.py" complexity="0" line-rate="1" branch-rate="0"> @@ -740,6 +745,93 @@ <line number="67" hits="1"/> </lines> </class> + <class name="clip.py" filename="encoders/clip.py" complexity="0" line-rate="0.939" branch-rate="0"> + <methods/> + <lines> + <line number="1" hits="1"/> + <line number="3" hits="1"/> + <line number="4" hits="1"/> + <line number="6" hits="1"/> + <line number="9" hits="1"/> + <line number="10" hits="1"/> + <line number="11" hits="1"/> + <line number="12" hits="1"/> + <line number="13" hits="1"/> + <line number="14" hits="1"/> + <line number="15" hits="1"/> + <line number="16" hits="1"/> + <line number="17" hits="1"/> + <line number="18" hits="1"/> + <line number="19" hits="1"/> + <line number="20" hits="1"/> + <line number="21" hits="1"/> + <line number="23" hits="1"/> + <line number="24" hits="1"/> + <line number="25" hits="1"/> + <line number="27" hits="1"/> + <line number="33" hits="1"/> + <line number="34" hits="1"/> + <line number="35" hits="1"/> + <line number="37" hits="1"/> + <line number="38" hits="1"/> + <line number="39" hits="1"/> + <line number="40" hits="1"/> + <line number="41" hits="1"/> + <line number="43" hits="1"/> + <line number="44" hits="1"/> + <line number="45" hits="1"/> + <line number="47" hits="1"/> + <line number="48" hits="1"/> + <line number="49" hits="1"/> + <line number="51" hits="1"/> + <line number="52" hits="1"/> + <line number="53" hits="1"/> + <line number="54" hits="1"/> + <line number="55" hits="1"/> + <line number="61" hits="1"/> + <line number="62" hits="1"/> + <line number="63" hits="1"/> + <line number="64" hits="1"/> + <line number="70" hits="1"/> + <line number="71" hits="1"/> + <line number="72" hits="0"/> + <line number="73" hits="0"/> + <line number="79" hits="1"/> + <line number="80" hits="1"/> + <line number="82" hits="1"/> + <line number="86" hits="1"/> + <line number="87" hits="1"/> + <line number="89" hits="1"/> + <line number="90" hits="1"/> + <line number="91" hits="1"/> + <line number="93" hits="1"/> + <line number="94" hits="1"/> + <line number="95" hits="0"/> + <line number="96" hits="1"/> + <line number="97" hits="0"/> + <line number="98" hits="1"/> + <line number="99" hits="1"/> + <line number="101" hits="0"/> + <line number="102" hits="1"/> + <line number="104" hits="1"/> + <line number="105" hits="1"/> + <line number="108" hits="1"/> + <line number="109" hits="1"/> + <line number="110" hits="1"/> + <line number="111" hits="1"/> + <line number="113" hits="1"/> + <line number="114" hits="1"/> + <line number="115" hits="1"/> + <line number="118" hits="1"/> + <line number="119" hits="1"/> + <line number="120" hits="1"/> + <line number="121" hits="1"/> + <line number="123" hits="1"/> + <line number="124" hits="1"/> + <line number="125" hits="1"/> + <line number="126" hits="1"/> + </lines> + </class> <class name="cohere.py" filename="encoders/cohere.py" complexity="0" line-rate="1" branch-rate="0"> <methods/> <lines> @@ -809,7 +901,7 @@ <line number="51" hits="0"/> </lines> </class> - <class name="huggingface.py" filename="encoders/huggingface.py" complexity="0" line-rate="0.4" branch-rate="0"> + <class name="huggingface.py" filename="encoders/huggingface.py" complexity="0" line-rate="0.9667" branch-rate="0"> <methods/> <lines> <line number="1" hits="1"/> @@ -833,45 +925,45 @@ <line number="25" hits="1"/> <line number="26" hits="1"/> <line number="27" hits="1"/> - <line number="33" hits="0"/> - <line number="34" hits="0"/> - <line number="35" hits="0"/> - <line number="36" hits="0"/> - <line number="42" hits="0"/> - <line number="44" hits="0"/> - <line number="49" hits="0"/> - <line number="51" hits="0"/> + <line number="33" hits="1"/> + <line number="34" hits="1"/> + <line number="35" hits="1"/> + <line number="36" hits="1"/> + <line number="42" hits="1"/> + <line number="44" hits="1"/> + <line number="49" hits="1"/> + <line number="51" hits="1"/> <line number="52" hits="0"/> - <line number="55" hits="0"/> - <line number="56" hits="0"/> - <line number="57" hits="0"/> - <line number="59" hits="0"/> + <line number="55" hits="1"/> + <line number="56" hits="1"/> + <line number="57" hits="1"/> + <line number="59" hits="1"/> <line number="61" hits="1"/> - <line number="68" hits="0"/> - <line number="69" hits="0"/> - <line number="70" hits="0"/> - <line number="72" hits="0"/> - <line number="76" hits="0"/> - <line number="77" hits="0"/> - <line number="79" hits="0"/> - <line number="80" hits="0"/> - <line number="83" hits="0"/> - <line number="84" hits="0"/> + <line number="68" hits="1"/> + <line number="69" hits="1"/> + <line number="70" hits="1"/> + <line number="72" hits="1"/> + <line number="76" hits="1"/> + <line number="77" hits="1"/> + <line number="79" hits="1"/> + <line number="80" hits="1"/> + <line number="83" hits="1"/> + <line number="84" hits="1"/> <line number="88" hits="0"/> - <line number="92" hits="0"/> - <line number="93" hits="0"/> - <line number="95" hits="0"/> - <line number="96" hits="0"/> - <line number="97" hits="0"/> + <line number="92" hits="1"/> + <line number="93" hits="1"/> + <line number="95" hits="1"/> + <line number="96" hits="1"/> + <line number="97" hits="1"/> <line number="99" hits="1"/> - <line number="100" hits="0"/> - <line number="101" hits="0"/> - <line number="104" hits="0"/> + <line number="100" hits="1"/> + <line number="101" hits="1"/> + <line number="104" hits="1"/> <line number="108" hits="1"/> - <line number="109" hits="0"/> - <line number="110" hits="0"/> - <line number="113" hits="0"/> - <line number="114" hits="0"/> + <line number="109" hits="1"/> + <line number="110" hits="1"/> + <line number="113" hits="1"/> + <line number="114" hits="1"/> </lines> </class> <class name="mistral.py" filename="encoders/mistral.py" complexity="0" line-rate="0.9512" branch-rate="0"> @@ -1042,14 +1134,14 @@ <line number="80" hits="1"/> </lines> </class> - <class name="vit.py" filename="encoders/vit.py" complexity="0" line-rate="0.4737" branch-rate="0"> + <class name="vit.py" filename="encoders/vit.py" complexity="0" line-rate="0.9254" branch-rate="0"> <methods/> <lines> <line number="1" hits="1"/> <line number="3" hits="1"/> - <line number="4" hits="1"/> <line number="5" hits="1"/> - <line number="7" hits="1"/> + <line number="8" hits="1"/> + <line number="9" hits="1"/> <line number="10" hits="1"/> <line number="11" hits="1"/> <line number="12" hits="1"/> @@ -1060,48 +1152,58 @@ <line number="17" hits="1"/> <line number="18" hits="1"/> <line number="19" hits="1"/> - <line number="20" hits="1"/> + <line number="21" hits="1"/> <line number="22" hits="1"/> <line number="23" hits="1"/> - <line number="24" hits="1"/> + <line number="25" hits="1"/> <line number="26" hits="1"/> <line number="27" hits="1"/> <line number="28" hits="1"/> <line number="29" hits="1"/> - <line number="30" hits="1"/> - <line number="36" hits="0"/> - <line number="37" hits="0"/> - <line number="38" hits="0"/> - <line number="39" hits="0"/> - <line number="40" hits="0"/> - <line number="46" hits="0"/> + <line number="35" hits="1"/> + <line number="36" hits="1"/> + <line number="37" hits="1"/> + <line number="38" hits="1"/> + <line number="39" hits="1"/> + <line number="45" hits="1"/> + <line number="46" hits="1"/> <line number="47" hits="0"/> - <line number="49" hits="0"/> - <line number="53" hits="0"/> - <line number="55" hits="0"/> - <line number="56" hits="0"/> - <line number="59" hits="0"/> - <line number="60" hits="0"/> - <line number="61" hits="0"/> - <line number="63" hits="0"/> + <line number="48" hits="0"/> + <line number="54" hits="1"/> + <line number="55" hits="1"/> + <line number="56" hits="1"/> + <line number="58" hits="1"/> + <line number="62" hits="1"/> + <line number="64" hits="1"/> <line number="65" hits="1"/> - <line number="66" hits="0"/> - <line number="67" hits="0"/> - <line number="68" hits="0"/> - <line number="69" hits="0"/> - <line number="71" hits="1"/> - <line number="72" hits="0"/> + <line number="67" hits="1"/> + <line number="69" hits="1"/> + <line number="70" hits="1"/> + <line number="71" hits="0"/> + <line number="72" hits="1"/> <line number="73" hits="0"/> - <line number="74" hits="0"/> - <line number="76" hits="1"/> - <line number="81" hits="0"/> - <line number="82" hits="0"/> - <line number="83" hits="0"/> - <line number="84" hits="0"/> - <line number="85" hits="0"/> - <line number="86" hits="0"/> - <line number="92" hits="0"/> - <line number="93" hits="0"/> + <line number="74" hits="1"/> + <line number="75" hits="1"/> + <line number="77" hits="0"/> + <line number="78" hits="1"/> + <line number="80" hits="1"/> + <line number="81" hits="1"/> + <line number="82" hits="1"/> + <line number="83" hits="1"/> + <line number="84" hits="1"/> + <line number="86" hits="1"/> + <line number="87" hits="1"/> + <line number="88" hits="1"/> + <line number="89" hits="1"/> + <line number="91" hits="1"/> + <line number="96" hits="1"/> + <line number="97" hits="1"/> + <line number="98" hits="1"/> + <line number="99" hits="1"/> + <line number="100" hits="1"/> + <line number="101" hits="1"/> + <line number="107" hits="1"/> + <line number="108" hits="1"/> </lines> </class> <class name="zure.py" filename="encoders/zure.py" complexity="0" line-rate="0.9589" branch-rate="0"> @@ -1184,7 +1286,7 @@ </class> </classes> </package> - <package name="index" line-rate="0.4842" branch-rate="0" complexity="0"> + <package name="index" line-rate="0.4932" branch-rate="0" complexity="0"> <classes> <class name="__init__.py" filename="index/__init__.py" complexity="0" line-rate="1" branch-rate="0"> <methods/> @@ -1212,13 +1314,13 @@ <line number="31" hits="1"/> <line number="36" hits="0"/> <line number="38" hits="1"/> - <line number="43" hits="0"/> - <line number="45" hits="1"/> - <line number="50" hits="0"/> - <line number="52" hits="1"/> - <line number="57" hits="0"/> - <line number="59" hits="1"/> + <line number="44" hits="0"/> + <line number="46" hits="1"/> + <line number="51" hits="0"/> + <line number="53" hits="1"/> + <line number="58" hits="0"/> <line number="60" hits="1"/> + <line number="61" hits="1"/> </lines> </class> <class name="local.py" filename="index/local.py" complexity="0" line-rate="0.8519" branch-rate="0"> @@ -1233,54 +1335,54 @@ <line number="16" hits="1"/> <line number="17" hits="1"/> <line number="19" hits="1"/> - <line number="20" hits="1"/> - <line number="22" hits="1"/> - <line number="25" hits="1"/> + <line number="21" hits="1"/> + <line number="23" hits="1"/> <line number="26" hits="1"/> <line number="27" hits="1"/> <line number="28" hits="1"/> - <line number="30" hits="0"/> - <line number="31" hits="1"/> + <line number="29" hits="1"/> + <line number="31" hits="0"/> <line number="32" hits="1"/> <line number="33" hits="1"/> <line number="34" hits="1"/> - <line number="36" hits="1"/> + <line number="35" hits="1"/> <line number="37" hits="1"/> <line number="38" hits="1"/> - <line number="40" hits="1"/> - <line number="47" hits="0"/> + <line number="39" hits="1"/> + <line number="41" hits="1"/> <line number="48" hits="0"/> <line number="49" hits="0"/> - <line number="51" hits="1"/> + <line number="50" hits="0"/> <line number="52" hits="1"/> - <line number="58" hits="1"/> - <line number="62" hits="1"/> + <line number="53" hits="1"/> + <line number="59" hits="1"/> <line number="63" hits="1"/> <line number="64" hits="1"/> - <line number="66" hits="1"/> - <line number="68" hits="1"/> + <line number="65" hits="1"/> + <line number="67" hits="1"/> <line number="69" hits="1"/> - <line number="71" hits="1"/> - <line number="75" hits="1"/> - <line number="80" hits="1"/> + <line number="70" hits="1"/> + <line number="72" hits="1"/> + <line number="76" hits="1"/> <line number="81" hits="1"/> <line number="82" hits="1"/> <line number="83" hits="1"/> - <line number="85" hits="0"/> - <line number="89" hits="1"/> - <line number="93" hits="0"/> - <line number="95" hits="1"/> + <line number="84" hits="1"/> + <line number="86" hits="0"/> + <line number="91" hits="1"/> + <line number="95" hits="0"/> <line number="97" hits="1"/> - <line number="98" hits="0"/> <line number="99" hits="1"/> - <line number="100" hits="1"/> + <line number="100" hits="0"/> + <line number="101" hits="1"/> <line number="102" hits="1"/> - <line number="103" hits="1"/> <line number="104" hits="1"/> - <line number="106" hits="0"/> + <line number="105" hits="1"/> + <line number="106" hits="1"/> + <line number="108" hits="0"/> </lines> </class> - <class name="pinecone.py" filename="index/pinecone.py" complexity="0" line-rate="0.2887" branch-rate="0"> + <class name="pinecone.py" filename="index/pinecone.py" complexity="0" line-rate="0.3" branch-rate="0"> <methods/> <lines> <line number="1" hits="1"/> @@ -1301,7 +1403,7 @@ <line number="22" hits="1"/> <line number="24" hits="1"/> <line number="25" hits="0"/> - <line number="27" hits="0"/> + <line number="26" hits="0"/> <line number="28" hits="0"/> <line number="29" hits="0"/> <line number="31" hits="1"/> @@ -1320,116 +1422,114 @@ <line number="51" hits="1"/> <line number="52" hits="0"/> <line number="53" hits="0"/> + <line number="54" hits="0"/> <line number="55" hits="0"/> - <line number="56" hits="0"/> - <line number="57" hits="0"/> + <line number="57" hits="1"/> <line number="58" hits="0"/> - <line number="60" hits="0"/> - <line number="62" hits="1"/> + <line number="59" hits="0"/> + <line number="61" hits="0"/> + <line number="62" hits="0"/> <line number="63" hits="0"/> - <line number="64" hits="0"/> - <line number="66" hits="0"/> - <line number="67" hits="0"/> <line number="68" hits="0"/> - <line number="73" hits="0"/> + <line number="69" hits="0"/> + <line number="70" hits="0"/> + <line number="71" hits="0"/> + <line number="73" hits="1"/> <line number="74" hits="0"/> <line number="75" hits="0"/> <line number="76" hits="0"/> - <line number="78" hits="1"/> <line number="79" hits="0"/> - <line number="80" hits="0"/> - <line number="81" hits="0"/> - <line number="84" hits="0"/> - <line number="91" hits="0"/> + <line number="86" hits="0"/> + <line number="87" hits="0"/> + <line number="88" hits="0"/> + <line number="89" hits="0"/> + <line number="90" hits="0"/> <line number="92" hits="0"/> - <line number="93" hits="0"/> <line number="94" hits="0"/> <line number="95" hits="0"/> - <line number="97" hits="0"/> - <line number="99" hits="0"/> - <line number="100" hits="0"/> - <line number="101" hits="0"/> - <line number="107" hits="0"/> - <line number="108" hits="0"/> - <line number="109" hits="0"/> + <line number="96" hits="0"/> + <line number="102" hits="0"/> + <line number="103" hits="0"/> + <line number="104" hits="0"/> + <line number="105" hits="0"/> + <line number="106" hits="0"/> + <line number="108" hits="1"/> <line number="110" hits="0"/> <line number="111" hits="0"/> - <line number="113" hits="1"/> - <line number="116" hits="0"/> - <line number="117" hits="0"/> - <line number="119" hits="0"/> - <line number="120" hits="0"/> - <line number="121" hits="0"/> - <line number="122" hits="0"/> + <line number="113" hits="0"/> + <line number="115" hits="1"/> <line number="123" hits="0"/> <line number="124" hits="0"/> <line number="125" hits="0"/> <line number="127" hits="0"/> - <line number="129" hits="1"/> - <line number="130" hits="0"/> - <line number="131" hits="0"/> <line number="132" hits="0"/> - <line number="134" hits="1"/> + <line number="133" hits="0"/> + <line number="134" hits="0"/> + <line number="136" hits="1"/> + <line number="137" hits="0"/> <line number="138" hits="0"/> <line number="139" hits="0"/> - <line number="140" hits="0"/> - <line number="141" hits="0"/> - <line number="143" hits="0"/> - <line number="144" hits="0"/> + <line number="141" hits="1"/> + <line number="145" hits="0"/> <line number="146" hits="0"/> - <line number="149" hits="0"/> + <line number="147" hits="0"/> + <line number="148" hits="0"/> <line number="150" hits="0"/> <line number="151" hits="0"/> - <line number="152" hits="0"/> - <line number="154" hits="0"/> - <line number="155" hits="0"/> + <line number="153" hits="0"/> <line number="156" hits="0"/> + <line number="157" hits="0"/> + <line number="158" hits="0"/> <line number="159" hits="0"/> - <line number="160" hits="0"/> + <line number="161" hits="0"/> + <line number="162" hits="0"/> <line number="163" hits="0"/> - <line number="165" hits="0"/> <line number="166" hits="0"/> <line number="167" hits="0"/> <line number="170" hits="0"/> - <line number="171" hits="0"/> + <line number="172" hits="0"/> <line number="173" hits="0"/> - <line number="176" hits="0"/> + <line number="174" hits="0"/> <line number="177" hits="0"/> <line number="178" hits="0"/> <line number="180" hits="0"/> - <line number="182" hits="1"/> - <line number="190" hits="0"/> - <line number="191" hits="0"/> - <line number="192" hits="0"/> - <line number="194" hits="1"/> - <line number="195" hits="0"/> - <line number="196" hits="0"/> + <line number="183" hits="0"/> + <line number="184" hits="0"/> + <line number="185" hits="0"/> + <line number="187" hits="0"/> + <line number="189" hits="1"/> <line number="197" hits="0"/> + <line number="198" hits="0"/> <line number="199" hits="0"/> <line number="201" hits="1"/> <line number="202" hits="0"/> - <line number="204" hits="1"/> - <line number="205" hits="0"/> + <line number="203" hits="0"/> + <line number="204" hits="0"/> <line number="206" hits="0"/> - <line number="207" hits="0"/> + <line number="208" hits="1"/> + <line number="209" hits="0"/> + <line number="211" hits="1"/> + <line number="212" hits="0"/> <line number="213" hits="0"/> - <line number="215" hits="1"/> - <line number="216" hits="0"/> - <line number="217" hits="0"/> - <line number="218" hits="0"/> - <line number="219" hits="0"/> + <line number="214" hits="0"/> + <line number="220" hits="0"/> + <line number="222" hits="1"/> + <line number="223" hits="0"/> <line number="224" hits="0"/> <line number="225" hits="0"/> <line number="226" hits="0"/> - <line number="228" hits="1"/> - <line number="229" hits="0"/> - <line number="231" hits="1"/> + <line number="231" hits="0"/> <line number="232" hits="0"/> + <line number="233" hits="0"/> + <line number="235" hits="1"/> + <line number="236" hits="0"/> + <line number="238" hits="1"/> + <line number="239" hits="0"/> </lines> </class> </classes> </package> - <package name="llms" line-rate="0.7321" branch-rate="0" complexity="0"> + <package name="llms" line-rate="0.915" branch-rate="0" complexity="0"> <classes> <class name="__init__.py" filename="llms/__init__.py" complexity="0" line-rate="1" branch-rate="0"> <methods/> @@ -1443,7 +1543,7 @@ <line number="8" hits="1"/> </lines> </class> - <class name="base.py" filename="llms/base.py" complexity="0" line-rate="0.7317" branch-rate="0"> + <class name="base.py" filename="llms/base.py" complexity="0" line-rate="0.9756" branch-rate="0"> <methods/> <lines> <line number="1" hits="1"/> @@ -1467,8 +1567,8 @@ <line number="31" hits="1"/> <line number="34" hits="1"/> <line number="35" hits="1"/> - <line number="36" hits="0"/> - <line number="37" hits="0"/> + <line number="36" hits="1"/> + <line number="37" hits="1"/> <line number="38" hits="1"/> <line number="39" hits="1"/> <line number="40" hits="1"/> @@ -1476,17 +1576,17 @@ <line number="43" hits="1"/> <line number="46" hits="1"/> <line number="48" hits="1"/> - <line number="76" hits="1"/> - <line number="77" hits="1"/> - <line number="79" hits="0"/> - <line number="80" hits="0"/> - <line number="82" hits="0"/> - <line number="83" hits="0"/> - <line number="84" hits="0"/> - <line number="85" hits="0"/> - <line number="86" hits="0"/> - <line number="87" hits="0"/> - <line number="88" hits="0"/> + <line number="98" hits="1"/> + <line number="99" hits="1"/> + <line number="101" hits="1"/> + <line number="102" hits="0"/> + <line number="104" hits="1"/> + <line number="105" hits="1"/> + <line number="106" hits="1"/> + <line number="107" hits="1"/> + <line number="108" hits="1"/> + <line number="109" hits="1"/> + <line number="110" hits="1"/> </lines> </class> <class name="cohere.py" filename="llms/cohere.py" complexity="0" line-rate="0.9655" branch-rate="0"> @@ -1523,49 +1623,49 @@ <line number="48" hits="1"/> </lines> </class> - <class name="llamacpp.py" filename="llms/llamacpp.py" complexity="0" line-rate="0" branch-rate="0"> + <class name="llamacpp.py" filename="llms/llamacpp.py" complexity="0" line-rate="0.9" branch-rate="0"> <methods/> <lines> - <line number="1" hits="0"/> - <line number="2" hits="0"/> - <line number="3" hits="0"/> - <line number="5" hits="0"/> - <line number="7" hits="0"/> - <line number="8" hits="0"/> - <line number="9" hits="0"/> - <line number="12" hits="0"/> - <line number="13" hits="0"/> - <line number="14" hits="0"/> - <line number="15" hits="0"/> - <line number="16" hits="0"/> - <line number="18" hits="0"/> - <line number="26" hits="0"/> - <line number="33" hits="0"/> - <line number="34" hits="0"/> - <line number="35" hits="0"/> - <line number="36" hits="0"/> - <line number="38" hits="0"/> - <line number="42" hits="0"/> - <line number="43" hits="0"/> - <line number="50" hits="0"/> - <line number="51" hits="0"/> - <line number="53" hits="0"/> + <line number="1" hits="1"/> + <line number="2" hits="1"/> + <line number="3" hits="1"/> + <line number="5" hits="1"/> + <line number="7" hits="1"/> + <line number="8" hits="1"/> + <line number="9" hits="1"/> + <line number="12" hits="1"/> + <line number="13" hits="1"/> + <line number="14" hits="1"/> + <line number="15" hits="1"/> + <line number="16" hits="1"/> + <line number="18" hits="1"/> + <line number="26" hits="1"/> + <line number="33" hits="1"/> + <line number="34" hits="1"/> + <line number="35" hits="1"/> + <line number="36" hits="1"/> + <line number="38" hits="1"/> + <line number="42" hits="1"/> + <line number="43" hits="1"/> + <line number="50" hits="1"/> + <line number="51" hits="1"/> + <line number="53" hits="1"/> <line number="54" hits="0"/> - <line number="55" hits="0"/> + <line number="55" hits="1"/> <line number="56" hits="0"/> <line number="57" hits="0"/> <line number="58" hits="0"/> - <line number="60" hits="0"/> - <line number="61" hits="0"/> - <line number="62" hits="0"/> - <line number="63" hits="0"/> - <line number="64" hits="0"/> - <line number="65" hits="0"/> - <line number="66" hits="0"/> - <line number="68" hits="0"/> - <line number="70" hits="0"/> - <line number="73" hits="0"/> - <line number="74" hits="0"/> + <line number="60" hits="1"/> + <line number="61" hits="1"/> + <line number="62" hits="1"/> + <line number="63" hits="1"/> + <line number="64" hits="1"/> + <line number="65" hits="1"/> + <line number="66" hits="1"/> + <line number="68" hits="1"/> + <line number="70" hits="1"/> + <line number="73" hits="1"/> + <line number="74" hits="1"/> </lines> </class> <class name="mistral.py" filename="llms/mistral.py" complexity="0" line-rate="0.8333" branch-rate="0"> @@ -1609,6 +1709,40 @@ <line number="57" hits="0"/> </lines> </class> + <class name="ollama.py" filename="llms/ollama.py" complexity="0" line-rate="1" branch-rate="0"> + <methods/> + <lines> + <line number="1" hits="1"/> + <line number="3" hits="1"/> + <line number="5" hits="1"/> + <line number="6" hits="1"/> + <line number="7" hits="1"/> + <line number="10" hits="1"/> + <line number="11" hits="1"/> + <line number="12" hits="1"/> + <line number="13" hits="1"/> + <line number="14" hits="1"/> + <line number="16" hits="1"/> + <line number="24" hits="1"/> + <line number="25" hits="1"/> + <line number="26" hits="1"/> + <line number="27" hits="1"/> + <line number="28" hits="1"/> + <line number="30" hits="1"/> + <line number="39" hits="1"/> + <line number="40" hits="1"/> + <line number="41" hits="1"/> + <line number="42" hits="1"/> + <line number="44" hits="1"/> + <line number="45" hits="1"/> + <line number="52" hits="1"/> + <line number="53" hits="1"/> + <line number="55" hits="1"/> + <line number="56" hits="1"/> + <line number="57" hits="1"/> + <line number="58" hits="1"/> + </lines> + </class> <class name="openai.py" filename="llms/openai.py" complexity="0" line-rate="0.8889" branch-rate="0"> <methods/> <lines> @@ -1738,24 +1872,47 @@ </class> </classes> </package> - <package name="splitters" line-rate="1" branch-rate="0" complexity="0"> + <package name="splitters" line-rate="0.374" branch-rate="0" complexity="0"> <classes> <class name="__init__.py" filename="splitters/__init__.py" complexity="0" line-rate="1" branch-rate="0"> <methods/> - <lines/> + <lines> + <line number="1" hits="1"/> + <line number="2" hits="1"/> + <line number="3" hits="1"/> + <line number="4" hits="1"/> + <line number="6" hits="1"/> + </lines> </class> - <class name="base.py" filename="splitters/base.py" complexity="0" line-rate="1" branch-rate="0"> + <class name="base.py" filename="splitters/base.py" complexity="0" line-rate="0.5" branch-rate="0"> <methods/> <lines> <line number="1" hits="1"/> <line number="3" hits="1"/> - <line number="5" hits="1"/> - <line number="8" hits="1"/> - <line number="9" hits="1"/> + <line number="4" hits="1"/> + <line number="6" hits="1"/> + <line number="7" hits="1"/> <line number="10" hits="1"/> <line number="11" hits="1"/> - <line number="13" hits="1"/> + <line number="12" hits="1"/> <line number="14" hits="1"/> + <line number="15" hits="1"/> + <line number="17" hits="1"/> + <line number="18" hits="1"/> + <line number="20" hits="1"/> + <line number="21" hits="0"/> + <line number="22" hits="0"/> + <line number="23" hits="0"/> + <line number="24" hits="0"/> + <line number="25" hits="0"/> + <line number="26" hits="0"/> + <line number="27" hits="0"/> + <line number="28" hits="0"/> + <line number="30" hits="0"/> + <line number="31" hits="0"/> + <line number="36" hits="0"/> + <line number="37" hits="0"/> + <line number="38" hits="0"/> </lines> </class> <class name="consecutive_sim.py" filename="splitters/consecutive_sim.py" complexity="0" line-rate="1" branch-rate="0"> @@ -1770,24 +1927,25 @@ <line number="16" hits="1"/> <line number="22" hits="1"/> <line number="23" hits="1"/> - <line number="25" hits="1"/> - <line number="27" hits="1"/> + <line number="24" hits="1"/> + <line number="26" hits="1"/> <line number="28" hits="1"/> - <line number="32" hits="1"/> + <line number="29" hits="1"/> <line number="33" hits="1"/> <line number="34" hits="1"/> <line number="35" hits="1"/> <line number="36" hits="1"/> <line number="37" hits="1"/> <line number="38" hits="1"/> - <line number="40" hits="1"/> + <line number="39" hits="1"/> <line number="41" hits="1"/> <line number="42" hits="1"/> <line number="43" hits="1"/> - <line number="50" hits="1"/> + <line number="44" hits="1"/> <line number="51" hits="1"/> <line number="52" hits="1"/> <line number="53" hits="1"/> + <line number="54" hits="1"/> </lines> </class> <class name="cumulative_sim.py" filename="splitters/cumulative_sim.py" complexity="0" line-rate="1" branch-rate="0"> @@ -1802,27 +1960,210 @@ <line number="16" hits="1"/> <line number="22" hits="1"/> <line number="23" hits="1"/> - <line number="25" hits="1"/> + <line number="24" hits="1"/> <line number="26" hits="1"/> - <line number="28" hits="1"/> + <line number="27" hits="1"/> <line number="29" hits="1"/> - <line number="32" hits="1"/> - <line number="33" hits="1"/> + <line number="30" hits="1"/> + <line number="34" hits="1"/> <line number="35" hits="1"/> - <line number="36" hits="1"/> <line number="37" hits="1"/> + <line number="38" hits="1"/> <line number="39" hits="1"/> <line number="42" hits="1"/> - <line number="43" hits="1"/> <line number="46" hits="1"/> <line number="47" hits="1"/> - <line number="48" hits="1"/> - <line number="53" hits="1"/> - <line number="54" hits="1"/> - <line number="61" hits="1"/> - <line number="66" hits="1"/> - <line number="67" hits="1"/> - <line number="69" hits="1"/> + <line number="50" hits="1"/> + <line number="51" hits="1"/> + <line number="52" hits="1"/> + <line number="57" hits="1"/> + <line number="58" hits="1"/> + <line number="65" hits="1"/> + <line number="70" hits="1"/> + <line number="71" hits="1"/> + <line number="73" hits="1"/> + </lines> + </class> + <class name="rolling_window.py" filename="splitters/rolling_window.py" complexity="0" line-rate="0.1056" branch-rate="0"> + <methods/> + <lines> + <line number="1" hits="1"/> + <line number="3" hits="1"/> + <line number="5" hits="1"/> + <line number="6" hits="1"/> + <line number="7" hits="1"/> + <line number="8" hits="1"/> + <line number="9" hits="1"/> + <line number="12" hits="1"/> + <line number="13" hits="1"/> + <line number="25" hits="0"/> + <line number="26" hits="0"/> + <line number="27" hits="0"/> + <line number="28" hits="0"/> + <line number="29" hits="0"/> + <line number="30" hits="0"/> + <line number="31" hits="0"/> + <line number="32" hits="0"/> + <line number="33" hits="0"/> + <line number="34" hits="0"/> + <line number="36" hits="1"/> + <line number="37" hits="0"/> + <line number="38" hits="0"/> + <line number="39" hits="0"/> + <line number="40" hits="0"/> + <line number="41" hits="0"/> + <line number="42" hits="0"/> + <line number="44" hits="1"/> + <line number="45" hits="0"/> + <line number="46" hits="0"/> + <line number="47" hits="0"/> + <line number="48" hits="0"/> + <line number="49" hits="0"/> + <line number="53" hits="0"/> + <line number="54" hits="0"/> + <line number="56" hits="1"/> + <line number="57" hits="0"/> + <line number="58" hits="0"/> + <line number="59" hits="0"/> + <line number="60" hits="0"/> + <line number="61" hits="0"/> + <line number="63" hits="1"/> + <line number="64" hits="0"/> + <line number="65" hits="0"/> + <line number="66" hits="0"/> + <line number="69" hits="0"/> + <line number="70" hits="0"/> + <line number="73" hits="0"/> + <line number="74" hits="0"/> + <line number="76" hits="0"/> + <line number="77" hits="0"/> + <line number="78" hits="0"/> + <line number="79" hits="0"/> + <line number="82" hits="0"/> + <line number="85" hits="0"/> + <line number="93" hits="0"/> + <line number="94" hits="0"/> + <line number="97" hits="0"/> + <line number="102" hits="0"/> + <line number="108" hits="0"/> + <line number="109" hits="0"/> + <line number="110" hits="0"/> + <line number="111" hits="0"/> + <line number="113" hits="0"/> + <line number="114" hits="0"/> + <line number="115" hits="0"/> + <line number="117" hits="0"/> + <line number="118" hits="0"/> + <line number="120" hits="1"/> + <line number="131" hits="0"/> + <line number="132" hits="0"/> + <line number="133" hits="0"/> + <line number="135" hits="0"/> + <line number="136" hits="0"/> + <line number="139" hits="0"/> + <line number="140" hits="0"/> + <line number="143" hits="0"/> + <line number="144" hits="0"/> + <line number="146" hits="0"/> + <line number="149" hits="0"/> + <line number="157" hits="0"/> + <line number="161" hits="0"/> + <line number="162" hits="0"/> + <line number="165" hits="0"/> + <line number="166" hits="0"/> + <line number="167" hits="0"/> + <line number="175" hits="0"/> + <line number="179" hits="0"/> + <line number="181" hits="0"/> + <line number="182" hits="0"/> + <line number="185" hits="0"/> + <line number="186" hits="0"/> + <line number="194" hits="0"/> + <line number="200" hits="0"/> + <line number="201" hits="0"/> + <line number="204" hits="0"/> + <line number="205" hits="0"/> + <line number="208" hits="0"/> + <line number="212" hits="0"/> + <line number="214" hits="1"/> + <line number="217" hits="0"/> + <line number="218" hits="0"/> + <line number="219" hits="0"/> + <line number="220" hits="0"/> + <line number="224" hits="0"/> + <line number="226" hits="0"/> + <line number="227" hits="0"/> + <line number="228" hits="0"/> + <line number="229" hits="0"/> + <line number="230" hits="0"/> + <line number="231" hits="0"/> + <line number="237" hits="0"/> + <line number="245" hits="0"/> + <line number="246" hits="0"/> + <line number="254" hits="0"/> + <line number="255" hits="0"/> + <line number="256" hits="0"/> + <line number="262" hits="0"/> + <line number="263" hits="0"/> + <line number="264" hits="0"/> + <line number="266" hits="1"/> + <line number="269" hits="0"/> + <line number="270" hits="0"/> + <line number="271" hits="0"/> + <line number="272" hits="0"/> + <line number="273" hits="0"/> + <line number="274" hits="0"/> + <line number="281" hits="0"/> + <line number="282" hits="0"/> + <line number="283" hits="0"/> + <line number="285" hits="0"/> + <line number="286" hits="0"/> + <line number="289" hits="0"/> + <line number="294" hits="0"/> + <line number="296" hits="0"/> + <line number="297" hits="0"/> + <line number="298" hits="0"/> + <line number="299" hits="0"/> + <line number="300" hits="0"/> + <line number="301" hits="0"/> + <line number="302" hits="0"/> + <line number="303" hits="0"/> + <line number="305" hits="0"/> + <line number="306" hits="0"/> + <line number="307" hits="0"/> + <line number="312" hits="1"/> + <line number="313" hits="0"/> + <line number="314" hits="0"/> + <line number="316" hits="0"/> + <line number="317" hits="0"/> + <line number="318" hits="0"/> + <line number="319" hits="0"/> + <line number="324" hits="0"/> + <line number="325" hits="0"/> + <line number="326" hits="0"/> + <line number="327" hits="0"/> + <line number="329" hits="0"/> + <line number="330" hits="0"/> + <line number="331" hits="0"/> + <line number="332" hits="0"/> + <line number="333" hits="0"/> + <line number="334" hits="0"/> + </lines> + </class> + <class name="utils.py" filename="splitters/utils.py" complexity="0" line-rate="0.3636" branch-rate="0"> + <methods/> + <lines> + <line number="1" hits="1"/> + <line number="2" hits="1"/> + <line number="5" hits="1"/> + <line number="22" hits="0"/> + <line number="55" hits="0"/> + <line number="56" hits="0"/> + <line number="57" hits="0"/> + <line number="60" hits="1"/> + <line number="61" hits="0"/> + <line number="62" hits="0"/> + <line number="63" hits="0"/> </lines> </class> </classes> @@ -1928,4 +2269,4 @@ </classes> </package> </packages> -</coverage> \ No newline at end of file +</coverage> diff --git a/semantic_router/encoders/clip.py b/semantic_router/encoders/clip.py index a910a999393b79bc39aa532edd9fa153f25d2651..89a85cf3ff7cce83c11456ecee5345d532da6c37 100644 --- a/semantic_router/encoders/clip.py +++ b/semantic_router/encoders/clip.py @@ -7,7 +7,7 @@ from semantic_router.encoders import BaseEncoder class CLIPEncoder(BaseEncoder): - name: str = "openai/clip-vit-base-patch32" + name: str = "openai/clip-vit-base-patch16" type: str = "huggingface" score_threshold: float = 0.2 tokenizer_kwargs: dict = {} diff --git a/tests/unit/encoders/test_clip.py b/tests/unit/encoders/test_clip.py index f861f1ee9fc9e2a8fbd855e4b7bdd7f61d41c3ed..2a9e814cbee7472ab3d93c6c8a49b78b77633e58 100644 --- a/tests/unit/encoders/test_clip.py +++ b/tests/unit/encoders/test_clip.py @@ -1,15 +1,23 @@ import numpy as np import pytest from PIL import Image +import torch from semantic_router.encoders import CLIPEncoder clip_encoder = CLIPEncoder() +if torch.cuda.is_available(): + device = "cuda" +elif torch.backends.mps.is_available(): + device = "mps" +else: + device = "cpu" + @pytest.fixture() def dummy_pil_image(): - return Image.fromarray(np.random.rand(1024, 512, 3).astype(np.uint8)) + return Image.fromarray(np.random.rand(512, 224, 3).astype(np.uint8)) @pytest.fixture() @@ -34,10 +42,10 @@ class TestVitEncoder: CLIPEncoder() def test_clip_encoder_initialization(self): - assert clip_encoder.name == "openai/clip-vit-base-patch32" + assert clip_encoder.name == "openai/clip-vit-base-patch16" assert clip_encoder.type == "huggingface" assert clip_encoder.score_threshold == 0.2 - assert clip_encoder.device == "cpu" + assert clip_encoder.device == device def test_clip_encoder_call_text(self): embeddings = clip_encoder(["hello", "world"]) @@ -59,7 +67,7 @@ class TestVitEncoder: def test_clip_device(self): device = clip_encoder._model.device.type - assert device == "cpu" + assert device == device def test_clip_encoder_ensure_rgb(self, dummy_black_and_white_img): rgb_image = clip_encoder._ensure_rgb(dummy_black_and_white_img) diff --git a/tests/unit/encoders/test_vit.py b/tests/unit/encoders/test_vit.py index 0435058c0cdbcf776b5a44356a085d82c9c30439..01aa32883ad4086d12eca953cc6b91b214551958 100644 --- a/tests/unit/encoders/test_vit.py +++ b/tests/unit/encoders/test_vit.py @@ -1,11 +1,19 @@ import numpy as np import pytest from PIL import Image +import torch from semantic_router.encoders import VitEncoder vit_encoder = VitEncoder() +if torch.cuda.is_available(): + device = "cuda" +elif torch.backends.mps.is_available(): + device = "mps" +else: + device = "cpu" + @pytest.fixture() def dummy_pil_image(): @@ -42,7 +50,7 @@ class TestVitEncoder: assert vit_encoder.name == "google/vit-base-patch16-224" assert vit_encoder.type == "huggingface" assert vit_encoder.score_threshold == 0.5 - assert vit_encoder.device == "cpu" + assert vit_encoder.device == device def test_vit_encoder_call(self, dummy_pil_image): encoded_images = vit_encoder([dummy_pil_image] * 3) @@ -59,7 +67,7 @@ class TestVitEncoder: def test_vit_encoder_process_images_device(self, dummy_pil_image): imgs = vit_encoder._process_images([dummy_pil_image] * 3)["pixel_values"] - assert imgs.device.type == "cpu" + assert imgs.device.type == device def test_vit_encoder_ensure_rgb(self, dummy_black_and_white_img): rgb_image = vit_encoder._ensure_rgb(dummy_black_and_white_img)