github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-26 03:54:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

lint: fix

Browse Source
This commit is contained in:
csunny
2023-05-25 20:39:04 +08:00
parent 4040e1592a
commit e95696c23d
23 changed files with 277 additions and 173 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pilot/prompts/base.py
View File

@@ -1,5 +1,3 @@
import json
from abc import ABC, abstractmethod
from pathlib import Path
@@ -8,7 +6,7 @@ from typing import Any, Callable, Dict, List, Mapping, Optional, Set, Union
import yaml
from pydantic import BaseModel, Extra, Field, root_validator
from pilot.scene.base_message import BaseMessage,HumanMessage,AIMessage, SystemMessage
from pilot.scene.base_message import BaseMessage, HumanMessage, AIMessage, SystemMessage
def get_buffer_string(
@@ -29,7 +27,6 @@ def get_buffer_string(
return "\n".join(string_messages)
class PromptValue(BaseModel, ABC):
@abstractmethod
def to_string(self) -> str:
@@ -39,6 +36,7 @@ class PromptValue(BaseModel, ABC):
def to_messages(self) -> List[BaseMessage]:
"""Return prompt as messages."""
class ChatPromptValue(PromptValue):
messages: List[BaseMessage]
@@ -48,4 +46,4 @@ class ChatPromptValue(PromptValue):
def to_messages(self) -> List[BaseMessage]:
"""Return prompt as messages."""
return self.messages
return self.messages