from openai import OpenAI

client = OpenAI(base_url="https://192.168.0.107:8000/v1", api_key="any")

stream = client.chat.completions.create(
    model = "/model",
    messages = [
        {"role": "system", "content": "Ты полезный ассистент."},
        {"role": "user", "content": "Сколько весит Луна?"}
    ],
    temperature = 0.1,
    stream = True  # Включаем потоковый режим
)

# Обрабатываем потоковый ответ
for chunk in stream:
    if chunk.choices[0].delta.content is not None:
        print(chunk.choices[0].delta.content, end="", flush=True)

from openai import OpenAI
client = OpenAI(
    base_url="https://192.168.0.108:8000/v1",
    api_key="any"
)

completion = client.chat.completions.create(
    model="/model",
    messages=[
        {"role": "system", "content": "Ты полезный ассистент."},
        {"role": "user", "content": "Классифицируй этот отзыв: С удовольствием и интересом посмотрел этот фильм. "}
    ],
    extra_body={"guided_choice": ["positive", "negative"]}
)
print(completion.choices[0].message.content)

completion = client.chat.completions.create(
    model="/model",
    messages=[
        {"role": "system", "content": "Ты полезный ассистент."},
        {"role": "user", "content": "Сгенерируй адрес электронной почты."}
    ],
    extra_body={"guided_regex": r"\w+@\w+\.com"},
)
print(completion.choices[0].message.content)

from pydantic import BaseModel
from enum import Enum

class EstateType(str, Enum):
    house = "дом"
    apartment = "квартира"

class CarDescription(BaseModel):
    price: int
    square: int
    area: int
    estate_type: EstateType

json_schema = CarDescription.model_json_schema()

completion = client.chat.completions.create(
    model="/model",
    messages=[
        {"role": "system", "content": "Ты полезный ассистент."},
        {
            "role": "user",
            "content": """Вытащи из следующего объявления инфомрмацию в формате JSON:
            {
                price: цена в рублях,
                square: площадь дома в кв. м.,
                area: площадь участка в сотках,
                estate_type: дом или квартира
            }

            
            === Объявление ===
            Предлагаем купить теплый двухэтажный кирпичный коттедж площадью 225 кв. м. для круглогодичного проживания дружной семьи, предпочитающей уют и свободу.
            На первом этаже – просторная уютная гостиная с камином, шикарным диваном, кухня, отделенная от столовой барной стойкой, кладовая, санузел.
            Два выхода – центральный и на задний дворик. На втором этаже – три спальни с балконами, два санузла с ванными.
            В каждой комнате установлены радиаторы. Все комнаты меблированы. Новый, стильный ремонт.
            Перед входом – парковка для двух автомобилей. Участок – 5 соток, есть зона для барбекю, веранда, баня.
            До ближайшей автобусной остановки – 300 м. В пешей доступности есть продуктовый магазин, аптека, детская площадка.
            Дом сдан в эксплуатацию. Документы для продажи готовы.
            Цена: 10 млн.р."""
        }
    ],
    extra_body={"guided_json": json_schema},
)
print(completion.choices[0].message.content)