FROM qwen2.5:3b
PARAMETER num_gpu 0

"""누구에게 물어볼지를 여기서 한 번만 정한다.

다른 파일은 전부 이렇게 쓴다 ─
    from config import client, MODEL
"""

from openai import OpenAI

# 어디로 보낼지. localhost = 내 컴퓨터 자신, 11434 = 올라마가 쓰는 문 번호.
# 끝의 /v1 은 "표준 방식으로 말을 걸겠다"는 표시다. 빠뜨리면 404 가 난다
BASE_URL = "http://localhost:11434/v1"

# 원래는 열쇠를 넣는 자리인데, 내 컴퓨터는 검사를 안 한다.
# 그래도 비워두면 패키지가 에러를 내므로 아무 글자나 채운다
API_KEY = "ollama"

# 방금 만든 판의 이름
MODEL = "qwen-cpu"

# 이 한 줄로 "말을 걸 상대"가 만들어진다. 다른 파일들은 이걸 가져다 쓴다
client = OpenAI(base_url=BASE_URL, api_key=API_KEY)

# config.py 에서 client 와 MODEL 을 가져온다.
# 접속 설정이 저기 한 군데에만 있으니 여기엔 주소도 열쇠도 안 적힌다
from config import client, MODEL

res = client.chat.completions.create(
    model=MODEL,

    # messages 는 주고받은 말의 목록이다. 지금은 우리가 한 말 하나뿐이다.
    #   role="user"       내가 하는 말
    #   role="assistant"  모델이 한 말 (이어서 물을 때 넣는다)
    #   role="system"     "너는 이런 역할이다" 라는 지시 (뒤에서 쓴다)
    messages=[
        {"role": "user", "content": "화장품 추천 서비스를 한 문장으로 소개해줘"},
        {"role": "system", "content": "한국어로 답변해줘" },
    ],
)

# 답이 여러 개 올 수도 있는 구조라 목록이다. 보통 하나뿐이라 [0] 으로 꺼낸다.
# choices[0].message.content 가 실제 답 글자다
print(res.choices[0].message.content)

# usage 에는 이번 호출에 쓴 토큰 수가 들어 있다.
# 내 컴퓨터에서 돌리면 돈은 안 들지만, 이 숫자만큼 시간이 든다
print("입력", res.usage.prompt_tokens, "출력", res.usage.completion_tokens)

import time                       # 시간을 재는 도구. 파이썬에 기본으로 들어 있다

from config import client, MODEL

prompt = """[고객] 41세 여성 건성
[이력] 병풀 링클 클렌징오일 28100(별점3) / 히알루론산 탄력 클렌징오일 25000(별점4)
[후보]
P115 비타민C 트러블 클렌징오일 20400
P140 히알루론산 시카 클렌징오일 23200
P096 레티놀 브라이트닝 클렌징오일 24400

어울릴 상품 두 개를 골라라."""

for i in range(3):                      # 세 번 돌린다. 한 번만 재면 못 믿는다
    started = time.time()                   # 지금 시각 (1970년부터 흐른 초)

    res = client.chat.completions.create(
        model=MODEL,
        messages=[{"role": "user", "content": prompt}],
    )

    took = time.time() - started            # 끝난 시각 - 시작 시각 = 걸린 초

    n_out = res.usage.completion_tokens
    print(f"{i + 1}회  {took:5.1f}초 · 출력 {n_out:3d}토큰 · 초당 {n_out / took:.1f}토큰")

# urllib 은 파이썬에 기본으로 들어 있는 인터넷 도구다 (URL 을 다루는 도구 묶음).
# 설치할 것이 없어서 어떤 컴퓨터에서든 그냥 돈다
import json
import urllib.request

# Request 는 "보낼 요청서"를 만든다. 아직 보내지는 않는다
req = urllib.request.Request(
    "http://localhost:11434/api/create",       # 11434 = 올라마의 문 번호

    # 보낼 내용. ollama create 에 준 것과 같은 내용을 형식만 바꾼 것이다.
    # json.dumps 는 파이썬 값을 글자로 바꾸고,
    # .encode() 는 그 글자를 바이트로 바꾼다 — 컴퓨터끼리는 바이트로 오간다
    data=json.dumps({
        "model": "qwen-cpu",                   # 만들 이름
        "from": "qwen2.5:3b",                  # 바탕이 될 모델
        "parameters": {"num_gpu": 0},          # Modelfile 의 PARAMETER 와 같다
        "stream": False,                       # 진행 상황을 조금씩 말고 한 번에 받는다
    }).encode(),

    # "보내는 내용이 JSON 형식이다" 라고 알려준다. 안 적으면 상대가 못 알아본다
    headers={"Content-Type": "application/json"},
)

# 이제 실제로 보낸다. timeout=180 은 "3분 기다려도 답이 없으면 포기한다"
with urllib.request.urlopen(req, timeout=180) as res:
    # read() 는 바이트로 받는다 -> decode() 로 사람이 읽는 글자로 되돌린다
    print(res.read().decode())