feat(core): APP use new SDK component (#1050)

This commit is contained in:
Fangyin Cheng
2024-01-10 10:39:04 +08:00
committed by GitHub
parent e11b72c724
commit fa8b5b190c
242 changed files with 2768 additions and 2163 deletions

View File

@@ -1,10 +1,10 @@
from abc import ABC, abstractmethod
from typing import Dict, Optional, List, Any
import logging
from abc import ABC, abstractmethod
from typing import Any, Dict, List, Optional
from dbgpt.core import ModelMessage
from dbgpt.model.base import ModelType
from dbgpt.model.adapter.base import LLMModelAdapter, register_model_adapter
from dbgpt.model.base import ModelType
logger = logging.getLogger(__name__)
@@ -42,7 +42,7 @@ class NewHFChatModelAdapter(LLMModelAdapter, ABC):
def load(self, model_path: str, from_pretrained_kwargs: dict):
try:
import transformers
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModel
from transformers import AutoModel, AutoModelForCausalLM, AutoTokenizer
except ImportError as exc:
raise ValueError(
"Could not import depend python package "