사용자 정의 비용 추적

import weave
from datetime import datetime

client = weave.init("my_custom_cost_model")

client.add_cost(
    llm_id="your_model_name",
    prompt_token_cost=0.01,
    completion_token_cost=0.02
)

client.add_cost(
    llm_id="your_model_name",
    prompt_token_cost=10,
    completion_token_cost=20,
    # 예를 들어 특정 날짜 이후에 모델 가격을 올리려는 경우
    effective_date=datetime(2025, 4, 22),
)

import weave

client = weave.init("my_custom_cost_model")

costs = client.query_costs(llm_ids=["your_model_name"])

cost = client.query_costs(costs[0].id)

import weave

client = weave.init("my_custom_cost_model")

costs = client.query_costs(llm_ids=["your_model_name"])
client.purge_costs([cost.id for cost in costs])

import weave

weave.init("project_costs")
@weave.op()
def get_costs_for_project(project_name: str):
    total_cost = 0
    requests = 0

    client = weave.init(project_name)
    # 프로젝트의 모든 call을 가져옵니다
    calls = list(
        client.get_calls(filter={"trace_roots_only": True}, include_costs=True)
    )

    for call in calls:
        # call에 비용이 있으면 총비용에 더합니다
        if call.summary["weave"] is not None and call.summary["weave"].get("costs", None) is not None:
            for k, cost in call.summary["weave"]["costs"].items():
                requests += cost["requests"]
                total_cost += cost["prompt_tokens_total_cost"]
                total_cost += cost["completion_tokens_total_cost"]

    # 총비용, 요청 수, call 수를 반환합니다
    return {
        "total_cost": total_cost,
        "requests": requests,
        "calls": len(calls),
    }

# 함수를 @weave.op()으로 데코레이팅했기 때문에
# 총합은 과거 비용 총계를 계산할 수 있도록 Weave에 저장됩니다
get_costs_for_project("my_custom_cost_model")