Fix embeddings
Some checks failed
ci/woodpecker/push/author-tests Pipeline failed

This commit is contained in:
Ryan Voots 2023-11-22 18:19:58 -05:00
parent 76747c731f
commit 15963fae0e
2 changed files with 10 additions and 2 deletions

View file

@ -68,7 +68,7 @@ class OpenAIAsync::Types::Requests::Completion :does(OpenAIAsync::Types::Request
} }
class OpenAIAsync::Types::Requests::Embedding :does(OpenAIAsync::Types::Requests::Base) :Struct { class OpenAIAsync::Types::Requests::Embedding :does(OpenAIAsync::Types::Requests::Base) :Struct {
method _endpoint() {...} method _endpoint() {"/embeddings"}
field $input :JSONStr; field $input :JSONStr;
field $model :JSONStr; field $model :JSONStr;
field $encoding_format :JSONStr = undef; field $encoding_format :JSONStr = undef;

View file

@ -67,7 +67,7 @@ class OpenAIAsync::Types::Results::ChatCompletionChunk :does(OpenAIAsync::Types:
class OpenAIAsync::Types::Results::Usage :does(OpenAIAsync::Types::Base) :Struct { class OpenAIAsync::Types::Results::Usage :does(OpenAIAsync::Types::Base) :Struct {
field $total_tokens :JSONNum; field $total_tokens :JSONNum;
field $prompt_tokens :JSONNum; field $prompt_tokens :JSONNum;
field $completion_tokens :JSONNum; # look at chat completions, is this the same field $completion_tokens :JSONNum = undef; # look at chat completions, is this the same
} }
class OpenAIAsync::Types::Results::LogProbs :does(OpenAIAsync::Types::Base) :Struct { class OpenAIAsync::Types::Results::LogProbs :does(OpenAIAsync::Types::Base) :Struct {
@ -95,7 +95,15 @@ class OpenAIAsync::Types::Results::Completion :does(OpenAIAsync::Types::Base) :S
field $object :JSONStr; field $object :JSONStr;
} }
class OpenAIAsync::Types::Results::Embedding :does(OpenAIAsync::Types::Base) :Struct { class OpenAIAsync::Types::Results::Embedding :does(OpenAIAsync::Types::Base) :Struct {
field $object :JSONStr;
field $model :JSONStr;
field $usage :MarshalTo(OpenAIAsync::Types::Results::Usage);
field $data :MarshalTo([OpenAIAsync::Types::Results::EmbeddingData]);
}
class OpenAIAsync::Types::Results::EmbeddingData :does(OpenAIAsync::Types::Base) :Struct {
field $index :JSONNum; field $index :JSONNum;
field $embedding :JSONList(JSONNum); field $embedding :JSONList(JSONNum);
field $object :JSONStr; field $object :JSONStr;