support function calling

Former-commit-id: 66533b3f65babf2429c92c0f8fafe4eff5e0ff63
This commit is contained in:
hiyouga
2024-01-18 09:54:23 +08:00
parent f7329b1a0e
commit a423274fd9
67 changed files with 1239 additions and 1079 deletions

View File

@@ -1,15 +1,17 @@
import time
from enum import Enum
from enum import Enum, unique
from pydantic import BaseModel, Field
from typing import List, Optional
@unique
class Role(str, Enum):
USER = "user"
ASSISTANT = "assistant"
SYSTEM = "system"
@unique
class Finish(str, Enum):
STOP = "stop"
LENGTH = "length"