§
    ˜Štj+  ã                   óB   — d dl Z d dlmZ d	d„Zd	d„Zd	d„Zd	d„Zd	d„ZdS )
é    N)ÚOllamaEmbeddingFunctionÚreturnc                  óÐ   — t          j        dd¬¦  «         t          ¦   «         }  | ddg¦  «        }|€J ‚t          |¦  «        dk    sJ ‚t	          d„ |D ¦   «         ¦  «        sJ ‚d S )NÚollamaúollama not installed©Úreasonú"Here is an article about llamas...úthis is another articleé   c              3   ó<   K  — | ]}t          |¦  «        d k    V — ŒdS )i€  N)Úlen)Ú.0Úes     ú]/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/chromadb/test/ef/test_ollama_ef.pyú	<genexpr>z,test_ollama_default_model.<locals>.<genexpr>   s,   è è € Ð1Ð1 �s�1‰vŒv˜Š}Ð1Ð1Ð1Ð1Ð1Ð1ó    )ÚpytestÚimportorskipr   r   Úall©ÚefÚ
embeddingss     r   Útest_ollama_default_modelr      s†   € Ý
Ô˜Ð)?Ð@Ñ@Ô@Ð@Ý	 Ñ	"Ô	"€BØ�Ð9Ð;TÐUÑVÔV€JØÐ!Ð!Ð!Ýˆz‰?Œ?˜aÒÐÐÐÝÐ1Ð1 jÐ1Ñ1Ô1Ñ1Ô1Ð1Ð1Ð1Ð1Ð1r   c                  ó
  — t          j        dd¬¦  «         d} t          | ¬¦  «        }t          j        t          ¦  «        5 } |ddg¦  «         d d d ¦  «         n# 1 swxY w Y   d| › d	�t          |j        ¦  «        v sJ ‚d S )
Nr   r   r   zunknown-model)Ú
model_namer
   r   zmodel "z" not found©r   r   r   ÚraisesÚ	ExceptionÚstrÚvalue)r   r   r   s      r   Útest_ollama_unknown_modelr"      sá   € Ý
Ô˜Ð)?Ð@Ñ@Ô@Ð@Ø €JÝ	 ¨JÐ	7Ñ	7Ô	7€BÝ	Œ•yÑ	!Ô	!ð N QØ
ˆÐ0Ð2KÐLÑMÔMÐMðNð Nð Nñ Nô Nð Nð Nð Nð Nð Nð Nøøøð Nð Nð Nð Nà,�ZÐ,Ð,Ð,µ°A´G±´Ð<Ð<Ð<Ð<Ð<Ð<s   ÁAÁA Á#A c                  ót   — t          j        dd¬¦  «         t          d¬¦  «        }  | ddg¦  «        }|€J ‚d S )Nr   r   r   z%http://localhost:11434/api/embeddings©Úurlr
   r   )r   r   r   r   s     r   Útest_ollama_backward_compatr&      sR   € Ý
Ô˜Ð)?Ð@Ñ@Ô@Ð@Ý	 Ð%LÐ	MÑ	MÔ	M€BØ�Ð9Ð;TÐUÑVÔV€JØÐ!Ð!Ð!Ð!Ð!r   c                  óþ   — t          j        dd¬¦  «         t          d¬¦  «        } t          j        t          ¦  «        5 } | ddg¦  «         d d d ¦  «         n# 1 swxY w Y   dt          |j        ¦  «        v sJ ‚d S )	Nr   r   r   z$http://localhost:11434/this_is_wrongr$   r
   r   Ú404r   )r   r   s     r   Útest_wrong_urlr)   !   sÒ   € Ý
Ô˜Ð)?Ð@Ñ@Ô@Ð@Ý	 Ð%KÐ	LÑ	LÔ	L€BÝ	Œ•yÑ	!Ô	!ð N QØ
ˆÐ0Ð2KÐLÑMÔMÐMðNð Nð Nñ Nô Nð Nð Nð Nð Nð Nð Nøøøð Nð Nð Nð Nà•C˜œ‘L”LÐ Ð Ð Ð Ð Ð s   Á AÁAÁ!Ac                  ó  — 	 ddl m}  t          j        d¦  «         n# t          $ r Y nw xY wt          j        t          ¦  «        5 }t          ¦   «          d d d ¦  «         n# 1 swxY w Y   dt          |j	        ¦  «        v sJ ‚d S )Nr   )ÚClientz"ollama python package is installedz*The ollama python package is not installed)
r   r+   r   ÚskipÚImportErrorr   Ú
ValueErrorr   r    r!   )r+   r   s     r   Útest_ollama_ask_user_to_installr/   )   sà   € ð:Ø!Ð!Ð!Ð!Ð!Ð!õ 	ŒÐ8Ñ9Ô9Ð9Ð9øõ ð ð ð Øˆðøøøõ 
Œ•zÑ	"Ô	"ð " aÝÑ!Ô!Ð!ð"ð "ð "ñ "ô "ð "ð "ð "ð "ð "ð "øøøð "ð "ð "ð "à7½3¸q¼w¹<¼<ÐGÐGÐGÐGÐGÐGs   ‚ �
*©*ÁA!Á!A%Á(A%)r   N)r   Ú<chromadb.utils.embedding_functions.ollama_embedding_functionr   r   r"   r&   r)   r/   © r   r   ú<module>r2      s    ðØ €€€ðð ð ð ð ð ð
2ð 2ð 2ð 2ð=ð =ð =ð =ð"ð "ð "ð "ð!ð !ð !ð !ð	Hð 	Hð 	Hð 	Hð 	Hð 	Hr   