o
    g                     @   sB   d dl Z d dlZd dlZd dlmZmZ d dlmZ dddZdS )    N)	HTTPErrorConnectError)OllamaEmbeddingFunctionreturnc               	   C   s   t jddu st jddu rtd ztt jdd} |   W n ttfy5   td Y nw t	t jdp>dt jd dd	}|d
dg}t
|dksWJ dS )z
    To set up the Ollama server, follow instructions at: https://github.com/ollama/ollama?tab=readme-ov-file
    Export the OLLAMA_SERVER_URL and OLLAMA_MODEL environment variables.
    OLLAMA_SERVER_URLNOLLAMA_MODELzNOLLAMA_SERVER_URL or OLLAMA_MODEL environment variable not set. Skipping test. z)Ollama server not running. Skipping test.znomic-embed-textz/embeddings)
model_nameurlz"Here is an article about llamas...zthis is another article   )osenvirongetpytestskiphttpxraise_for_statusr   r   r   len)responseef
embeddings r   V/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/test/ef/test_ollama_ef.pytest_ollama
   s"   r   )r   N)r   r   r   r   r   "chromadb.utils.embedding_functionsr   r   r   r   r   r   <module>   s    