test_openaischema.py

gistfile1.txt

from instructor import OpenAISchema
from pydantic import Field
from typing import List
import enum

import openai

class Source(enum.Enum):
    VIDEO = "VIDEO"
    TRANSCRIPT = "TRANSCRIPT"
    DOCUMENT = "DOCUMENT"

class Search(OpenAISchema):
    query: str = Field(..., description="The search query.")
    source: Source

    def search(self):
        return f"Fake results: `{self. query}` from {self. source}"
    
class MultiSearch(OpenAISchema):
    "correct segmentation of `Search` tasks"
    tasks: List[Search]

print(MultiSearch.openai_schema)
# MultiSearch.openai_schena['name']