go : add support for whisper_full_lang_id() (#1010)

* * Add support for whisper_full_lang_id() to go bindings

* Expose token.id so we can test beg, eot etc

---------

Co-authored-by: Jay Binks <redacted>

diff --git a/bindings/go/whisper.go b/bindings/go/whisper.go
index 451f3f8d6563787e4bc3dd4c6f29614aa9cab9db..8a5efa7de0cd012ff57c3cea9769cff14925e290 100644 (file)
--- a/bindings/go/whisper.go
+++ b/bindings/go/whisper.go
@@ -338,6 +338,18 @@ func (ctx *Context) Whisper_full_parallel(params Params, samples []float32, proc
        }
 }
 
+// Return the id of the autodetected language, returns -1 if not found
+// Added to whisper.cpp in
+// https://github.com/ggerganov/whisper.cpp/commit/a1c1583cc7cd8b75222857afc936f0638c5683d6
+//
+// Examples:
+//
+//     "de" -> 2
+//     "german" -> 2
+func (ctx *Context) Whisper_full_lang_id() int {
+       return int(C.whisper_full_lang_id((*C.struct_whisper_context)(ctx)))
+}
+
 // Number of generated text segments.
 // A segment can be a few words, a sentence, or even a paragraph.
 func (ctx *Context) Whisper_full_n_segments() int {
@@ -451,3 +463,7 @@ func (t TokenData) T0() int64 {
 func (t TokenData) T1() int64 {
        return int64(t.t1)
 }
+
+func (t TokenData) Id() Token {
+       return Token(t.id)
+}