harn-stdlib 0.8.26

Embedded Harn standard library source catalog
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

fn __budget_envelope(opts) {
  return opts?.budget
}

/**
 * agent_budget_pre_call_blocked.
 *
 * @effects: [host]
 * @allocation: heap
 * @errors: []
 * @api_stability: experimental
 * @example: agent_budget_pre_call_blocked(session, opts)
 */
pub fn agent_budget_pre_call_blocked(session, opts) {
  let envelope = __budget_envelope(opts)
  if envelope == nil {
    return false
  }
  return __host_agent_budget_pre_call_blocked(session.session_id, envelope)
}

/**
 * agent_budget_post_call_blocked.
 *
 * @effects: []
 * @allocation: heap
 * @errors: []
 * @api_stability: experimental
 * @example: agent_budget_post_call_blocked(totals, opts)
 */
pub fn agent_budget_post_call_blocked(totals, opts) {
  let envelope = __budget_envelope(opts)
  if envelope == nil && opts?.token_budget == nil {
    return false
  }
  if opts?.token_budget != nil && totals?.tokens_used != nil {
    if totals.tokens_used >= opts.token_budget {
      return true
    }
  }
  if envelope?.total_budget_usd != nil && totals?.cost_usd != nil {
    if totals.cost_usd >= envelope.total_budget_usd {
      return true
    }
  }
  return false
}