diff --git a/coverage.xml b/coverage.xml
index 5850def2dfb8bf3ee91ce0e20a28f5312ec8c0b2..7eb662b902de37fa6581d52548d1cf54db5e5be4 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">
+<coverage version="7.4.0" timestamp="1707722055292" lines-valid="1402" lines-covered="1188" line-rate="0.8474" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0">
<!-- Generated by coverage.py: https://coverage.readthedocs.io/en/7.4.0 -->
<!-- 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>/home/dustinmorris/semantic-router/semantic_router</source>
</sources>
<packages>
- <package name="." line-rate="0.9263" branch-rate="0" complexity="0">
+ <package name="." line-rate="0.9055" 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,396 @@
<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.9055" 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="245" hits="1"/>
+ <line number="246" hits="0"/>
+ <line number="252" hits="1"/>
<line number="253" hits="1"/>
+ <line number="254" hits="1"/>
+ <line number="255" hits="1"/>
<line number="256" hits="1"/>
- <line number="257" hits="1"/>
<line number="258" hits="1"/>
+ <line number="259" hits="1"/>
+ <line number="260" hits="1"/>
+ <line number="261" hits="1"/>
+ <line number="262" hits="1"/>
+ <line number="264" hits="1"/>
<line number="265" hits="1"/>
<line number="266" hits="1"/>
+ <line number="267" hits="1"/>
+ <line number="269" hits="1"/>
+ <line number="270" hits="1"/>
<line number="272" hits="1"/>
- <line number="277" hits="1"/>
+ <line number="274" hits="1"/>
+ <line number="275" hits="1"/>
<line number="278" hits="1"/>
- <line number="280" hits="1"/>
+ <line number="279" 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="291" hits="1"/>
+ <line number="290" 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="298" hits="1"/>
<line number="299" hits="1"/>
+ <line number="301" hits="1"/>
<line number="302" hits="1"/>
- <line number="303" hits="1"/>
- <line number="306" hits="1"/>
- <line number="307" hits="1"/>
- <line number="309" hits="1"/>
+ <line number="304" hits="1"/>
+ <line number="305" 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="315" hits="1"/>
<line number="318" hits="1"/>
- <line number="319" hits="1"/>
- <line number="325" hits="1"/>
- <line number="326" hits="1"/>
- <line number="327" hits="1"/>
- <line number="329" hits="1"/>
+ <line number="321" hits="1"/>
+ <line number="322" hits="1"/>
+ <line number="323" hits="1"/>
<line number="330" hits="1"/>
<line number="331" hits="1"/>
+ <line number="337" hits="1"/>
+ <line number="340" hits="1"/>
+ <line number="341" hits="1"/>
+ <line number="342" hits="1"/>
+ <line number="344" hits="1"/>
+ <line number="346" hits="1"/>
+ <line number="348" hits="1"/>
+ <line number="349" hits="1"/>
+ <line number="351" hits="1"/>
+ <line number="352" hits="1"/>
+ <line number="354" hits="1"/>
+ <line number="355" hits="1"/>
+ <line number="357" hits="1"/>
+ <line number="358" hits="1"/>
+ <line number="359" hits="1"/>
+ <line number="360" hits="1"/>
+ <line number="361" hits="1"/>
+ <line number="362" hits="1"/>
+ <line number="363" hits="1"/>
+ <line number="365" hits="1"/>
+ <line number="368" hits="1"/>
+ <line number="369" hits="1"/>
+ <line number="372" hits="1"/>
+ <line number="373" hits="1"/>
+ <line number="375" hits="1"/>
+ <line number="376" hits="1"/>
+ <line number="378" hits="1"/>
+ <line number="379" hits="1"/>
+ <line number="380" hits="1"/>
+ <line number="382" hits="1"/>
+ <line number="384" hits="1"/>
+ <line number="388" hits="1"/>
+ <line number="389" hits="1"/>
+ <line number="390" hits="1"/>
+ <line number="394" hits="1"/>
+ <line number="395" hits="1"/>
+ <line number="401" hits="1"/>
+ <line number="402" hits="1"/>
+ <line number="403" hits="1"/>
+ <line number="405" hits="1"/>
+ <line number="406" hits="1"/>
+ <line number="407" hits="1"/>
+ <line number="409" hits="1"/>
+ <line number="411" hits="1"/>
+ <line number="415" hits="1"/>
+ <line number="417" hits="1"/>
+ <line number="424" hits="1"/>
+ <line number="426" hits="1"/>
+ <line number="427" hits="1"/>
+ <line number="429" hits="1"/>
+ <line number="430" hits="1"/>
+ <line number="432" hits="1"/>
+ <line number="437" hits="1"/>
+ <line number="439" hits="1"/>
+ <line number="441" hits="1"/>
+ <line number="442" hits="0"/>
+ <line number="443" hits="0"/>
+ <line number="445" hits="1"/>
+ <line number="447" hits="1"/>
+ <line number="451" hits="1"/>
+ <line number="452" hits="1"/>
+ <line number="453" hits="1"/>
+ <line number="455" hits="1"/>
+ <line number="459" hits="1"/>
+ <line number="460" hits="1"/>
+ <line number="461" hits="1"/>
+ <line number="462" hits="1"/>
+ <line number="463" hits="1"/>
+ <line number="464" hits="1"/>
+ <line number="465" hits="1"/>
+ <line number="468" hits="1"/>
+ <line number="474" hits="1"/>
+ <line number="475" hits="1"/>
+ <line number="476" hits="1"/>
+ <line number="478" hits="1"/>
+ <line number="479" hits="1"/>
+ <line number="480" hits="1"/>
+ <line number="488" hits="1"/>
+ <line number="492" hits="1"/>
</lines>
</class>
<class name="linear.py" filename="linear.py" complexity="0" line-rate="1" branch-rate="0">
@@ -339,7 +425,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 +436,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,101 +466,189 @@
<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.8929" 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="13" hits="1"/>
+ <line number="5" hits="1"/>
+ <line number="7" 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="23" hits="1"/>
+ <line number="21" 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="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="43" hits="1"/>
+ <line number="43" hits="0"/>
<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="45" hits="1"/>
+ <line number="46" hits="1"/>
+ <line number="47" hits="1"/>
+ <line number="48" hits="0"/>
+ <line number="49" hits="0"/>
+ <line number="51" hits="0"/>
<line number="53" hits="1"/>
<line number="54" hits="1"/>
- <line number="56" hits="1"/>
<line number="57" hits="1"/>
<line number="58" hits="1"/>
<line number="59" hits="1"/>
<line number="61" hits="1"/>
<line number="62" hits="1"/>
- <line number="65" hits="1"/>
- <line number="66" hits="1"/>
- <line number="68" hits="1"/>
- <line number="74" hits="1"/>
+ <line number="63" hits="1"/>
+ <line number="64" hits="1"/>
+ <line number="66" hits="1"/>
+ <line number="67" hits="1"/>
+ <line number="69" hits="1"/>
+ <line number="70" hits="0"/>
+ <line number="72" hits="1"/>
+ <line number="73" hits="1"/>
<line number="75" hits="1"/>
+ <line number="76" hits="0"/>
+ <line number="79" hits="1"/>
+ <line number="80" hits="1"/>
+ <line number="81" hits="1"/>
+ <line number="82" hits="1"/>
+ </lines>
+ </class>
+ <class name="text.py" filename="text.py" complexity="0" line-rate="0.7778" 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="7" hits="1"/>
+ <line number="8" hits="1"/>
+ <line number="9" hits="1"/>
+ <line number="12" hits="1"/>
+ <line number="14" hits="1"/>
+ <line number="25" hits="1"/>
+ <line number="26" hits="1"/>
+ <line number="29" hits="1"/>
+ <line number="30" hits="1"/>
+ <line number="32" hits="1"/>
+ <line number="33" hits="0"/>
+ <line number="34" hits="0"/>
+ <line number="35" hits="0"/>
+ <line number="36" hits="0"/>
+ <line number="39" hits="0"/>
+ <line number="40" hits="0"/>
+ <line number="41" hits="0"/>
+ <line number="42" hits="0"/>
+ <line number="43" hits="0"/>
+ <line number="45" hits="0"/>
+ <line number="46" hits="0"/>
+ <line number="47" hits="0"/>
+ <line number="48" hits="0"/>
+ <line number="50" hits="1"/>
+ <line number="56" hits="1"/>
+ <line number="58" hits="1"/>
+ <line number="59" hits="1"/>
+ <line number="61" hits="1"/>
+ <line number="83" hits="1"/>
+ <line number="84" hits="1"/>
+ <line number="87" hits="1"/>
+ <line number="88" hits="1"/>
+ <line number="92" hits="1"/>
+ <line number="94" hits="1"/>
+ <line number="106" hits="1"/>
+ <line number="107" hits="1"/>
+ <line number="109" hits="1"/>
+ <line number="111" hits="1"/>
+ <line number="124" hits="1"/>
+ <line number="125" hits="1"/>
+ <line number="126" hits="1"/>
+ <line number="131" hits="1"/>
+ <line number="132" hits="1"/>
+ <line number="134" hits="1"/>
+ <line number="145" hits="1"/>
+ <line number="146" hits="1"/>
+ <line number="147" hits="1"/>
+ <line number="149" hits="1"/>
+ <line number="163" hits="1"/>
+ <line number="164" hits="1"/>
+ <line number="167" hits="1"/>
+ <line number="169" hits="1"/>
+ <line number="171" hits="0"/>
+ <line number="174" hits="1"/>
+ <line number="175" hits="1"/>
+ <line number="176" hits="1"/>
+ <line number="177" hits="1"/>
+ <line number="180" hits="1"/>
+ <line number="184" hits="1"/>
+ <line number="185" hits="1"/>
+ <line number="187" hits="1"/>
+ <line number="190" hits="1"/>
+ <line number="191" hits="0"/>
+ <line number="194" hits="1"/>
+ <line number="199" hits="1"/>
+ <line number="200" hits="0"/>
+ <line number="202" hits="1"/>
+ <line number="205" hits="1"/>
</lines>
</class>
</classes>
</package>
- <package name="encoders" line-rate="0.9369" branch-rate="0" complexity="0">
+ <package name="encoders" line-rate="0.8771" 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 +659,25 @@
<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="9" hits="1"/>
+ <line number="11" 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">
@@ -564,28 +742,30 @@
<line number="9" hits="1"/>
<line number="10" hits="1"/>
<line number="11" hits="1"/>
- <line number="13" hits="1"/>
- <line number="19" hits="1"/>
- <line number="20" hits="1"/>
+ <line number="12" hits="1"/>
+ <line number="14" 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="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="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="46" hits="1"/>
+ <line number="47" hits="1"/>
+ <line number="48" 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,24 +790,22 @@
<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>
</class>
- <class name="huggingface.py" filename="encoders/huggingface.py" complexity="0" line-rate="0.9667" branch-rate="0">
+ <class name="huggingface.py" filename="encoders/huggingface.py" complexity="0" line-rate="0.4" branch-rate="0">
<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,87 +814,86 @@
<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="33" hits="1"/>
- <line number="34" hits="1"/>
- <line number="40" hits="1"/>
- <line number="42" hits="1"/>
- <line number="47" hits="1"/>
- <line number="49" hits="1"/>
- <line number="50" hits="0"/>
- <line number="53" hits="1"/>
- <line number="54" hits="1"/>
- <line number="55" hits="1"/>
- <line number="57" hits="1"/>
- <line number="59" hits="1"/>
- <line number="66" hits="1"/>
- <line number="67" hits="1"/>
- <line number="68" hits="1"/>
- <line number="70" hits="1"/>
- <line number="74" hits="1"/>
- <line number="75" 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="93" hits="1"/>
- <line number="94" hits="1"/>
- <line number="95" hits="1"/>
- <line number="97" hits="1"/>
- <line number="98" 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="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="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="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="99" hits="1"/>
- <line number="102" hits="1"/>
- <line number="106" hits="1"/>
- <line number="107" hits="1"/>
+ <line number="100" hits="0"/>
+ <line number="101" hits="0"/>
+ <line number="104" hits="0"/>
<line number="108" hits="1"/>
- <line number="111" hits="1"/>
- <line number="112" hits="1"/>
+ <line number="109" hits="0"/>
+ <line number="110" hits="0"/>
+ <line number="113" hits="0"/>
+ <line number="114" hits="0"/>
</lines>
</class>
- <class name="openai.py" filename="encoders/openai.py" complexity="0" line-rate="0.9767" branch-rate="0">
+ <class name="mistral.py" filename="encoders/mistral.py" complexity="0" line-rate="0.95" 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="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="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="23" hits="1"/>
- <line number="24" hits="1"/>
+ <line number="19" hits="1"/>
<line number="25" hits="1"/>
<line number="26" hits="1"/>
<line number="27" hits="1"/>
- <line number="28" hits="0"/>
+ <line number="28" hits="1"/>
<line number="29" hits="1"/>
<line number="30" hits="1"/>
<line number="31" hits="1"/>
<line number="32" hits="1"/>
- <line number="34" hits="1"/>
- <line number="35" hits="1"/>
+ <line number="33" hits="0"/>
+ <line number="34" hits="0"/>
<line number="36" hits="1"/>
<line number="37" hits="1"/>
<line number="38" hits="1"/>
- <line number="41" hits="1"/>
- <line number="42" hits="1"/>
+ <line number="39" hits="1"/>
+ <line number="40" hits="1"/>
<line number="43" hits="1"/>
<line number="44" hits="1"/>
<line number="45" hits="1"/>
@@ -728,79 +905,395 @@
<line number="51" hits="1"/>
<line number="52" hits="1"/>
<line number="54" hits="1"/>
- <line number="59" hits="1"/>
- <line number="61" hits="1"/>
- <line number="62" hits="1"/>
+ <line number="55" hits="1"/>
+ <line number="56" hits="1"/>
+ <line number="57" hits="1"/>
</lines>
</class>
- </classes>
- </package>
- <package name="llms" line-rate="0.9043" branch-rate="0" complexity="0">
- <classes>
- <class name="__init__.py" filename="llms/__init__.py" complexity="0" line-rate="1" 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="3" hits="1"/>
- <line number="4" hits="1"/>
- <line number="6" hits="1"/>
- </lines>
- </class>
- <class name="base.py" filename="llms/base.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="8" hits="1"/>
- <line number="9" hits="1"/>
- <line number="11" hits="1"/>
- <line number="12" hits="1"/>
- <line number="14" hits="1"/>
- <line number="15" hits="1"/>
- </lines>
- </class>
- <class name="cohere.py" filename="llms/cohere.py" complexity="0" line-rate="0.9655" 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="10" hits="1"/>
<line number="11" hits="1"/>
- <line number="13" hits="1"/>
- <line number="18" 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="21" 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="0"/>
+ <line number="29" hits="1"/>
+ <line number="30" hits="1"/>
+ <line number="31" hits="1"/>
+ <line number="32" 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="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="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="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.7214" branch-rate="0" complexity="0">
+ <classes>
+ <class name="__init__.py" filename="llms/__init__.py" complexity="0" line-rate="1" 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="5" hits="1"/>
+ <line number="6" hits="1"/>
+ <line number="8" hits="1"/>
+ </lines>
+ </class>
+ <class name="base.py" filename="llms/base.py" complexity="0" line-rate="0.7317" 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="10" hits="1"/>
+ <line number="11" hits="1"/>
+ <line number="13" hits="1"/>
+ <line number="14" 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="0"/>
+ <line number="37" hits="0"/>
+ <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="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"/>
+ </lines>
+ </class>
+ <class name="cohere.py" filename="llms/cohere.py" complexity="0" line-rate="0.9655" 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="10" hits="1"/>
+ <line number="11" hits="1"/>
+ <line number="13" 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="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="41" hits="1"/>
+ <line number="43" hits="1"/>
+ <line number="44" hits="0"/>
+ <line number="45" hits="1"/>
+ <line number="47" hits="1"/>
+ <line number="48" hits="1"/>
+ </lines>
+ </class>
+ <class name="llamacpp.py" filename="llms/llamacpp.py" complexity="0" line-rate="0" 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="54" hits="0"/>
+ <line number="55" hits="0"/>
+ <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"/>
+ </lines>
+ </class>
+ <class name="mistral.py" filename="llms/mistral.py" complexity="0" line-rate="0.8286" 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="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="31" hits="1"/>
- <line number="32" hits="1"/>
- <line number="33" hits="1"/>
+ <line number="31" hits="0"/>
+ <line number="32" hits="0"/>
+ <line number="35" hits="1"/>
+ <line number="36" 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="0"/>
- <line number="43" hits="1"/>
- <line number="45" hits="1"/>
- <line number="46" hits="1"/>
+ <line number="42" hits="1"/>
+ <line number="49" hits="1"/>
+ <line number="51" hits="1"/>
+ <line number="52" 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="openai.py" filename="llms/openai.py" complexity="0" line-rate="0.8571" 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"/>
@@ -819,28 +1312,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 +1354,203 @@
<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="splitters" line-rate="1" branch-rate="0" complexity="0">
<classes>
- <class name="__init__.py" filename="utils/__init__.py" complexity="0" line-rate="1" branch-rate="0">
+ <class name="__init__.py" filename="splitters/__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="base.py" filename="splitters/base.py" complexity="0" line-rate="1" 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="10" hits="1"/>
+ <line number="11" hits="1"/>
<line number="13" hits="1"/>
+ <line number="14" hits="1"/>
+ </lines>
+ </class>
+ <class name="consecutive_sim.py" filename="splitters/consecutive_sim.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="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="28" 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="40" hits="1"/>
+ <line number="41" hits="1"/>
+ <line number="42" hits="1"/>
+ <line number="43" hits="1"/>
+ <line number="50" hits="1"/>
+ <line number="51" hits="1"/>
+ <line number="52" hits="1"/>
+ <line number="53" hits="1"/>
+ </lines>
+ </class>
+ <class name="cumulative_sim.py" filename="splitters/cumulative_sim.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="16" hits="1"/>
+ <line number="22" hits="1"/>
+ <line number="23" hits="1"/>
+ <line number="25" hits="1"/>
+ <line number="26" hits="1"/>
+ <line number="28" hits="1"/>
+ <line number="29" hits="1"/>
+ <line number="32" hits="1"/>
+ <line number="33" hits="1"/>
+ <line number="35" hits="1"/>
+ <line number="36" hits="1"/>
+ <line number="37" 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"/>
+ </lines>
+ </class>
+ </classes>
+ </package>
+ <package name="utils" line-rate="0.4688" 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.3667" 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="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"/>
@@ -991,42 +1589,6 @@
<line number="43" hits="1"/>
</lines>
</class>
- <class name="splitters.py" filename="utils/splitters.py" complexity="0" line-rate="1" branch-rate="0">
- <methods/>
- <lines>
- <line number="1" hits="1"/>
- <line number="3" 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="42" 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="50" hits="1"/>
- <line number="51" hits="1"/>
- <line number="53" hits="1"/>
- <line number="54" hits="1"/>
- <line number="56" hits="1"/>
- <line number="61" hits="1"/>
- <line number="62" hits="1"/>
- <line number="65" hits="1"/>
- <line number="66" hits="1"/>
- <line number="69" hits="1"/>
- <line number="74" hits="1"/>
- <line number="75" hits="1"/>
- </lines>
- </class>
</classes>
</package>
</packages>
diff --git a/poetry.lock b/poetry.lock
index 2498a9a6c5327741198c1a5e1ac6ed2986abe8a9..360f2988bf165befd1edf211f79c4a629589e17d 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -944,13 +944,13 @@ trio = ["trio (>=0.22.0,<0.23.0)"]
[[package]]
name = "httpx"
-version = "0.26.0"
+version = "0.25.2"
description = "The next generation HTTP client."
optional = false
python-versions = ">=3.8"
files = [
- {file = "httpx-0.26.0-py3-none-any.whl", hash = "sha256:8915f5a3627c4d47b73e8202457cb28f1266982d1159bd5779d86a80c0eab1cd"},
- {file = "httpx-0.26.0.tar.gz", hash = "sha256:451b55c30d5185ea6b23c2c793abf9bb237d2a7dfb901ced6ff69ad37ec1dfaf"},
+ {file = "httpx-0.25.2-py3-none-any.whl", hash = "sha256:a05d3d052d9b2dfce0e3896636467f8a5342fb2b902c819428e1ac65413ca118"},
+ {file = "httpx-0.25.2.tar.gz", hash = "sha256:8b8fcaa0c8ea7b05edd69a094e63a2094c4efcb48129fb757361bc423c0ad9e8"},
]
[package.dependencies]
@@ -1319,6 +1319,22 @@ files = [
[package.dependencies]
traitlets = "*"
+[[package]]
+name = "mistralai"
+version = "0.0.12"
+description = ""
+optional = false
+python-versions = ">=3.8,<4.0"
+files = [
+ {file = "mistralai-0.0.12-py3-none-any.whl", hash = "sha256:d489d1f0a31bf0edbe15c6d12f68b943148d2a725a088be0d8a5d4c888f8436c"},
+ {file = "mistralai-0.0.12.tar.gz", hash = "sha256:fe652836146a15bdce7691a95803a32c53c641c5400093447ffa93bf2ed296b2"},
+]
+
+[package.dependencies]
+httpx = ">=0.25.2,<0.26.0"
+orjson = ">=3.9.10,<4.0.0"
+pydantic = ">=2.5.2,<3.0.0"
+
[[package]]
name = "mmh3"
version = "3.1.0"
@@ -1866,6 +1882,65 @@ typing-extensions = ">=4.7,<5"
[package.extras]
datalib = ["numpy (>=1)", "pandas (>=1.2.3)", "pandas-stubs (>=1.1.0.11)"]
+[[package]]
+name = "orjson"
+version = "3.9.13"
+description = "Fast, correct Python JSON library supporting dataclasses, datetimes, and numpy"
+optional = false
+python-versions = ">=3.8"
+files = [
+ {file = "orjson-3.9.13-cp310-cp310-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:fa6b67f8bef277c2a4aadd548d58796854e7d760964126c3209b19bccc6a74f1"},
+ {file = "orjson-3.9.13-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:b812417199eeb169c25f67815cfb66fd8de7ff098bf57d065e8c1943a7ba5c8f"},
+ {file = "orjson-3.9.13-cp310-cp310-manylinux_2_17_armv7l.manylinux2014_armv7l.whl", hash = "sha256:7ccd5bd222e5041069ad9d9868ab59e6dbc53ecde8d8c82b919954fbba43b46b"},
+ {file = "orjson-3.9.13-cp310-cp310-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:eaaf80957c38e9d3f796f355a80fad945e72cd745e6b64c210e635b7043b673e"},
+ {file = "orjson-3.9.13-cp310-cp310-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:60da7316131185d0110a1848e9ad15311e6c8938ee0b5be8cbd7261e1d80ee8f"},
+ {file = "orjson-3.9.13-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:5b98cd948372f0eb219bc309dee4633db1278687161e3280d9e693b6076951d2"},
+ {file = "orjson-3.9.13-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:3869d65561f10071d3e7f35ae58fd377056f67d7aaed5222f318390c3ad30339"},
+ {file = "orjson-3.9.13-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:43fd6036b16bb6742d03dae62f7bdf8214d06dea47e4353cde7e2bd1358d186f"},
+ {file = "orjson-3.9.13-cp310-none-win32.whl", hash = "sha256:0d3ba9d88e20765335260d7b25547d7c571eee2b698200f97afa7d8c7cd668fc"},
+ {file = "orjson-3.9.13-cp310-none-win_amd64.whl", hash = "sha256:6e47153db080f5e87e8ba638f1a8b18995eede6b0abb93964d58cf11bcea362f"},
+ {file = "orjson-3.9.13-cp311-cp311-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:4584e8eb727bc431baaf1bf97e35a1d8a0109c924ec847395673dfd5f4ef6d6f"},
+ {file = "orjson-3.9.13-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:2f37f0cdd026ef777a4336e599d8194c8357fc14760c2a5ddcfdf1965d45504b"},
+ {file = "orjson-3.9.13-cp311-cp311-manylinux_2_17_armv7l.manylinux2014_armv7l.whl", hash = "sha256:d714595d81efab11b42bccd119977d94b25d12d3a806851ff6bfd286a4bce960"},
+ {file = "orjson-3.9.13-cp311-cp311-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:9171e8e1a1f221953e38e84ae0abffe8759002fd8968106ee379febbb5358b33"},
+ {file = "orjson-3.9.13-cp311-cp311-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:1ab9dbdec3f13f3ea6f937564ce21651844cfbf2725099f2f490426acf683c23"},
+ {file = "orjson-3.9.13-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:811ac076855e33e931549340288e0761873baf29276ad00f221709933c644330"},
+ {file = "orjson-3.9.13-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:860d0f5b42d0c0afd73fa4177709f6e1b966ba691fcd72175affa902052a81d6"},
+ {file = "orjson-3.9.13-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:838b898e8c1f26eb6b8d81b180981273f6f5110c76c22c384979aca854194f1b"},
+ {file = "orjson-3.9.13-cp311-none-win32.whl", hash = "sha256:d3222db9df629ef3c3673124f2e05fb72bc4a320c117e953fec0d69dde82e36d"},
+ {file = "orjson-3.9.13-cp311-none-win_amd64.whl", hash = "sha256:978117122ca4cc59b28af5322253017f6c5fc03dbdda78c7f4b94ae984c8dd43"},
+ {file = "orjson-3.9.13-cp312-cp312-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:031df1026c7ea8303332d78711f180231e3ae8b564271fb748a03926587c5546"},
+ {file = "orjson-3.9.13-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:0fd9a2101d04e85086ea6198786a3f016e45475f800712e6833e14bf9ce2832f"},
+ {file = "orjson-3.9.13-cp312-cp312-manylinux_2_17_armv7l.manylinux2014_armv7l.whl", hash = "sha256:446d9ad04204e79229ae19502daeea56479e55cbc32634655d886f5a39e91b44"},
+ {file = "orjson-3.9.13-cp312-cp312-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:b57c0954a9fdd2b05b9cec0f5a12a0bdce5bf021a5b3b09323041613972481ab"},
+ {file = "orjson-3.9.13-cp312-cp312-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:266e55c83f81248f63cc93d11c5e3a53df49a5d2598fa9e9db5f99837a802d5d"},
+ {file = "orjson-3.9.13-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:31372ba3a9fe8ad118e7d22fba46bbc18e89039e3bfa89db7bc8c18ee722dca8"},
+ {file = "orjson-3.9.13-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:e3b0c4da61f39899561e08e571f54472a09fa71717d9797928af558175ae5243"},
+ {file = "orjson-3.9.13-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:2cc03a35bfc71c8ebf96ce49b82c2a7be6af4b3cd3ac34166fdb42ac510bbfff"},
+ {file = "orjson-3.9.13-cp312-none-win_amd64.whl", hash = "sha256:49b7e3fe861cb246361825d1a238f2584ed8ea21e714bf6bb17cebb86772e61c"},
+ {file = "orjson-3.9.13-cp38-cp38-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:62e9a99879c4d5a04926ac2518a992134bfa00d546ea5a4cae4b9be454d35a22"},
+ {file = "orjson-3.9.13-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:d92a3e835a5100f1d5b566fff79217eab92223ca31900dba733902a182a35ab0"},
+ {file = "orjson-3.9.13-cp38-cp38-manylinux_2_17_armv7l.manylinux2014_armv7l.whl", hash = "sha256:23f21faf072ed3b60b5954686f98157e073f6a8068eaa58dbde83e87212eda84"},
+ {file = "orjson-3.9.13-cp38-cp38-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:828c502bb261588f7de897e06cb23c4b122997cb039d2014cb78e7dabe92ef0c"},
+ {file = "orjson-3.9.13-cp38-cp38-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:16946d095212a3dec552572c5d9bca7afa40f3116ad49695a397be07d529f1fa"},
+ {file = "orjson-3.9.13-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:3deadd8dc0e9ff844b5b656fa30a48dbee1c3b332d8278302dd9637f6b09f627"},
+ {file = "orjson-3.9.13-cp38-cp38-musllinux_1_2_aarch64.whl", hash = "sha256:9b1b5adc5adf596c59dca57156b71ad301d73956f5bab4039b0e34dbf50b9fa0"},
+ {file = "orjson-3.9.13-cp38-cp38-musllinux_1_2_x86_64.whl", hash = "sha256:ddc089315d030c54f0f03fb38286e2667c05009a78d659f108a8efcfbdf2e585"},
+ {file = "orjson-3.9.13-cp38-none-win32.whl", hash = "sha256:ae77275a28667d9c82d4522b681504642055efa0368d73108511647c6499b31c"},
+ {file = "orjson-3.9.13-cp38-none-win_amd64.whl", hash = "sha256:730385fdb99a21fce9bb84bb7fcbda72c88626facd74956bda712834b480729d"},
+ {file = "orjson-3.9.13-cp39-cp39-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:7e8e4a571d958910272af8d53a9cbe6599f9f5fd496a1bc51211183bb2072cbd"},
+ {file = "orjson-3.9.13-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:cfad553a36548262e7da0f3a7464270e13900b898800fb571a5d4b298c3f8356"},
+ {file = "orjson-3.9.13-cp39-cp39-manylinux_2_17_armv7l.manylinux2014_armv7l.whl", hash = "sha256:0d691c44604941945b00e0a13b19a7d9c1a19511abadf0080f373e98fdeb6b31"},
+ {file = "orjson-3.9.13-cp39-cp39-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:a8c83718346de08d68b3cb1105c5d91e5fc39885d8610fdda16613d4e3941459"},
+ {file = "orjson-3.9.13-cp39-cp39-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:63ef57a53bfc2091a7cd50a640d9ae866bd7d92a5225a1bab6baa60ef62583f2"},
+ {file = "orjson-3.9.13-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:9156b96afa38db71344522f5517077eaedf62fcd2c9148392ff93d801128809c"},
+ {file = "orjson-3.9.13-cp39-cp39-musllinux_1_2_aarch64.whl", hash = "sha256:31fb66b41fb2c4c817d9610f0bc7d31345728d7b5295ac78b63603407432a2b2"},
+ {file = "orjson-3.9.13-cp39-cp39-musllinux_1_2_x86_64.whl", hash = "sha256:8a730bf07feacb0863974e67b206b7c503a62199de1cece2eb0d4c233ec29c11"},
+ {file = "orjson-3.9.13-cp39-none-win32.whl", hash = "sha256:5ef58869f3399acbbe013518d8b374ee9558659eef14bca0984f67cb1fbd3c37"},
+ {file = "orjson-3.9.13-cp39-none-win_amd64.whl", hash = "sha256:9bcf56efdb83244cde070e82a69c0f03c47c235f0a5cb6c81d9da23af7fbaae4"},
+ {file = "orjson-3.9.13.tar.gz", hash = "sha256:fc6bc65b0cf524ee042e0bc2912b9206ef242edfba7426cf95763e4af01f527a"},
+]
+
[[package]]
name = "packaging"
version = "23.2"
@@ -3388,4 +3463,4 @@ local = ["llama-cpp-python", "torch", "transformers"]
[metadata]
lock-version = "2.0"
python-versions = "^3.9"
-content-hash = "10453196b0249ab854bdcd965ce8631ad9e0db33ae4423d35b250cb8b07b9898"
+content-hash = "9bce8a548105a7226534f2f2e7b1a4d9ec22cab06ada5e4b16ef12156a0983c3"
diff --git a/pyproject.toml b/pyproject.toml
index b396601d8ca30dd802c5116f5493b0571a585f17..c997d0075ca209bb1b2104cda227050fc2a9d0dc 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,6 +19,7 @@ python = "^3.9"
pydantic = "^2.5.3"
openai = "^1.10.0"
cohere = "^4.32"
+mistralai= "^0.0.12"
numpy = "^1.25.2"
colorlog = "^6.8.0"
pyyaml = "^6.0.1"
diff --git a/semantic_router/encoders/__init__.py b/semantic_router/encoders/__init__.py
index b46681541f26631f37d1b3fcfc14410997ea3d85..4f27e96334dd34bf215f0f416db84178bcdde6d9 100644
--- a/semantic_router/encoders/__init__.py
+++ b/semantic_router/encoders/__init__.py
@@ -3,6 +3,7 @@ from semantic_router.encoders.bm25 import BM25Encoder
from semantic_router.encoders.cohere import CohereEncoder
from semantic_router.encoders.fastembed import FastEmbedEncoder
from semantic_router.encoders.huggingface import HuggingFaceEncoder
+from semantic_router.encoders.mistral import MistralEncoder
from semantic_router.encoders.openai import OpenAIEncoder
from semantic_router.encoders.tfidf import TfidfEncoder
from semantic_router.encoders.zure import AzureOpenAIEncoder
@@ -16,4 +17,5 @@ __all__ = [
"TfidfEncoder",
"FastEmbedEncoder",
"HuggingFaceEncoder",
+ "MistralEncoder",
]
diff --git a/semantic_router/encoders/mistral.py b/semantic_router/encoders/mistral.py
new file mode 100644
index 0000000000000000000000000000000000000000..88857de3e61e19984f5ee1036c1097bd9245c8b2
--- /dev/null
+++ b/semantic_router/encoders/mistral.py
@@ -0,0 +1,57 @@
+"""This file contains the MistralEncoder class which is used to encode text using MistralAI"""
+import os
+from time import sleep
+from typing import List, Optional
+
+from mistralai.client import MistralClient
+from mistralai.exceptions import MistralException
+from mistralai.models.embeddings import EmbeddingResponse
+
+from semantic_router.encoders import BaseEncoder
+
+
+class MistralEncoder(BaseEncoder):
+ """Class to encode text using MistralAI"""
+
+ client: Optional[MistralClient]
+ type: str = "mistral"
+
+ def __init__(
+ self,
+ name: Optional[str] = None,
+ mistralai_api_key: Optional[str] = None,
+ score_threshold: float = 0.82,
+ ):
+ if name is None:
+ name = os.getenv("MISTRAL_MODEL_NAME", "mistral-embed")
+ super().__init__(name=name, score_threshold=score_threshold)
+ api_key = mistralai_api_key or os.getenv("MISTRALAI_API_KEY")
+ if api_key is None:
+ raise ValueError("Mistral API key not provided")
+ try:
+ self.client = MistralClient(api_key=api_key)
+ except Exception as e:
+ raise ValueError(f"Unable to connect to MistralAI {e.args}: {e}") from e
+
+ def __call__(self, docs: List[str]) -> List[List[float]]:
+ if self.client is None:
+ raise ValueError("Mistral client not initialized")
+ embeds = None
+ error_message = ""
+
+ # Exponential backoff
+ for _ in range(3):
+ try:
+ embeds = self.client.embeddings(model=self.name, input=docs)
+ if embeds.data:
+ break
+ except MistralException as e:
+ sleep(2**_)
+ error_message = str(e)
+ except Exception as e:
+ raise ValueError(f"Unable to connect to MistralAI {e.args}: {e}") from e
+
+ if not embeds or not isinstance(embeds, EmbeddingResponse) or not embeds.data:
+ raise ValueError(f"No embeddings returned from MistralAI: {error_message}")
+ embeddings = [embeds_obj.embedding for embeds_obj in embeds.data]
+ return embeddings
diff --git a/semantic_router/llms/__init__.py b/semantic_router/llms/__init__.py
index b216f0b0e62fd8bcdf0339ca0b79dbcc910e0c1e..4e2eef16f35d5ef726ed121c6159ab27446ada6a 100644
--- a/semantic_router/llms/__init__.py
+++ b/semantic_router/llms/__init__.py
@@ -1,7 +1,15 @@
from semantic_router.llms.base import BaseLLM
from semantic_router.llms.cohere import CohereLLM
+from semantic_router.llms.mistral import MistralAILLM
from semantic_router.llms.openai import OpenAILLM
from semantic_router.llms.openrouter import OpenRouterLLM
from semantic_router.llms.zure import AzureOpenAILLM
-__all__ = ["BaseLLM", "OpenAILLM", "OpenRouterLLM", "CohereLLM", "AzureOpenAILLM"]
+__all__ = [
+ "BaseLLM",
+ "OpenAILLM",
+ "OpenRouterLLM",
+ "CohereLLM",
+ "AzureOpenAILLM",
+ "MistralAILLM",
+]
diff --git a/semantic_router/llms/mistral.py b/semantic_router/llms/mistral.py
new file mode 100644
index 0000000000000000000000000000000000000000..0f74bcbde93e17f3faf1e95dd4d3bbe791a4155c
--- /dev/null
+++ b/semantic_router/llms/mistral.py
@@ -0,0 +1,56 @@
+import os
+from typing import List, Optional
+
+from mistralai.client import MistralClient
+
+from semantic_router.llms import BaseLLM
+from semantic_router.schema import Message
+from semantic_router.utils.logger import logger
+
+
+class MistralAILLM(BaseLLM):
+ client: Optional[MistralClient]
+ temperature: Optional[float]
+ max_tokens: Optional[int]
+
+ def __init__(
+ self,
+ name: Optional[str] = None,
+ mistralai_api_key: Optional[str] = None,
+ temperature: float = 0.01,
+ max_tokens: int = 200,
+ ):
+ if name is None:
+ name = os.getenv("MISTRALAI_CHAT_MODEL_NAME", "mistral-tiny")
+ super().__init__(name=name)
+ api_key = mistralai_api_key or os.getenv("MISTRALAI_API_KEY")
+ if api_key is None:
+ raise ValueError("MistralAI API key cannot be 'None'.")
+ try:
+ self.client = MistralClient(api_key=api_key)
+ except Exception as e:
+ raise ValueError(
+ f"MistralAI API client failed to initialize. Error: {e}"
+ ) from e
+ self.temperature = temperature
+ self.max_tokens = max_tokens
+
+ def __call__(self, messages: List[Message]) -> str:
+ if self.client is None:
+ raise ValueError("MistralAI client is not initialized.")
+ try:
+ completion = self.client.chat(
+ model=self.name,
+ messages=[m.to_mistral() for m in messages],
+ temperature=self.temperature,
+ max_tokens=self.max_tokens,
+ )
+
+ output = completion.choices[0].message.content
+
+ if not output:
+ raise Exception("No output generated")
+ return output
+ except Exception as e:
+ logger.error(f"LLM error: {e}")
+ raise Exception(f"LLM error: {e}") from e
diff --git a/semantic_router/schema.py b/semantic_router/schema.py
index 46ee7f590c3275b314b9ecb3e52adc54f16012fe..6a5a0637a1190652b39f8d779eb23138e53ebfe3 100644
--- a/semantic_router/schema.py
+++ b/semantic_router/schema.py
@@ -8,6 +8,7 @@ from semantic_router.encoders import (
BaseEncoder,
CohereEncoder,
FastEmbedEncoder,
+ MistralEncoder,
OpenAIEncoder,
)
@@ -17,6 +18,7 @@ class EncoderType(Enum):
FASTEMBED = "fastembed"
OPENAI = "openai"
COHERE = "cohere"
+ MISTRAL = "mistral"
class RouteChoice(BaseModel):
@@ -43,6 +45,8 @@ class Encoder:
self.model = OpenAIEncoder(name=name)
elif self.type == EncoderType.COHERE:
self.model = CohereEncoder(name=name)
+ elif self.type == EncoderType.MISTRAL:
+ self.model = MistralEncoder(name=name)
else:
raise ValueError
@@ -65,6 +69,9 @@ class Message(BaseModel):
def to_llamacpp(self):
return {"role": self.role, "content": self.content}
+ def to_mistral(self):
+ return {"role": self.role, "content": self.content}
+
def __str__(self):
return f"{self.role}: {self.content}"
diff --git a/semantic_router/splitters/base.py b/semantic_router/splitters/base.py
index c7ca37c2ed2660448b4549181e354aa0e5b8727a..edeba73b7e98d366c5e2b5991cbe827853660367 100644
--- a/semantic_router/splitters/base.py
+++ b/semantic_router/splitters/base.py
@@ -1,6 +1,7 @@
from typing import List
from pydantic.v1 import BaseModel
+
from semantic_router.encoders import BaseEncoder
diff --git a/semantic_router/splitters/consecutive_sim.py b/semantic_router/splitters/consecutive_sim.py
index 6bd08845e0d3a45706bc99dd0740ab1355ae3840..55a29a5c62b479a6cd046ddb233ece4760b4f92f 100644
--- a/semantic_router/splitters/consecutive_sim.py
+++ b/semantic_router/splitters/consecutive_sim.py
@@ -1,8 +1,10 @@
from typing import List
-from semantic_router.splitters.base import BaseSplitter
-from semantic_router.encoders import BaseEncoder
+
import numpy as np
+
+from semantic_router.encoders import BaseEncoder
from semantic_router.schema import DocumentSplit
+from semantic_router.splitters.base import BaseSplitter
class ConsecutiveSimSplitter(BaseSplitter):
diff --git a/semantic_router/splitters/cumulative_sim.py b/semantic_router/splitters/cumulative_sim.py
index ba8f4bd31faa15d268acb729aaedb4147d9c97e0..f7a6475ad809a8b1eb877f592cf9ca0799941ba2 100644
--- a/semantic_router/splitters/cumulative_sim.py
+++ b/semantic_router/splitters/cumulative_sim.py
@@ -1,8 +1,10 @@
from typing import List
-from semantic_router.splitters.base import BaseSplitter
+
import numpy as np
-from semantic_router.schema import DocumentSplit
+
from semantic_router.encoders import BaseEncoder
+from semantic_router.schema import DocumentSplit
+from semantic_router.splitters.base import BaseSplitter
class CumulativeSimSplitter(BaseSplitter):
diff --git a/semantic_router/text.py b/semantic_router/text.py
index 978da135eb71f3912c8d49318c743f3c91fc2b47..6038888d1926c59f256dd8b5df8c15851f84e784 100644
--- a/semantic_router/text.py
+++ b/semantic_router/text.py
@@ -1,13 +1,12 @@
-from colorama import Fore
-from colorama import Style
+from typing import List, Literal, Tuple, Union
+from colorama import Fore, Style
from pydantic.v1 import BaseModel, Field
-from typing import Union, List, Literal, Tuple
+
+from semantic_router.encoders import BaseEncoder
+from semantic_router.schema import DocumentSplit, Message
from semantic_router.splitters.consecutive_sim import ConsecutiveSimSplitter
from semantic_router.splitters.cumulative_sim import CumulativeSimSplitter
-from semantic_router.encoders import BaseEncoder
-from semantic_router.schema import Message
-from semantic_router.schema import DocumentSplit
# Define a type alias for the splitter to simplify the annotation
SplitterType = Union[ConsecutiveSimSplitter, CumulativeSimSplitter, None]
diff --git a/tests/unit/encoders/test_mistral.py b/tests/unit/encoders/test_mistral.py
new file mode 100644
index 0000000000000000000000000000000000000000..c2a91c128d22966da177dfc812e37d3bda215b09
--- /dev/null
+++ b/tests/unit/encoders/test_mistral.py
@@ -0,0 +1,124 @@
+import pytest
+from mistralai.exceptions import MistralException
+from mistralai.models.embeddings import EmbeddingObject, EmbeddingResponse, UsageInfo
+
+from semantic_router.encoders import MistralEncoder
+
+
+@pytest.fixture
+def mistralai_encoder(mocker):
+ mocker.patch("mistralai.client.MistralClient")
+ return MistralEncoder(mistralai_api_key="test_api_key")
+
+
+class TestMistralEncoder:
+ def test_mistralai_encoder_init_success(self, mocker):
+ encoder = MistralEncoder(mistralai_api_key="test_api_key")
+ assert encoder.client is not None
+
+ def test_mistralai_encoder_init_no_api_key(self, mocker):
+ mocker.patch("os.getenv", return_value=None)
+ with pytest.raises(ValueError) as _:
+ MistralEncoder()
+
+ def test_mistralai_encoder_call_uninitialized_client(self, mistralai_encoder):
+ # Set the client to None to simulate an uninitialized client
+ mistralai_encoder.client = None
+ with pytest.raises(ValueError) as e:
+ mistralai_encoder(["test document"])
+ assert "Mistral client not initialized" in str(e.value)
+
+ def test_mistralai_encoder_init_exception(self, mocker):
+ mocker.patch(
+ "mistralai.client.MistralClient",
+ side_effect=Exception("Initialization error"),
+ )
+ with pytest.raises(ValueError) as e:
+ MistralEncoder()
+ assert "Mistral API key not provided" in str(e.value)
+
+ def test_mistralai_encoder_call_success(self, mistralai_encoder, mocker):
+ mock_embeddings = mocker.Mock()
+ mock_embeddings.data = [
+ EmbeddingObject(embedding=[0.1, 0.2], index=0, object="embedding")
+ ]
+
+ mocker.patch("os.getenv", return_value="fake-api-key")
+ mocker.patch("time.sleep", return_value=None) # To speed up the test
+
+ mock_embedding = EmbeddingObject(
+ index=0, object="embedding", embedding=[0.1, 0.2]
+ )
+ # Mock the CreateEmbeddingResponse object
+ mock_response = EmbeddingResponse(
+ id="test-id",
+ model="mistral-embed",
+ object="list",
+ usage=UsageInfo(prompt_tokens=1, total_tokens=20, completion_tokens=None),
+ data=[mock_embedding],
+ )
+
+ responses = [MistralException("mistralai error"), mock_response]
+ mocker.patch.object(
+ mistralai_encoder.client, "embeddings", side_effect=responses
+ )
+ embeddings = mistralai_encoder(["test document"])
+ assert embeddings == [[0.1, 0.2]]
+
+ def test_mistralai_encoder_call_with_retries(self, mistralai_encoder, mocker):
+ mocker.patch("os.getenv", return_value="fake-api-key")
+ mocker.patch("time.sleep", return_value=None) # To speed up the test
+ mocker.patch.object(
+ mistralai_encoder.client,
+ "embeddings",
+ side_effect=MistralException("Test error"),
+ )
+ with pytest.raises(ValueError) as e:
+ mistralai_encoder(["test document"])
+ assert "No embeddings returned from MistralAI: Test error" in str(e.value)
+
+ def test_mistralai_encoder_call_failure_non_mistralai_error(
+ self, mistralai_encoder, mocker
+ ):
+ mocker.patch("os.getenv", return_value="fake-api-key")
+ mocker.patch("time.sleep", return_value=None) # To speed up the test
+ mocker.patch.object(
+ mistralai_encoder.client,
+ "embeddings",
+ side_effect=Exception("Non-MistralException"),
+ )
+ with pytest.raises(ValueError) as e:
+ mistralai_encoder(["test document"])
+
+ assert (
+ "Unable to connect to MistralAI ('Non-MistralException',): Non-MistralException"
+ in str(e.value)
+ )
+
+ def test_mistralai_encoder_call_successful_retry(self, mistralai_encoder, mocker):
+ mock_embeddings = mocker.Mock()
+ mock_embeddings.data = [
+ EmbeddingObject(embedding=[0.1, 0.2], index=0, object="embedding")
+ ]
+
+ mocker.patch("os.getenv", return_value="fake-api-key")
+ mocker.patch("time.sleep", return_value=None) # To speed up the test
+
+ mock_embedding = EmbeddingObject(
+ index=0, object="embedding", embedding=[0.1, 0.2]
+ )
+ # Mock the CreateEmbeddingResponse object
+ mock_response = EmbeddingResponse(
+ id="test-id",
+ model="mistral-embed",
+ object="list",
+ usage=UsageInfo(prompt_tokens=1, total_tokens=20, completion_tokens=None),
+ data=[mock_embedding],
+ )
+
+ responses = [MistralException("mistralai error"), mock_response]
+ mocker.patch.object(
+ mistralai_encoder.client, "embeddings", side_effect=responses
+ )
+ embeddings = mistralai_encoder(["test document"])
+ assert embeddings == [[0.1, 0.2]]
diff --git a/tests/unit/llms/test_llm_mistral.py b/tests/unit/llms/test_llm_mistral.py
new file mode 100644
index 0000000000000000000000000000000000000000..4e01ca5af528ea3d92b316c04b2fcf1f19f068c6
--- /dev/null
+++ b/tests/unit/llms/test_llm_mistral.py
@@ -0,0 +1,57 @@
+import pytest
+
+from semantic_router.llms import MistralAILLM
+from semantic_router.schema import Message
+
+
+@pytest.fixture
+def mistralai_llm(mocker):
+ mocker.patch("mistralai.client.MistralClient")
+ return MistralAILLM(mistralai_api_key="test_api_key")
+
+
+class TestMistralAILLM:
+ def test_mistralai_llm_init_with_api_key(self, mistralai_llm):
+ assert mistralai_llm.client is not None, "Client should be initialized"
+ assert mistralai_llm.name == "mistral-tiny", "Default name not set correctly"
+
+ def test_mistralai_llm_init_success(self, mocker):
+ mocker.patch("os.getenv", return_value="fake-api-key")
+ llm = MistralAILLM()
+ assert llm.client is not None
+
+ def test_mistralai_llm_init_without_api_key(self, mocker):
+ mocker.patch("os.getenv", return_value=None)
+ with pytest.raises(ValueError) as _:
+ MistralAILLM()
+
+ def test_mistralai_llm_call_uninitialized_client(self, mistralai_llm):
+ # Set the client to None to simulate an uninitialized client
+ mistralai_llm.client = None
+ with pytest.raises(ValueError) as e:
+ llm_input = [Message(role="user", content="test")]
+ mistralai_llm(llm_input)
+ assert "MistralAI client is not initialized." in str(e.value)
+
+ def test_mistralai_llm_init_exception(self, mocker):
+ mocker.patch(
+ "mistralai.client.MistralClient",
+ side_effect=Exception("Initialization error"),
+ )
+ with pytest.raises(ValueError) as e:
+ MistralAILLM()
+ assert "MistralAI API key cannot be 'None'." in str(e.value)
+
+ def test_mistralai_llm_call_success(self, mistralai_llm, mocker):
+ mock_completion = mocker.MagicMock()
+ mock_completion.choices[0].message.content = "test"
+
+ mocker.patch("os.getenv", return_value="fake-api-key")
+ mocker.patch.object(
+ mistralai_llm.client,
+ "chat",
+ return_value=mock_completion,
+ )
+ llm_input = [Message(role="user", content="test")]
+ output = mistralai_llm(llm_input)
+ assert output == "test"
diff --git a/tests/unit/test_splitters.py b/tests/unit/test_splitters.py
index 165fc2dd4327320f12ed3ea3a68c440b650af1dd..5ee28504dad622cd8cb52c62553f651a5739e990 100644
--- a/tests/unit/test_splitters.py
+++ b/tests/unit/test_splitters.py
@@ -1,15 +1,15 @@
from unittest.mock import Mock, create_autospec
-import pytest
import numpy as np
+import pytest
-from semantic_router.text import Conversation
-from semantic_router.schema import Message
-from semantic_router.splitters.consecutive_sim import ConsecutiveSimSplitter
-from semantic_router.splitters.cumulative_sim import CumulativeSimSplitter
from semantic_router.encoders.base import BaseEncoder
from semantic_router.encoders.cohere import CohereEncoder
+from semantic_router.schema import Message
from semantic_router.splitters.base import BaseSplitter
+from semantic_router.splitters.consecutive_sim import ConsecutiveSimSplitter
+from semantic_router.splitters.cumulative_sim import CumulativeSimSplitter
+from semantic_router.text import Conversation
def test_consecutive_sim_splitter():
diff --git a/tests/unit/test_text.py b/tests/unit/test_text.py
index e7490ec14d43593efa608261e4e6ef1d76915347..51328b8b0035b0e977268c063352165c6b71d902 100644
--- a/tests/unit/test_text.py
+++ b/tests/unit/test_text.py
@@ -1,12 +1,16 @@
-import pytest
from unittest.mock import Mock
-from semantic_router.text import Conversation, Message
-from semantic_router.splitters.consecutive_sim import ConsecutiveSimSplitter
-from semantic_router.splitters.cumulative_sim import CumulativeSimSplitter
+
+import pytest
+
from semantic_router.encoders.cohere import (
CohereEncoder,
-) # Adjust this import based on your project structure
+)
+
+# Adjust this import based on your project structure
from semantic_router.schema import DocumentSplit
+from semantic_router.splitters.consecutive_sim import ConsecutiveSimSplitter
+from semantic_router.splitters.cumulative_sim import CumulativeSimSplitter
+from semantic_router.text import Conversation, Message
@pytest.fixture