dspy-optimizer-experiment/schema.py at main · kmad/dspy-optimizer-experiment · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
from pydantic import BaseModel
from typing import Optional, List


class ExtractedEntities(BaseModel):
    Company: Optional[List[str]] = None
    Date: Optional[List[str]] = None
    Location: Optional[List[str]] = None
    Money: Optional[List[str]] = None
    Person: Optional[List[str]] = None
    Product: Optional[List[str]] = None
    Quantity: Optional[List[str]] = None