ei+@ddlZddlmZddZddZddZddZddZy) N)OllamaEmbeddingFunctionctjddt}|ddg}|Jt|dk(sJt d|DsJy)Nollamaollama not installedreason"Here is an article about llamas...this is another articlec38K|]}t|dk(yw)iN)len).0es i/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/chromadb/test/ef/test_ollama_ef.py z,test_ollama_default_model..s1s1v}1s)pytest importorskiprr allef embeddingss rtest_ollama_default_modelrs^ )?@ "B9;TUVJ  !! ! z?a   1j1 11 1ctjddd}t|}tjt5}|ddgdddd|d t j vsJy#1swY'xYw) Nrrrz unknown-model) model_namer r zmodel "z " not foundrrrraises Exceptionstrvalue)rrrs rtest_ollama_unknown_modelr!sw )?@ J J 7B y !NQ 02K LMN ZL ,AGG << <NNs A00A9cftjddtd}|ddg}|Jy)Nrrrz%http://localhost:11434/api/embeddingsurlr r )rrrrs rtest_ollama_backward_compatr%s; )?@ %L MB9;TUVJ  !! !rctjddtd}tjt5}|ddgddddt j vsJy#1swY#xYw) Nrrrz$http://localhost:11434/this_is_wrongr#r r 404r)rrs rtest_wrong_urlr(!sj )?@ %K LB y !NQ 02K LMN CL  NNs A**A3c ddlm}tjdtj t 5}tddddtjvsJy#t$rYQwxYw#1swY2xYw)Nr)Clientz"ollama python package is installedz*The ollama python package is not installed) rr*rskip ImportErrorr ValueErrorrrr )r*rs rtest_ollama_ask_user_to_installr.)sl:!  89 z ""a!" 73qww< GG G    ""sA# A2# A/.A/2A;)returnN)rr2s'  2="! Hr