Add speakers cost
This commit is contained in:
parent
9e7f018d6f
commit
7ce2b309f5
1 changed files with 20 additions and 0 deletions
|
|
@ -3058,6 +3058,26 @@ def finalizeSpeakerParse():
|
||||||
THREAD_CTX.in_speaker = False
|
THREAD_CTX.in_speaker = False
|
||||||
except Exception:
|
except Exception:
|
||||||
pass
|
pass
|
||||||
|
# Record token usage so it appears in the TOTAL string
|
||||||
|
try:
|
||||||
|
with LOCK:
|
||||||
|
TOKENS[0] += resp[1][0]
|
||||||
|
TOKENS[1] += resp[1][1]
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
# Emit a one-time summary line for speaker translation using the same format
|
||||||
|
try:
|
||||||
|
cost = calculateCost(resp[1][0], resp[1][1], MODEL)
|
||||||
|
totalTokenstring = (
|
||||||
|
Fore.YELLOW + "[Input: " + str(resp[1][0]) + "]"
|
||||||
|
"[Output: "
|
||||||
|
+ str(resp[1][1])
|
||||||
|
+ "]" "[Cost: ${:,.4f}".format(cost)
|
||||||
|
+ "]"
|
||||||
|
)
|
||||||
|
tqdm.write("Speakers: " + totalTokenstring + Fore.GREEN + " \u2713 " + Fore.RESET)
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
tl_list = resp[0]
|
tl_list = resp[0]
|
||||||
with _speakerCacheLock:
|
with _speakerCacheLock:
|
||||||
for orig, tl in zip(to_translate, tl_list):
|
for orig, tl in zip(to_translate, tl_list):
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue