diff --git a/coverage.xml b/coverage.xml
index 5850def2dfb8bf3ee91ce0e20a28f5312ec8c0b2..f6013daa21871d362b16d3f529b6f5a06feb834c 100644
--- a/coverage.xml
+++ b/coverage.xml
@@ -1,12 +1,12 @@
<?xml version="1.0" ?>
-<coverage version="7.4.0" timestamp="1704881517474" lines-valid="893" lines-covered="778" line-rate="0.8712" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0">
- <!-- Generated by coverage.py: https://coverage.readthedocs.io/en/7.4.0 -->
+<coverage version="7.3.3" timestamp="1706806167202" lines-valid="1260" lines-covered="1115" line-rate="0.8849" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0">
+ <!-- Generated by coverage.py: https://coverage.readthedocs.io/en/7.3.3 -->
<!-- Based on https://raw.githubusercontent.com/cobertura/web/master/htdocs/xml/coverage-04.dtd -->
<sources>
- <source>/Users/jakit/customers/aurelio/semantic-router/semantic_router</source>
+ <source>/Users/johnny_silicio/aurelio_ai/semantic-router/semantic_router</source>
</sources>
<packages>
- <package name="." line-rate="0.9263" branch-rate="0" complexity="0">
+ <package name="." line-rate="0.8662" branch-rate="0" complexity="0">
<classes>
<class name="__init__.py" filename="__init__.py" complexity="0" line-rate="1" branch-rate="0">
<methods/>
@@ -15,310 +15,438 @@
<line number="2" hits="1"/>
<line number="3" hits="1"/>
<line number="5" hits="1"/>
+ <line number="7" hits="1"/>
</lines>
</class>
- <class name="hybrid_layer.py" filename="hybrid_layer.py" complexity="0" line-rate="0.9796" branch-rate="0">
+ <class name="hybrid_layer.py" filename="hybrid_layer.py" complexity="0" line-rate="0.9808" branch-rate="0">
<methods/>
<lines>
<line number="1" hits="1"/>
- <line number="2" hits="1"/>
+ <line number="3" hits="1"/>
<line number="4" hits="1"/>
- <line number="8" hits="1"/>
- <line number="9" hits="1"/>
- <line number="10" hits="1"/>
- <line number="13" hits="1"/>
- <line number="14" hits="1"/>
+ <line number="6" hits="1"/>
+ <line number="11" hits="1"/>
+ <line number="12" 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="26" hits="1"/>
- <line number="27" hits="1"/>
+ <line number="21" hits="1"/>
+ <line number="28" hits="1"/>
<line number="29" hits="1"/>
- <line number="30" hits="0"/>
- <line number="31" hits="0"/>
- <line number="33" hits="1"/>
+ <line number="31" hits="1"/>
+ <line number="32" hits="0"/>
+ <line number="33" hits="0"/>
<line number="35" hits="1"/>
<line number="37" hits="1"/>
- <line number="41" hits="1"/>
- <line number="43" hits="1"/>
+ <line number="38" hits="1"/>
+ <line number="39" hits="1"/>
+ <line number="42" 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="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="57" hits="1"/>
- <line number="58" hits="1"/>
+ <line number="59" hits="1"/>
+ <line number="60" hits="1"/>
+ <line number="62" hits="1"/>
<line number="63" hits="1"/>
- <line number="64" hits="1"/>
<line number="65" hits="1"/>
<line number="67" hits="1"/>
- <line number="68" hits="1"/>
- <line number="69" hits="1"/>
+ <line number="70" hits="1"/>
+ <line number="72" hits="1"/>
<line number="73" hits="1"/>
- <line number="74" hits="1"/>
<line number="76" hits="1"/>
<line number="78" hits="1"/>
- <line number="79" hits="1"/>
<line number="81" hits="1"/>
- <line number="83" hits="1"/>
+ <line number="82" hits="1"/>
+ <line number="84" hits="1"/>
<line number="85" hits="1"/>
<line number="86" hits="1"/>
- <line number="89" hits="1"/>
+ <line number="88" hits="1"/>
<line number="90" hits="1"/>
- <line number="93" hits="1"/>
+ <line number="91" hits="1"/>
<line number="94" hits="1"/>
<line number="95" hits="1"/>
- <line number="102" hits="1"/>
+ <line number="98" hits="1"/>
+ <line number="99" hits="1"/>
+ <line number="100" hits="1"/>
+ <line number="106" hits="1"/>
+ <line number="107" hits="1"/>
<line number="109" hits="1"/>
<line number="115" hits="1"/>
- <line number="120" hits="1"/>
- <line number="121" hits="1"/>
- <line number="123" hits="1"/>
+ <line number="116" hits="1"/>
+ <line number="118" hits="1"/>
<line number="124" hits="1"/>
- <line number="126" hits="1"/>
- <line number="128" hits="1"/>
+ <line number="129" hits="1"/>
<line number="130" hits="1"/>
- <line number="131" hits="1"/>
<line number="132" hits="1"/>
- <line number="134" hits="1"/>
+ <line number="133" hits="1"/>
<line number="135" hits="1"/>
- <line number="136" hits="1"/>
<line number="137" hits="1"/>
<line number="139" hits="1"/>
<line number="140" hits="1"/>
<line number="141" hits="1"/>
<line number="143" hits="1"/>
<line number="144" hits="1"/>
+ <line number="145" hits="1"/>
<line number="146" hits="1"/>
- <line number="147" hits="1"/>
+ <line number="148" hits="1"/>
<line number="149" hits="1"/>
- <line number="151" hits="1"/>
+ <line number="150" hits="1"/>
<line number="152" hits="1"/>
<line number="153" 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="163" hits="1"/>
+ <line number="162" hits="1"/>
+ <line number="164" hits="1"/>
+ <line number="165" hits="1"/>
<line number="166" hits="1"/>
<line number="167" hits="1"/>
+ <line number="168" hits="1"/>
+ <line number="169" hits="1"/>
<line number="170" hits="1"/>
- <line number="171" hits="1"/>
- <line number="173" hits="1"/>
- <line number="174" 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="183" hits="1"/>
+ <line number="185" hits="1"/>
+ <line number="186" hits="1"/>
+ <line number="187" hits="1"/>
+ <line number="189" hits="1"/>
</lines>
</class>
- <class name="layer.py" filename="layer.py" complexity="0" line-rate="0.8827" branch-rate="0">
+ <class name="layer.py" filename="layer.py" complexity="0" line-rate="0.795" 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="4" 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="12" hits="1"/>
<line number="13" hits="1"/>
- <line number="16" hits="1"/>
- <line number="17" hits="1"/>
+ <line number="14" hits="1"/>
+ <line number="15" 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="0"/>
- <line number="23" hits="0"/>
- <line number="24" hits="0"/>
+ <line number="22" hits="1"/>
+ <line number="24" 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="31" hits="1"/>
- <line number="32" hits="1"/>
- <line number="33" hits="0"/>
+ <line number="31" hits="0"/>
+ <line number="32" hits="0"/>
+ <line number="34" hits="1"/>
+ <line number="35" hits="1"/>
<line number="36" hits="0"/>
- <line number="38" hits="1"/>
<line number="39" hits="0"/>
- <line number="40" hits="0"/>
- <line number="41" hits="0"/>
- <line number="44" hits="1"/>
- <line number="50" hits="1"/>
- <line number="52" hits="1"/>
- <line number="58" hits="1"/>
- <line number="59" hits="1"/>
- <line number="63" hits="1"/>
- <line number="64" hits="0"/>
- <line number="65" hits="1"/>
- <line number="66" hits="0"/>
- <line number="67" hits="1"/>
- <line number="68" hits="0"/>
- <line number="69" hits="1"/>
- <line number="70" hits="0"/>
- <line number="71" hits="1"/>
+ <line number="41" hits="1"/>
+ <line number="42" hits="0"/>
+ <line number="43" hits="0"/>
+ <line number="44" hits="0"/>
+ <line number="47" hits="1"/>
+ <line number="53" hits="1"/>
+ <line number="55" hits="1"/>
+ <line number="61" hits="1"/>
+ <line number="62" hits="1"/>
+ <line number="66" hits="1"/>
+ <line number="67" hits="0"/>
+ <line number="68" hits="1"/>
+ <line number="69" hits="0"/>
+ <line number="70" hits="1"/>
+ <line number="71" hits="0"/>
<line number="72" hits="1"/>
- <line number="73" hits="1"/>
+ <line number="73" hits="0"/>
+ <line number="74" hits="1"/>
<line number="75" hits="1"/>
<line number="76" 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"/>
<line number="86" hits="1"/>
- <line number="90" hits="1"/>
- <line number="91" hits="1"/>
- <line number="92" hits="1"/>
+ <line number="87" hits="1"/>
+ <line number="89" hits="1"/>
<line number="93" hits="1"/>
<line number="94" hits="1"/>
<line number="95" hits="1"/>
- <line number="99" hits="1"/>
- <line number="101" hits="1"/>
- <line number="102" hits="1"/>
- <line number="108" hits="1"/>
- <line number="110" hits="1"/>
+ <line number="96" hits="1"/>
+ <line number="97" hits="1"/>
+ <line number="98" hits="1"/>
+ <line number="102" hits="0"/>
+ <line number="104" hits="1"/>
+ <line number="105" hits="1"/>
<line number="111" hits="1"/>
+ <line number="113" hits="1"/>
<line number="114" hits="1"/>
- <line number="115" hits="1"/>
- <line number="119" hits="1"/>
+ <line number="117" hits="1"/>
+ <line number="118" hits="1"/>
<line number="122" hits="1"/>
- <line number="123" hits="0"/>
<line number="125" hits="1"/>
- <line number="126" hits="1"/>
- <line number="127" hits="1"/>
+ <line number="126" hits="0"/>
<line number="128" hits="1"/>
<line number="129" hits="1"/>
+ <line number="130" hits="1"/>
<line number="131" hits="1"/>
<line number="132" hits="1"/>
- <line number="133" hits="1"/>
+ <line number="134" hits="1"/>
<line number="135" hits="1"/>
<line number="136" hits="1"/>
- <line number="137" hits="1"/>
<line number="138" hits="1"/>
<line number="139" hits="1"/>
<line number="140" hits="1"/>
+ <line number="141" hits="1"/>
<line number="142" hits="1"/>
<line number="143" hits="1"/>
- <line number="144" hits="0"/>
+ <line number="145" hits="1"/>
<line number="146" hits="1"/>
- <line number="147" hits="1"/>
+ <line number="147" hits="0"/>
+ <line number="149" hits="1"/>
<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="162" hits="1"/>
- <line number="163" hits="1"/>
- <line number="164" hits="1"/>
+ <line number="157" hits="1"/>
+ <line number="159" hits="1"/>
<line number="165" hits="1"/>
<line number="166" hits="1"/>
- <line number="170" hits="1"/>
- <line number="172" hits="1"/>
+ <line number="167" hits="1"/>
+ <line number="168" hits="1"/>
+ <line number="169" hits="1"/>
<line number="173" hits="1"/>
- <line number="174" hits="1"/>
<line number="175" hits="1"/>
+ <line number="176" hits="1"/>
<line number="177" hits="1"/>
- <line number="179" hits="1"/>
+ <line number="178" hits="1"/>
+ <line number="180" hits="1"/>
<line number="181" hits="1"/>
<line number="182" hits="1"/>
- <line number="183" hits="1"/>
<line number="184" hits="1"/>
- <line number="185" hits="1"/>
- <line number="187" hits="1"/>
+ <line number="186" hits="1"/>
<line number="188" hits="1"/>
- <line number="189" hits="0"/>
- <line number="190" hits="0"/>
- <line number="195" hits="0"/>
- <line number="196" hits="0"/>
- <line number="198" hits="0"/>
- <line number="199" hits="1"/>
- <line number="202" hits="1"/>
+ <line number="189" hits="1"/>
+ <line number="190" hits="1"/>
+ <line number="191" hits="1"/>
+ <line number="194" hits="1"/>
+ <line number="195" hits="1"/>
+ <line number="197" hits="1"/>
+ <line number="203" hits="1"/>
<line number="204" hits="1"/>
- <line number="205" hits="0"/>
- <line number="211" hits="1"/>
+ <line number="205" hits="1"/>
+ <line number="206" hits="1"/>
+ <line number="208" hits="1"/>
+ <line number="210" hits="1"/>
<line number="212" hits="1"/>
- <line number="213" hits="1"/>
<line number="214" hits="1"/>
<line number="215" hits="1"/>
+ <line number="216" hits="1"/>
<line number="217" hits="1"/>
- <line number="218" hits="1"/>
- <line number="219" hits="1"/>
- <line number="220" hits="1"/>
- <line number="221" hits="1"/>
+ <line number="222" hits="1"/>
<line number="223" hits="1"/>
<line number="224" hits="1"/>
<line number="225" hits="1"/>
- <line number="226" hits="1"/>
<line number="228" hits="1"/>
- <line number="229" hits="1"/>
- <line number="231" hits="1"/>
- <line number="234" hits="1"/>
- <line number="235" hits="1"/>
- <line number="237" hits="1"/>
- <line number="238" hits="1"/>
+ <line number="229" hits="0"/>
+ <line number="230" hits="0"/>
+ <line number="236" hits="0"/>
+ <line number="237" hits="0"/>
+ <line number="239" hits="0"/>
<line number="240" hits="1"/>
- <line number="241" hits="1"/>
<line number="243" hits="1"/>
- <line number="244" hits="1"/>
- <line number="246" hits="1"/>
- <line number="248" hits="1"/>
- <line number="250" hits="1"/>
- <line number="253" hits="1"/>
- <line number="256" hits="1"/>
- <line number="257" hits="1"/>
- <line number="258" hits="1"/>
- <line number="265" hits="1"/>
- <line number="266" hits="1"/>
- <line number="272" hits="1"/>
- <line number="277" hits="1"/>
+ <line number="245" hits="1"/>
+ <line number="250" hits="0"/>
+ <line number="251" hits="0"/>
+ <line number="252" hits="0"/>
+ <line number="253" hits="0"/>
+ <line number="255" hits="0"/>
+ <line number="257" hits="0"/>
+ <line number="260" hits="0"/>
+ <line number="262" hits="0"/>
+ <line number="263" hits="0"/>
+ <line number="264" hits="0"/>
+ <line number="265" hits="0"/>
+ <line number="266" hits="0"/>
+ <line number="267" hits="0"/>
+ <line number="269" hits="0"/>
+ <line number="271" hits="1"/>
+ <line number="272" hits="0"/>
<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="294" hits="1"/>
<line number="295" hits="1"/>
<line number="296" hits="1"/>
- <line number="297" hits="1"/>
- <line number="299" hits="1"/>
- <line number="302" hits="1"/>
- <line number="303" hits="1"/>
- <line number="306" hits="1"/>
+ <line number="298" hits="1"/>
+ <line number="300" hits="1"/>
+ <line number="301" hits="1"/>
+ <line number="304" hits="1"/>
+ <line number="305" hits="1"/>
<line number="307" hits="1"/>
- <line number="309" hits="1"/>
+ <line number="308" hits="1"/>
<line number="310" hits="1"/>
- <line number="312" hits="1"/>
+ <line number="311" hits="1"/>
<line number="313" hits="1"/>
<line number="314" hits="1"/>
<line number="316" hits="1"/>
<line number="318" hits="1"/>
<line number="319" 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="327" hits="1"/>
- <line number="329" hits="1"/>
+ <line number="328" hits="1"/>
<line number="330" hits="1"/>
<line number="331" hits="1"/>
+ <line number="336" hits="1"/>
+ <line number="337" hits="1"/>
+ <line number="339" hits="1"/>
+ <line number="341" hits="1"/>
+ <line number="344" hits="1"/>
+ <line number="347" hits="1"/>
+ <line number="348" hits="1"/>
+ <line number="349" hits="1"/>
+ <line number="356" hits="1"/>
+ <line number="357" hits="1"/>
+ <line number="363" hits="1"/>
+ <line number="366" hits="1"/>
+ <line number="367" hits="1"/>
+ <line number="368" hits="1"/>
+ <line number="370" hits="1"/>
+ <line number="372" hits="1"/>
+ <line number="374" hits="1"/>
+ <line number="375" hits="1"/>
+ <line number="377" hits="1"/>
+ <line number="378" hits="1"/>
+ <line number="380" hits="1"/>
+ <line number="381" hits="1"/>
+ <line number="383" hits="1"/>
+ <line number="384" hits="1"/>
+ <line number="385" hits="1"/>
+ <line number="386" hits="1"/>
+ <line number="387" hits="1"/>
+ <line number="388" hits="1"/>
+ <line number="389" hits="1"/>
+ <line number="391" hits="1"/>
+ <line number="394" hits="1"/>
+ <line number="395" hits="1"/>
+ <line number="398" hits="1"/>
+ <line number="399" hits="1"/>
+ <line number="401" hits="1"/>
+ <line number="402" hits="1"/>
+ <line number="404" hits="1"/>
+ <line number="405" hits="0"/>
+ <line number="406" hits="0"/>
+ <line number="407" hits="0"/>
+ <line number="408" hits="0"/>
+ <line number="409" hits="0"/>
+ <line number="410" hits="0"/>
+ <line number="412" hits="0"/>
+ <line number="416" hits="0"/>
+ <line number="417" hits="0"/>
+ <line number="418" hits="0"/>
+ <line number="419" hits="0"/>
+ <line number="420" hits="0"/>
+ <line number="422" hits="0"/>
+ <line number="425" hits="1"/>
+ <line number="426" hits="1"/>
+ <line number="427" hits="1"/>
+ <line number="429" hits="1"/>
+ <line number="431" hits="1"/>
+ <line number="435" hits="1"/>
+ <line number="436" hits="1"/>
+ <line number="437" hits="1"/>
+ <line number="441" hits="1"/>
+ <line number="442" hits="1"/>
+ <line number="448" hits="1"/>
+ <line number="449" hits="1"/>
+ <line number="450" hits="1"/>
+ <line number="452" hits="1"/>
+ <line number="453" hits="1"/>
+ <line number="454" hits="1"/>
+ <line number="456" hits="1"/>
+ <line number="458" hits="1"/>
+ <line number="462" hits="1"/>
+ <line number="464" hits="1"/>
+ <line number="471" hits="1"/>
+ <line number="473" hits="1"/>
+ <line number="474" hits="1"/>
+ <line number="476" hits="1"/>
+ <line number="477" hits="1"/>
+ <line number="479" hits="1"/>
+ <line number="484" hits="1"/>
+ <line number="486" hits="1"/>
+ <line number="488" hits="1"/>
+ <line number="489" hits="0"/>
+ <line number="490" hits="0"/>
+ <line number="492" hits="1"/>
+ <line number="494" hits="1"/>
+ <line number="498" hits="1"/>
+ <line number="499" hits="1"/>
+ <line number="500" hits="1"/>
+ <line number="502" hits="1"/>
+ <line number="506" hits="1"/>
+ <line number="507" hits="1"/>
+ <line number="508" hits="1"/>
+ <line number="509" hits="1"/>
+ <line number="510" hits="1"/>
+ <line number="511" hits="1"/>
+ <line number="512" hits="1"/>
+ <line number="515" hits="1"/>
+ <line number="521" hits="1"/>
+ <line number="522" hits="1"/>
+ <line number="523" hits="1"/>
+ <line number="525" hits="1"/>
+ <line number="526" hits="1"/>
+ <line number="527" hits="1"/>
+ <line number="535" hits="1"/>
+ <line number="539" hits="1"/>
+ <line number="542" hits="1"/>
+ <line number="543" hits="0"/>
+ <line number="544" hits="0"/>
+ <line number="545" hits="0"/>
+ <line number="548" hits="0"/>
+ <line number="560" hits="0"/>
+ <line number="571" hits="0"/>
+ <line number="574" hits="0"/>
+ <line number="577" hits="0"/>
+ <line number="580" hits="0"/>
+ <line number="581" hits="0"/>
+ <line number="582" hits="0"/>
+ <line number="583" hits="0"/>
</lines>
</class>
<class name="linear.py" filename="linear.py" complexity="0" line-rate="1" branch-rate="0">
@@ -339,7 +467,7 @@
<line number="30" hits="1"/>
</lines>
</class>
- <class name="route.py" filename="route.py" complexity="0" line-rate="0.9342" branch-rate="0">
+ <class name="route.py" filename="route.py" complexity="0" line-rate="0.9103" branch-rate="0">
<methods/>
<lines>
<line number="1" hits="1"/>
@@ -350,26 +478,26 @@
<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"/>
@@ -380,54 +508,56 @@
<line number="50" hits="1"/>
<line number="51" hits="1"/>
<line number="52" hits="1"/>
+ <line number="56" hits="0"/>
<line number="57" hits="0"/>
- <line number="60" hits="0"/>
- <line number="63" hits="1"/>
- <line number="64" hits="1"/>
- <line number="66" hits="1"/>
- <line number="67" hits="1"/>
+ <line number="62" hits="0"/>
+ <line number="65" hits="0"/>
+ <line number="68" hits="1"/>
<line number="69" hits="1"/>
- <line number="70" hits="1"/>
<line number="71" hits="1"/>
- <line number="73" hits="1"/>
+ <line number="72" hits="1"/>
<line number="74" hits="1"/>
+ <line number="75" hits="1"/>
+ <line number="76" hits="1"/>
<line number="78" hits="1"/>
<line number="79" hits="1"/>
- <line number="80" hits="1"/>
- <line number="81" hits="1"/>
<line number="83" hits="1"/>
<line number="84" hits="1"/>
+ <line number="85" hits="1"/>
<line number="86" 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="93" hits="0"/>
+ <line number="92" hits="1"/>
+ <line number="94" hits="1"/>
<line number="95" hits="1"/>
<line number="96" hits="1"/>
- <line number="97" hits="1"/>
- <line number="99" hits="1"/>
- <line number="124" hits="1"/>
- <line number="125" hits="1"/>
- <line number="126" hits="1"/>
- <line number="127" hits="0"/>
+ <line number="98" hits="0"/>
+ <line number="100" hits="1"/>
+ <line number="101" hits="1"/>
+ <line number="102" hits="1"/>
+ <line number="104" hits="1"/>
<line number="129" hits="1"/>
+ <line number="130" hits="1"/>
<line number="131" hits="1"/>
- <line number="133" hits="1"/>
+ <line number="132" hits="0"/>
<line number="134" hits="1"/>
- <line number="135" hits="1"/>
<line number="136" hits="1"/>
- <line number="137" hits="0"/>
+ <line number="138" hits="1"/>
+ <line number="139" hits="1"/>
+ <line number="140" hits="1"/>
+ <line number="141" hits="1"/>
+ <line number="142" hits="0"/>
</lines>
</class>
- <class name="schema.py" filename="schema.py" complexity="0" line-rate="0.9574" branch-rate="0">
+ <class name="schema.py" filename="schema.py" complexity="0" line-rate="0.9608" branch-rate="0">
<methods/>
<lines>
<line number="1" hits="1"/>
- <line number="3" hits="1"/>
+ <line number="2" hits="1"/>
<line number="4" hits="1"/>
- <line number="6" hits="1"/>
- <line number="12" hits="1"/>
+ <line number="5" hits="1"/>
+ <line number="7" hits="1"/>
<line number="13" hits="1"/>
<line number="16" hits="1"/>
<line number="17" hits="1"/>
@@ -437,44 +567,48 @@
<line number="23" hits="1"/>
<line number="24" hits="1"/>
<line number="25" hits="1"/>
- <line number="28" hits="1"/>
- <line number="29" hits="1"/>
+ <line number="26" hits="1"/>
+ <line number="27" hits="1"/>
<line number="30" hits="1"/>
<line number="31" hits="1"/>
<line number="32" 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="39" hits="1"/>
- <line number="40" hits="0"/>
+ <line number="40" hits="1"/>
<line number="41" hits="1"/>
- <line number="42" hits="1"/>
+ <line number="42" hits="0"/>
<line number="43" hits="1"/>
<line number="44" hits="1"/>
- <line number="46" hits="0"/>
- <line number="48" hits="1"/>
- <line number="49" hits="1"/>
- <line number="52" hits="1"/>
- <line number="53" hits="1"/>
+ <line number="45" hits="1"/>
+ <line number="46" hits="1"/>
+ <line number="48" hits="0"/>
+ <line number="50" hits="1"/>
+ <line number="51" hits="1"/>
<line number="54" hits="1"/>
+ <line number="55" hits="1"/>
<line number="56" hits="1"/>
- <line number="57" hits="1"/>
<line number="58" hits="1"/>
<line number="59" hits="1"/>
+ <line number="60" hits="1"/>
<line number="61" hits="1"/>
- <line number="62" hits="1"/>
- <line number="65" hits="1"/>
+ <line number="63" hits="1"/>
+ <line number="64" hits="1"/>
<line number="66" hits="1"/>
- <line number="68" hits="1"/>
- <line number="74" hits="1"/>
- <line number="75" hits="1"/>
+ <line number="67" hits="1"/>
+ <line number="70" hits="1"/>
+ <line number="71" hits="1"/>
+ <line number="73" hits="1"/>
+ <line number="81" hits="1"/>
+ <line number="82" hits="1"/>
</lines>
</class>
</classes>
</package>
- <package name="encoders" line-rate="0.9369" branch-rate="0" complexity="0">
+ <package name="encoders" line-rate="0.9615" branch-rate="0" complexity="0">
<classes>
<class name="__init__.py" filename="encoders/__init__.py" complexity="0" line-rate="1" branch-rate="0">
<methods/>
@@ -485,21 +619,24 @@
<line number="4" hits="1"/>
<line number="5" hits="1"/>
<line number="6" hits="1"/>
+ <line number="7" hits="1"/>
<line number="8" hits="1"/>
+ <line number="10" hits="1"/>
</lines>
</class>
<class name="base.py" filename="encoders/base.py" complexity="0" line-rate="1" branch-rate="0">
<methods/>
<lines>
<line number="1" hits="1"/>
- <line number="4" hits="1"/>
- <line number="5" hits="1"/>
+ <line number="3" 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="12" hits="1"/>
- <line number="13" hits="1"/>
+ <line number="14" hits="1"/>
+ <line number="15" hits="1"/>
</lines>
</class>
<class name="bm25.py" filename="encoders/bm25.py" complexity="0" line-rate="0.9574" branch-rate="0">
@@ -575,17 +712,17 @@
<line number="26" hits="1"/>
<line number="27" hits="1"/>
<line number="28" hits="1"/>
- <line number="30" hits="1"/>
- <line number="31" hits="1"/>
<line number="32" 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="39" hits="1"/>
</lines>
</class>
- <class name="fastembed.py" filename="encoders/fastembed.py" complexity="0" line-rate="0.7" branch-rate="0">
+ <class name="fastembed.py" filename="encoders/fastembed.py" complexity="0" line-rate="0.8667" branch-rate="0">
<methods/>
<lines>
<line number="1" hits="1"/>
@@ -610,12 +747,12 @@
<line number="33" hits="1"/>
<line number="40" hits="1"/>
<line number="42" hits="1"/>
- <line number="43" hits="0"/>
+ <line number="43" hits="1"/>
<line number="45" hits="1"/>
- <line number="46" hits="0"/>
- <line number="47" hits="0"/>
- <line number="48" hits="0"/>
- <line number="49" hits="0"/>
+ <line number="46" hits="1"/>
+ <line number="47" hits="1"/>
+ <line number="48" hits="1"/>
+ <line number="49" hits="1"/>
<line number="50" hits="0"/>
<line number="51" hits="0"/>
</lines>
@@ -624,10 +761,8 @@
<methods/>
<lines>
<line number="1" hits="1"/>
- <line number="2" hits="1"/>
<line number="3" hits="1"/>
- <line number="6" hits="1"/>
- <line number="7" hits="1"/>
+ <line number="5" hits="1"/>
<line number="8" hits="1"/>
<line number="9" hits="1"/>
<line number="10" hits="1"/>
@@ -636,76 +771,131 @@
<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="24" hits="1"/>
<line number="25" hits="1"/>
- <line number="31" hits="1"/>
- <line number="32" hits="1"/>
+ <line number="26" hits="1"/>
+ <line number="27" hits="1"/>
<line number="33" hits="1"/>
<line number="34" hits="1"/>
- <line number="40" hits="1"/>
+ <line number="35" hits="1"/>
+ <line number="36" hits="1"/>
<line number="42" hits="1"/>
- <line number="47" hits="1"/>
+ <line number="44" hits="1"/>
<line number="49" hits="1"/>
- <line number="50" hits="0"/>
- <line number="53" hits="1"/>
- <line number="54" hits="1"/>
+ <line number="51" hits="1"/>
+ <line number="52" hits="0"/>
<line number="55" hits="1"/>
+ <line number="56" hits="1"/>
<line number="57" hits="1"/>
<line number="59" hits="1"/>
- <line number="66" hits="1"/>
- <line number="67" hits="1"/>
+ <line number="61" hits="1"/>
<line number="68" hits="1"/>
+ <line number="69" hits="1"/>
<line number="70" hits="1"/>
- <line number="74" hits="1"/>
- <line number="75" hits="1"/>
+ <line number="72" hits="1"/>
+ <line number="76" hits="1"/>
<line number="77" hits="1"/>
- <line number="78" hits="1"/>
- <line number="81" hits="1"/>
- <line number="82" hits="1"/>
- <line number="86" hits="0"/>
- <line number="90" hits="1"/>
- <line number="91" 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="1"/>
<line number="93" hits="1"/>
- <line number="94" hits="1"/>
<line number="95" hits="1"/>
+ <line number="96" hits="1"/>
<line number="97" hits="1"/>
- <line number="98" hits="1"/>
<line number="99" hits="1"/>
- <line number="102" hits="1"/>
- <line number="106" hits="1"/>
- <line number="107" hits="1"/>
+ <line number="100" hits="1"/>
+ <line number="101" hits="1"/>
+ <line number="104" hits="1"/>
<line number="108" hits="1"/>
- <line number="111" hits="1"/>
- <line number="112" hits="1"/>
+ <line number="109" hits="1"/>
+ <line number="110" hits="1"/>
+ <line number="113" hits="1"/>
+ <line number="114" hits="1"/>
</lines>
</class>
- <class name="openai.py" filename="encoders/openai.py" complexity="0" line-rate="0.9767" branch-rate="0">
+ <class name="openai.py" filename="encoders/openai.py" complexity="0" line-rate="0.9783" branch-rate="0">
<methods/>
<lines>
<line number="1" hits="1"/>
<line number="2" hits="1"/>
- <line number="4" 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="10" hits="1"/>
+ <line number="11" hits="1"/>
+ <line number="14" hits="1"/>
+ <line number="15" hits="1"/>
+ <line number="16" hits="1"/>
+ <line number="17" hits="1"/>
+ <line number="19" 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="0"/>
+ <line number="32" hits="1"/>
+ <line number="33" hits="1"/>
+ <line number="34" hits="1"/>
+ <line number="35" 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="48" hits="1"/>
+ <line number="49" hits="1"/>
+ <line number="50" hits="1"/>
+ <line number="55" hits="1"/>
+ <line number="56" hits="1"/>
+ <line number="57" hits="1"/>
+ <line number="58" hits="1"/>
+ <line number="59" hits="1"/>
+ <line number="60" hits="1"/>
+ <line number="61" hits="1"/>
+ <line number="62" hits="1"/>
+ <line number="63" hits="1"/>
+ <line number="65" hits="1"/>
+ <line number="70" hits="1"/>
+ <line number="72" hits="1"/>
+ <line number="73" hits="1"/>
+ </lines>
+ </class>
+ <class name="tfidf.py" filename="encoders/tfidf.py" complexity="0" line-rate="0.9844" 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="9" hits="1"/>
<line number="10" hits="1"/>
<line number="13" hits="1"/>
<line number="14" hits="1"/>
<line number="15" hits="1"/>
<line number="17" 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="26" hits="1"/>
- <line number="27" hits="1"/>
- <line number="28" hits="0"/>
+ <line number="27" hits="0"/>
<line number="29" hits="1"/>
<line number="30" hits="1"/>
<line number="31" hits="1"/>
@@ -715,7 +905,8 @@
<line number="36" hits="1"/>
<line number="37" hits="1"/>
<line number="38" hits="1"/>
- <line number="41" hits="1"/>
+ <line number="39" hits="1"/>
+ <line number="40" hits="1"/>
<line number="42" hits="1"/>
<line number="43" hits="1"/>
<line number="44" hits="1"/>
@@ -723,19 +914,114 @@
<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="57" hits="1"/>
+ <line number="58" hits="1"/>
+ <line number="60" hits="1"/>
+ <line number="61" hits="1"/>
+ <line number="63" hits="1"/>
+ <line number="64" 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="1"/>
+ <line number="73" hits="1"/>
+ <line number="75" hits="1"/>
+ <line number="76" hits="1"/>
+ <line number="77" hits="1"/>
+ <line number="80" hits="1"/>
+ </lines>
+ </class>
+ <class name="zure.py" filename="encoders/zure.py" complexity="0" line-rate="0.9444" 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="9" hits="1"/>
+ <line number="10" 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="22" hits="1"/>
+ <line number="31" hits="1"/>
+ <line number="32" 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="39" hits="1"/>
+ <line number="40" hits="1"/>
+ <line number="41" hits="1"/>
+ <line number="42" hits="1"/>
+ <line number="43" hits="0"/>
+ <line number="44" hits="1"/>
+ <line number="45" hits="1"/>
+ <line number="49" hits="1"/>
+ <line number="50" hits="1"/>
+ <line number="51" hits="1"/>
+ <line number="52" hits="0"/>
+ <line number="53" hits="1"/>
<line number="54" hits="1"/>
+ <line number="55" hits="1"/>
+ <line number="56" hits="0"/>
+ <line number="57" hits="1"/>
+ <line number="58" hits="1"/>
<line number="59" hits="1"/>
+ <line number="60" hits="0"/>
<line number="61" hits="1"/>
- <line number="62" hits="1"/>
+ <line number="68" hits="1"/>
+ <line number="69" hits="1"/>
+ <line number="76" hits="1"/>
+ <line number="77" 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"/>
+ <line number="88" hits="1"/>
+ <line number="89" hits="1"/>
+ <line number="90" hits="1"/>
+ <line number="93" hits="1"/>
+ <line number="94" hits="1"/>
+ <line number="95" hits="1"/>
+ <line number="97" 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="107" hits="1"/>
+ <line number="112" hits="1"/>
+ <line number="114" hits="1"/>
+ <line number="115" hits="1"/>
</lines>
</class>
</classes>
</package>
- <package name="llms" line-rate="0.9043" branch-rate="0" complexity="0">
+ <package name="llms" line-rate="0.9071" branch-rate="0" complexity="0">
<classes>
<class name="__init__.py" filename="llms/__init__.py" complexity="0" line-rate="1" branch-rate="0">
<methods/>
@@ -744,21 +1030,54 @@
<line number="2" hits="1"/>
<line number="3" hits="1"/>
<line number="4" hits="1"/>
- <line number="6" hits="1"/>
+ <line number="5" hits="1"/>
+ <line number="7" hits="1"/>
</lines>
</class>
- <class name="base.py" filename="llms/base.py" complexity="0" line-rate="1" 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"/>
- <line number="3" hits="1"/>
- <line number="5" hits="1"/>
- <line number="8" hits="1"/>
- <line number="9" hits="1"/>
+ <line number="2" 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="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="19" hits="1"/>
+ <line number="20" hits="1"/>
+ <line number="22" hits="1"/>
+ <line number="26" 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="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="40" hits="1"/>
+ <line number="41" hits="1"/>
+ <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="1"/>
+ <line number="80" hits="0"/>
+ <line number="82" hits="1"/>
+ <line number="83" hits="1"/>
+ <line number="84" hits="1"/>
+ <line number="85" hits="1"/>
+ <line number="86" hits="1"/>
+ <line number="87" hits="1"/>
+ <line number="88" hits="1"/>
</lines>
</class>
<class name="cohere.py" filename="llms/cohere.py" complexity="0" line-rate="0.9655" branch-rate="0">
@@ -782,25 +1101,70 @@
<line number="25" hits="1"/>
<line number="26" hits="1"/>
<line number="27" hits="1"/>
- <line number="29" hits="1"/>
- <line number="30" hits="1"/>
<line number="31" hits="1"/>
<line number="32" hits="1"/>
<line number="33" hits="1"/>
- <line number="39" hits="1"/>
+ <line number="34" hits="1"/>
+ <line number="35" hits="1"/>
<line number="41" hits="1"/>
- <line number="42" hits="0"/>
<line number="43" hits="1"/>
+ <line number="44" hits="0"/>
<line number="45" hits="1"/>
- <line number="46" hits="1"/>
+ <line number="47" hits="1"/>
+ <line number="48" hits="1"/>
</lines>
</class>
- <class name="openai.py" filename="llms/openai.py" complexity="0" line-rate="0.8571" 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="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="1"/>
+ <line number="56" hits="0"/>
+ <line number="57" hits="0"/>
+ <line number="58" 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="openai.py" filename="llms/openai.py" complexity="0" line-rate="0.8571" branch-rate="0">
+ <methods/>
+ <lines>
+ <line number="1" hits="1"/>
+ <line number="2" hits="1"/>
+ <line number="4" hits="1"/>
<line number="6" hits="1"/>
<line number="7" hits="1"/>
<line number="8" hits="1"/>
@@ -819,28 +1183,28 @@
<line number="30" hits="1"/>
<line number="31" hits="1"/>
<line number="32" 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="39" hits="1"/>
<line number="40" hits="1"/>
- <line number="47" hits="1"/>
+ <line number="41" hits="1"/>
+ <line number="42" hits="1"/>
<line number="49" hits="1"/>
- <line number="50" hits="0"/>
<line number="51" hits="1"/>
<line number="52" hits="0"/>
- <line number="53" hits="0"/>
+ <line number="53" hits="1"/>
<line number="54" hits="0"/>
+ <line number="55" hits="0"/>
+ <line number="56" hits="0"/>
</lines>
</class>
<class name="openrouter.py" filename="llms/openrouter.py" complexity="0" line-rate="0.8649" branch-rate="0">
<methods/>
<lines>
<line number="1" hits="1"/>
- <line number="3" hits="1"/>
- <line number="5" hits="1"/>
+ <line number="2" hits="1"/>
+ <line number="4" hits="1"/>
<line number="6" hits="1"/>
<line number="7" hits="1"/>
<line number="8" hits="1"/>
@@ -861,98 +1225,114 @@
<line number="35" hits="1"/>
<line number="36" hits="1"/>
<line number="37" 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="46" hits="1"/>
+ <line number="47" hits="1"/>
+ <line number="54" hits="1"/>
+ <line number="56" hits="1"/>
+ <line number="57" hits="0"/>
+ <line number="58" hits="1"/>
+ <line number="59" hits="0"/>
+ <line number="60" hits="0"/>
+ <line number="61" hits="0"/>
+ </lines>
+ </class>
+ <class name="zure.py" filename="llms/zure.py" complexity="0" line-rate="0.8684" branch-rate="0">
+ <methods/>
+ <lines>
+ <line number="1" hits="1"/>
+ <line number="2" hits="1"/>
+ <line number="4" hits="1"/>
+ <line number="6" hits="1"/>
+ <line number="7" hits="1"/>
+ <line number="8" 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="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="0"/>
+ <line number="31" hits="1"/>
+ <line number="32" hits="1"/>
+ <line number="33" hits="1"/>
+ <line number="34" hits="1"/>
+ <line number="35" hits="1"/>
<line number="38" hits="1"/>
<line number="39" hits="1"/>
+ <line number="40" 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="52" hits="1"/>
+ <line number="46" hits="1"/>
+ <line number="47" hits="1"/>
<line number="54" hits="1"/>
- <line number="55" hits="0"/>
<line number="56" hits="1"/>
<line number="57" hits="0"/>
- <line number="58" hits="0"/>
+ <line number="58" hits="1"/>
<line number="59" hits="0"/>
+ <line number="60" hits="0"/>
+ <line number="61" hits="0"/>
</lines>
</class>
</classes>
</package>
- <package name="utils" line-rate="0.5246" branch-rate="0" complexity="0">
+ <package name="utils" line-rate="0.6667" branch-rate="0" complexity="0">
<classes>
<class name="__init__.py" filename="utils/__init__.py" complexity="0" line-rate="1" branch-rate="0">
<methods/>
<lines/>
</class>
- <class name="function_call.py" filename="utils/function_call.py" complexity="0" line-rate="0.2456" branch-rate="0">
+ <class name="function_call.py" filename="utils/function_call.py" complexity="0" line-rate="0.3667" 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="4" hits="1"/>
+ <line number="6" hits="1"/>
<line number="7" hits="1"/>
<line number="8" hits="1"/>
- <line number="9" hits="1"/>
+ <line number="11" hits="1"/>
<line number="12" hits="1"/>
- <line number="13" hits="1"/>
+ <line number="13" hits="0"/>
<line number="14" hits="0"/>
<line number="15" hits="0"/>
<line number="16" hits="0"/>
- <line number="17" hits="0"/>
+ <line number="18" hits="0"/>
<line number="19" hits="0"/>
<line number="20" hits="0"/>
- <line number="21" hits="0"/>
- <line number="25" hits="0"/>
+ <line number="24" hits="0"/>
+ <line number="26" hits="0"/>
<line number="27" hits="0"/>
<line number="28" hits="0"/>
- <line number="29" hits="0"/>
- <line number="35" hits="1"/>
- <line number="41" hits="1"/>
+ <line number="34" hits="1"/>
+ <line number="40" hits="1"/>
<line number="44" hits="1"/>
<line number="47" hits="0"/>
<line number="49" hits="0"/>
- <line number="77" hits="0"/>
- <line number="78" hits="0"/>
- <line number="79" hits="0"/>
- <line number="80" hits="0"/>
- <line number="82" hits="0"/>
- <line number="84" hits="0"/>
- <line number="85" hits="0"/>
- <line number="86" hits="0"/>
- <line number="87" hits="0"/>
- <line number="90" hits="1"/>
- <line number="92" hits="0"/>
- <line number="94" hits="0"/>
- <line number="95" hits="0"/>
- <line number="96" hits="0"/>
- <line number="97" hits="0"/>
- <line number="101" 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="107" hits="0"/>
- <line number="108" hits="0"/>
- <line number="112" hits="1"/>
- <line number="115" hits="0"/>
- <line number="117" hits="0"/>
- <line number="118" hits="0"/>
- <line number="119" hits="0"/>
- <line number="120" hits="0"/>
- <line number="122" hits="0"/>
- <line number="123" hits="0"/>
- <line number="124" hits="0"/>
+ <line number="50" hits="0"/>
+ <line number="51" hits="0"/>
+ <line number="52" hits="0"/>
+ <line number="54" hits="0"/>
+ <line number="55" hits="0"/>
+ <line number="56" hits="0"/>
</lines>
</class>
<class name="llm.py" filename="utils/llm.py" complexity="0" line-rate="0" branch-rate="0">
<methods/>
<lines>
<line number="1" hits="0"/>
- <line number="3" hits="0"/>
- <line number="5" hits="0"/>
+ <line number="2" hits="0"/>
+ <line number="4" hits="0"/>
<line number="6" hits="0"/>
<line number="9" hits="0"/>
<line number="10" hits="0"/>
@@ -996,16 +1376,13 @@
<lines>
<line number="1" hits="1"/>
<line number="3" hits="1"/>
+ <line number="4" hits="1"/>
<line number="6" hits="1"/>
- <line number="31" hits="1"/>
- <line number="32" hits="1"/>
- <line number="33" hits="1"/>
- <line number="34" 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="9" hits="1"/>
+ <line number="10" hits="1"/>
+ <line number="11" hits="1"/>
+ <line number="12" hits="1"/>
+ <line number="15" hits="1"/>
<line number="42" hits="1"/>
<line number="43" hits="1"/>
<line number="44" hits="1"/>
@@ -1014,17 +1391,27 @@
<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="56" hits="1"/>
- <line number="61" hits="1"/>
+ <line number="55" hits="1"/>
<line number="62" hits="1"/>
+ <line number="63" 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="71" hits="1"/>
+ <line number="72" hits="1"/>
<line number="74" hits="1"/>
- <line number="75" hits="1"/>
+ <line number="79" hits="1"/>
+ <line number="80" hits="1"/>
+ <line number="87" hits="1"/>
+ <line number="88" hits="1"/>
+ <line number="91" hits="1"/>
+ <line number="96" hits="1"/>
+ <line number="97" hits="1"/>
</lines>
</class>
</classes>
diff --git a/docs/00-introduction.ipynb b/docs/00-introduction.ipynb
index 2ff5a9852c3a8fb4988ad5991f09d85c5e663c31..7d4fed5d974a7b31077a45f638681b7c8ebbf62d 100644
--- a/docs/00-introduction.ipynb
+++ b/docs/00-introduction.ipynb
@@ -53,7 +53,7 @@
},
{
"cell_type": "code",
- "execution_count": 1,
+ "execution_count": null,
"metadata": {},
"outputs": [],
"source": [
@@ -81,7 +81,7 @@
},
{
"cell_type": "code",
- "execution_count": 2,
+ "execution_count": null,
"metadata": {},
"outputs": [],
"source": [
@@ -108,7 +108,7 @@
},
{
"cell_type": "code",
- "execution_count": 3,
+ "execution_count": null,
"metadata": {},
"outputs": [],
"source": [
@@ -136,17 +136,9 @@
},
{
"cell_type": "code",
- "execution_count": 4,
+ "execution_count": null,
"metadata": {},
- "outputs": [
- {
- "name": "stderr",
- "output_type": "stream",
- "text": [
- "\u001b[32m2024-01-07 18:08:29 INFO semantic_router.utils.logger Initializing RouteLayer\u001b[0m\n"
- ]
- }
- ],
+ "outputs": [],
"source": [
"from semantic_router.layer import RouteLayer\n",
"\n",
@@ -162,40 +154,18 @@
},
{
"cell_type": "code",
- "execution_count": 5,
+ "execution_count": null,
"metadata": {},
- "outputs": [
- {
- "data": {
- "text/plain": [
- "RouteChoice(name='politics', function_call=None)"
- ]
- },
- "execution_count": 5,
- "metadata": {},
- "output_type": "execute_result"
- }
- ],
+ "outputs": [],
"source": [
"rl(\"don't you love politics?\")"
]
},
{
"cell_type": "code",
- "execution_count": 6,
+ "execution_count": null,
"metadata": {},
- "outputs": [
- {
- "data": {
- "text/plain": [
- "RouteChoice(name='chitchat', function_call=None)"
- ]
- },
- "execution_count": 6,
- "metadata": {},
- "output_type": "execute_result"
- }
- ],
+ "outputs": [],
"source": [
"rl(\"how's the weather today?\")"
]
@@ -209,24 +179,30 @@
},
{
"cell_type": "code",
- "execution_count": 7,
+ "execution_count": null,
"metadata": {},
- "outputs": [
- {
- "data": {
- "text/plain": [
- "RouteChoice(name=None, function_call=None)"
- ]
- },
- "execution_count": 7,
- "metadata": {},
- "output_type": "execute_result"
- }
- ],
+ "outputs": [],
"source": [
"rl(\"I'm interested in learning about llama 2\")"
]
},
+ {
+ "cell_type": "markdown",
+ "metadata": {},
+ "source": [
+ "We can also retrieve multiple routes with its associated score:"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": [
+ "\n",
+ "rl.retrieve_multiple_routes(\"Hi! How are you doing in politics??\")"
+ ]
+ },
{
"cell_type": "markdown",
"metadata": {},
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 08afccfab19b590a1fc975e9325c69d6cb961d0b..31c7cfa081e9b9143d6b406c7d8f711374f72fc0 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -242,6 +242,32 @@ class RouteLayer:
# if no route passes threshold, return empty route choice
return RouteChoice()
+ def retrieve_multiple_routes(
+ self,
+ text: Optional[str] = None,
+ vector: Optional[List[float]] = None,
+ ) -> List[RouteChoice]:
+ if vector is None:
+ if text is None:
+ raise ValueError("Either text or vector must be provided")
+ vector_arr = self._encode(text=text)
+ else:
+ vector_arr = np.array(vector)
+ # get relevant utterances
+ results = self._retrieve(xq=vector_arr)
+
+ # decide most relevant routes
+ categories_with_scores = self._semantic_classify_multiple_routes(results, self.score_threshold)
+
+ route_choices = []
+ for category, score in categories_with_scores:
+ route = self.check_for_matching_routes(category)
+ if route:
+ route_choice = RouteChoice(name=route.name, similarity_score=score, route=route)
+ route_choices.append(route_choice)
+
+ return route_choices
+
def __str__(self):
return (
f"RouteLayer(encoder={self.encoder}, "
@@ -375,6 +401,27 @@ class RouteLayer:
logger.warning("No classification found for semantic classifier.")
return "", []
+ def _semantic_classify_multiple_routes(self, query_results: List[dict], threshold: float) -> List[Tuple[str, float]]:
+ scores_by_class: Dict[str, List[float]] = {}
+ for result in query_results:
+ score = result["score"]
+ route = result["route"]
+ if route in scores_by_class:
+ scores_by_class[route].append(score)
+ else:
+ scores_by_class[route] = [score]
+
+
+ # Filter classes based on threshold and find max score for each
+ classes_above_threshold = []
+ for route, scores in scores_by_class.items():
+ if self._pass_threshold(scores, threshold):
+ max_score = max(scores)
+ classes_above_threshold.append((route, max_score))
+
+ return classes_above_threshold
+
+
def _pass_threshold(self, scores: List[float], threshold: float) -> bool:
if scores:
return max(scores) > threshold
@@ -490,3 +537,47 @@ def threshold_random_search(
for i, route in enumerate(route_names)
}
return score_thresholds
+
+
+if __name__ == "__main__":
+ from semantic_router import Route
+ from semantic_router.layer import RouteLayer
+ from semantic_router.encoders import OpenAIEncoder
+
+ # Define routes with example phrases
+ politics = Route(
+ name="politics",
+ utterances=[
+ "isn't politics the best thing ever",
+ "why don't you tell me about your political opinions",
+ "don't you just love the president",
+ "don't you just hate the president",
+ "they're going to destroy this country!",
+ "they will save the country!",
+ ],
+ )
+
+ chitchat = Route(
+ name="chitchat",
+ utterances=[
+ "how's the weather today?",
+ "how are things going?",
+ "lovely weather today",
+ "the weather is horrendous",
+ "let's go to the chippy",
+ ],
+ )
+
+ routes = [politics, chitchat]
+
+ # Initialize the encoder
+ encoder = OpenAIEncoder()
+
+ # Initialize the RouteLayer with the encoder and routes
+ rl = RouteLayer(encoder=encoder, routes=routes)
+
+ # Test the RouteLayer with example queries
+ print(rl.retrieve_multiple_routes("how's the weather today?")) # Expected to match the chitchat route
+ print(rl.retrieve_multiple_routes("don't you love politics?")) # Expected to match the politics route
+ print(rl.retrieve_multiple_routes("I'm interested in learning about llama 2")) # Expected to return None since it doesn't match any route
+ print(rl.retrieve_multiple_routes("Hi! How are you doing in politics??"))
\ No newline at end of file