Управление контекстом диалога

python3 -m venv new-env
source new-env/bin/activate

pip install openai

import openai

YC_MODEL = "yandexgpt"

def print_conversation(conv_id):
    print(f"\nConversation {conv_id}")
    items = client.conversations.items.list(conversation_id=conv_id)
    print(f"\nFound {len(items.data)} items in conversation:")
    
    for item in items.data:
        print(f"\nItem: {item}")
    print("-" * 80)


client = openai.OpenAI(
    api_key=YC_API_KEY,
    base_url="https://ai.api.cloud.yandex.net/v1",
    project=YC_FOLDER_ID
)

# 1) создаём conversation
conv = client.conversations.create()
print("conversation id:", conv.id)
print_conversation(conv.id)

# 2) первое сообщение с системной инструкцией и пользовательским вводом
r1 = client.responses.create(
    model=f"gpt://{YC_FOLDER_ID}/{YC_MODEL}",
    conversation=conv.id,
    input=[
        {"role": "developer", "content": "Ты мой ассистент."},
        {"role": "user", "content": "Привет! Запомни: я живу в Серпухове."}
    ]
)
print("assistant:", r1.output_text)
print_conversation(conv.id)


# 3) продолжаем в том же conversation
r2 = client.responses.create(
    model=f"gpt://{YC_FOLDER_ID}/{YC_MODEL}",
    conversation=conv.id,
    input="В каком городе я живу?"
)
print("assistant:", r2.output_text)
print_conversation(conv.id)


# 4) вызов инструмента web_search
r3 = client.responses.create(
    model=f"gpt://{YC_FOLDER_ID}/{YC_MODEL}",
    conversation=conv.id,
    input="Какая погода будет на выходных?",
    tools=[{
        "type": "web_search",
    }]
)
print("assistant:", r3.output_text)
print_conversation(conv.id)

export YC_FOLDER_ID=<идентификатор_каталога>
export YC_API_KEY=<API-ключ>

python dialogue.py

import openai     

YC_MODEL = "yandexgpt"

previous_id = None  # параметр для сохранения ID последнего ответа ассистента

client = openai.OpenAI(
    api_key=YC_API_KEY,
    project=YC_FOLDER_ID,
    base_url="https://ai.api.cloud.yandex.net/v1",
)

print("Чат с агентом (введите 'выход' для выхода)\n")

while True:
    user_input = input("Вы: ")
    if user_input.lower() in ("exit", "quit", "выход"):
        print("Чат завершён.")
        break

    response = client.responses.create(
        model=f"gpt://{YC_FOLDER_ID}/{YC_MODEL}",
        input=user_input,
        instructions="Ты — текстовый агент, который ведёт диалог и даёт информативные ответы на вопросы пользователя.",
        previous_response_id=previous_id  # передаем контекст, если он есть
    )

    # сохраняем ID для следующего шага
    previous_id = response.id

    # выводим ответ агента
    print("Агент:", response.output_text)

export YC_FOLDER_ID=<идентификатор_каталога>
export YC_API_KEY=<API-ключ>

python dialogue.py

import openai     

YC_MODEL = "yandexgpt"

client = openai.OpenAI(
    api_key=YC_API_KEY,
    base_url="https://ai.api.cloud.yandex.net/v1",
    project=YC_FOLDER_ID,
)

response = client.responses.create(
    model=f"gpt://{YC_FOLDER_ID}/{YC_MODEL}",
    input=[
        {"role": "user", "content": "Привет! Помоги с выбором подарка самому себе."},
        {"role": "assistant", "content": "Привет! Расскажи о себе"},
        {"role": "user", "content": "Меня зовут Глеб."},
        {"role": "user", "content": "Я люблю путешествовать."},
        {"role": "user", "content": "Увлекаюсь аниме и дизайном."},
    ],
)

print(response.output_text)

export YC_FOLDER_ID=<идентификатор_каталога>
export YC_API_KEY=<API-ключ>

python dialogue.py