**kwargs consistiency

This commit is contained in:
2024-10-30 18:46:33 -04:00
parent f828f9991b
commit 1f66bac645
+27
View File
@@ -0,0 +1,27 @@
from typing import Optional, Type
from pydantic import BaseModel
from simplemind import Session
# Example usage:
if __name__ == "__main__":
# Create a session with default settings
session = Session(llm_provider="openai", llm_model="gpt-4o-mini")
# Use the session for text generation
response = session.generate_text("Tell me about the future of AI")
# Create a conversation using session defaults
conversation = session.create_conversation()
# Generate structured data with Pydantic
class Recipe(BaseModel):
name: str
ingredients: list[str]
instructions: list[str]
recipe = session.generate_data(
"Give me a recipe for chocolate chip cookies", response_model=Recipe
)
print(locals())