mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-04-26 01:06:27 +02:00
24 lines
654 B
Python
24 lines
654 B
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
@Time : 2023/5/11 14:45
|
|
@Author : alexanderwu
|
|
@File : llm.py
|
|
"""
|
|
|
|
from typing import Optional
|
|
|
|
from metagpt.config import CONFIG, LLMProviderEnum
|
|
from metagpt.provider.base_llm import BaseLLM
|
|
from metagpt.provider.human_provider import HumanProvider
|
|
from metagpt.provider.llm_provider_registry import LLM_REGISTRY
|
|
|
|
_ = HumanProvider() # Avoid pre-commit error
|
|
|
|
|
|
def LLM(provider: Optional[LLMProviderEnum] = None) -> BaseLLM:
|
|
"""get the default llm provider"""
|
|
if provider is None:
|
|
provider = CONFIG.get_default_llm_provider_enum()
|
|
|
|
return LLM_REGISTRY.get_provider(provider)
|