[1bdb11]: / aitrika / llm / base_llm.py

Download this file

21 lines (15 with data), 312 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
from abc import ABC, abstractmethod
class BaseLLM(ABC):
"""
Base LLM class.
"""
@abstractmethod
def query(self, query: str) -> str:
"""
Query method for LLM.
Args:
query (str): Query
Returns:
str: Response
"""
pass