o
    RTg                     @   s,   d dl mZ d dlmZ G dd deZdS )    )
LLMService)
ChatOllamac                   @   s8   e Zd ZddedefddZdd	ed
edefddZdS )OllamaLLMService	llama3:8bhttp://127.0.0.1:11434modelbase_urlc                 C   s   t ||d| _d S )N)r   r   )r   chat)selfr   r    r   B/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/ollama_llm.py__init__   s   zOllamaLLMService.__init__   ffffff?prompt
max_tokenstemperaturec                 C   s   | j |}|j S )N)r	   invokecontentstrip)r
   r   r   r   responser   r   r   call   s   
zOllamaLLMService.callN)r   r   )r   r   )__name__
__module____qualname__strr   intfloatr   r   r   r   r   r      s    r   N)marker.llms.llm_servicer   langchain_community.chat_modelsr   r   r   r   r   r   <module>   s    