MD2204/multi_modality
1
1o
2 ��wi � @ sT d dl Z d dlZd dlmZ d dlmZmZmZmZ defdd�Z de3fdd�Z dS ) � N)�4ChatOllama)� OLLAMA_MODEL�OLLAMA_BASE_URL�OLLAMA_API_KEY�OLLAMA_TEMPERATURE�returnc C s& i } t r dt � �| d< tttt| d�S )z Initialize ChatOllama instance.zBearer �
Authorization)�model�base_url� temperature�headersN)r r r r r )r � r
�GC:\Users\MoEdBouk\Desktop\AI ENGENEER PATH\multimodality_RAG\src\llm.py�get_ollama_llm s �r c 5 C sr z t dt� dt� d�� t� } | �d�}t d|j� �� W dS ty8 } zt d|� �� W Y d}~d S d}~ww )6zTest connection to Ollama.zTesting connection to z with model z...zHello, are you there?u% ✓ Connection successful. Response: Tu ❌ Connection failed: NF)�printr r r �invoke�content� Exception)�llm�response�er
r
r �test_llm_connection s 7��r ) �os�requestsZlangchain_ollamar �8src.configr r r r r �boolr r
r
r
r �<module> s