#!/usr/bin/env python
# -*- encoding: utf-8 -*-
"""
@Time    :   2024/09/03 15:14:06
@Author  :   ChenHao
@Description  :   对接ollama模型
@Contact :   jerrychen1990@gmail.com
"""

from loguru import logger

from liteai.utils import set_logger
from tests.base import BasicTestCase


class TestOllama(BasicTestCase):
    @classmethod
    def setUpClass(cls):
        set_logger(__name__)
        logger.info("start test ollama job")

    def test_basic_llm(self):
        super().basic_llm(model="ollama/glm4", base_url="http://36.103.167.117:11437/")
