Skip to content

Commit 457c58b

Browse files
committed
Update the base model and generator to use tools and outputs
1 parent d048c39 commit 457c58b

File tree

4 files changed

+253
-106
lines changed

4 files changed

+253
-106
lines changed

outlines/__init__.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@
44
from outlines import grammars as grammars
55
from outlines import inputs as inputs
66
from outlines import models as models
7+
from outlines import outputs as outputs
78
from outlines import processors as processors
9+
from outlines import tools as tools
810
from outlines import types as types
911
from outlines.applications import Application as Application
1012
from outlines.caching import clear_cache as clear_cache
@@ -13,10 +15,17 @@
1315
from outlines.generator import Generator as Generator
1416
from outlines.inputs import Audio as Audio
1517
from outlines.inputs import Image as Image
18+
from outlines.inputs import ToolCall as ToolCall
19+
from outlines.inputs import SystemMessage as SystemMessage
20+
from outlines.inputs import UserMessage as UserMessage
21+
from outlines.inputs import AssistantMessage as AssistantMessage
22+
from outlines.inputs import ToolMessage as ToolMessage
23+
from outlines.inputs import Chat as Chat
1624
from outlines.inputs import Video as Video
1725
from outlines.models import * # noqa: F403
1826
from outlines.templates import Template as Template
1927
from outlines.templates import Vision as Vision
28+
from outlines.tools import ToolDef as ToolDef
2029
from outlines.types import cfg as cfg
2130
from outlines.types import json_schema as json_schema
2231
from outlines.types import regex as regex

0 commit comments

Comments
 (0)