Streaming input!

[rgbkrk]

This adds a new decorator @incremental_display that lets ChatLab know what you'd like to render streaming arguments with.

import warnings
from graphviz import Digraph
from pydantic import BaseModel, Field
from typing import List
from chatlab import Chat, system
from chatlab.decorators import incremental_display

class Node(BaseModel):
    id: int
    label: str
    color: str

class Edge(BaseModel):
    source: int
    target: int
    label: str
    color: str = Field("black", description="A valid Graphviz color name or hex code.")

class KnowledgeGraph(BaseModel):
    nodes: List[Node] = Field(..., default_factory=list)
    edges: List[Edge] = Field(..., default_factory=list)


def visualize_knowledge_graph(kg: KnowledgeGraph, comment: str = "Knowledge Graph"):
    """Visualizes a knowledge graph using graphviz."""
    if kg is None:
        return

    dot = Digraph(comment=comment)

    for node in kg.nodes:
        dot.node(str(node.id), node.label, color=node.color)

    for edge in kg.edges:
        dot.edge(str(edge.source), str(edge.target), label=edge.label, color=edge.color)

    return dot


@incremental_display(visualize_knowledge_graph)
def store_knowledge_graph(kg: KnowledgeGraph, comment: str):
    """Creates a graphviz diagram for the user and stores it in their database."""
    return "Stored graph, visual shown"

chat = Chat(
    model="gpt-4-turbo-preview",
    chat_functions=[store_knowledge_graph],
)

await chat("Visually teach me about Ice Cream", temperature=0.7)