使用 Lyzr SDK 构建个性化礼品助手

import streamlit as st
from lyzr_automata.ai_models.openai import OpenAIModel
from lyzr_automata import Agent, Task
from lyzr_automata.pipelines.linear_sync_pipeline import LinearSyncPipeline
from PIL import Image
from lyzr_automata.tasks.task_literals import InputType, OutputType
import os

os.environ["OPENAI_API_KEY"] = st.secrets["apikey"]

st.title("Personalized Gift Assistant")
st.markdown("Welcome to Personalized Gift Assistant! Let us help you find the perfect gift for any occasion, tailored to your recipient's unique interests and your budget.")
st.markdown("1) Mention your receiver's age.")
st.markdown("2) Mention your receiver's interest.")
st.markdown("3) Mention the occasion.")
st.markdown("4) Mention your budget.")
input = st.text_input("Please enter the above details:", placeholder="Type here")

open_ai_text_completion_model = OpenAIModel(
    api_key=st.secrets["apikey"],
    parameters={
        "model": "gpt-4-turbo-preview",
        "temperature": 0.2,
        "max_tokens": 1500,
    },
)

def generation(input):
    generator_agent = Agent(
        role="Expert GIFT CONSULTANT",
        prompt_persona="Your task is to CURATE a personalized list of 5-7 GIFTS for the user and provide EXPLANATIONS for each choice, taking into account the RECEIVER'S AGE, RECEIVER'S INTERESTS, the OCCASION, and the BUDGET.")
    prompt = """
[Prompts here]
"""
    generator_agent_task = Task(
        name="Generation",
        model=open_ai_text_completion_model,
        agent=generator_agent,
        instructions=prompt,
        default_input=input,
        output_type=OutputType.TEXT,
        input_type=InputType.TEXT,
    ).execute()
    return generator_agent_task

if st.button("Assist!"):
    solution = generation(input)
    st.markdown(solution)