24 lines
741 B
Python
24 lines
741 B
Python
"""This module is adapt from https://github.com/zeno-ml/zeno-build"""
|
|
try:
|
|
from .providers.gemini_utils import generate_from_gemini_completion
|
|
except:
|
|
print('Google Cloud not set up, skipping import of providers.gemini_utils.generate_from_gemini_completion')
|
|
|
|
from .providers.hf_utils import generate_from_huggingface_completion
|
|
from .providers.openai_utils import (
|
|
generate_from_openai_chat_completion,
|
|
generate_from_openai_completion,
|
|
)
|
|
from .providers.api_utils import (
|
|
generate_with_api,
|
|
)
|
|
from .utils import call_llm
|
|
|
|
__all__ = [
|
|
"generate_from_openai_completion",
|
|
"generate_from_openai_chat_completion",
|
|
"generate_from_huggingface_completion",
|
|
"generate_from_gemini_completion",
|
|
"call_llm",
|
|
]
|