jacob-mathison/query-orchestration
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

Merged in feature/textExtractionsPart2 (pull request #193)

Browse Source 
Continue finishing the parts of the text extraction plan

* add missing fields
This commit is contained in:
Jay Brown
2025-12-02 19:13:08 +00:00
parent c45e1dd427
commit 3bdc27f4de
21 changed files with 2392 additions and 215 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mocks/queryapi/mock_ClientInterface.go
+147
View File
@@ -1964,6 +1964,80 @@ func (_c *MockClientInterface_GetDocumentsByLabel_Call) RunAndReturn(run func(co
return _c
}
// GetFieldExtractionByVersion provides a mock function with given fields: ctx, params, reqEditors
func (_m *MockClientInterface) GetFieldExtractionByVersion(ctx context.Context, params *queryapi.GetFieldExtractionByVersionParams, reqEditors ...queryapi.RequestEditorFn) (*http.Response, error) {
_va := make([]interface{}, len(reqEditors))
for _i := range reqEditors {
_va[_i] = reqEditors[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, params)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
if len(ret) == 0 {
panic("no return value specified for GetFieldExtractionByVersion")
}
var r0 *http.Response
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *queryapi.GetFieldExtractionByVersionParams, ...queryapi.RequestEditorFn) (*http.Response, error)); ok {
return rf(ctx, params, reqEditors...)
}
if rf, ok := ret.Get(0).(func(context.Context, *queryapi.GetFieldExtractionByVersionParams, ...queryapi.RequestEditorFn) *http.Response); ok {
r0 = rf(ctx, params, reqEditors...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*http.Response)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *queryapi.GetFieldExtractionByVersionParams, ...queryapi.RequestEditorFn) error); ok {
r1 = rf(ctx, params, reqEditors...)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockClientInterface_GetFieldExtractionByVersion_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetFieldExtractionByVersion'
type MockClientInterface_GetFieldExtractionByVersion_Call struct {
*mock.Call
}
// GetFieldExtractionByVersion is a helper method to define mock.On call
// - ctx context.Context
// - params *queryapi.GetFieldExtractionByVersionParams
// - reqEditors ...queryapi.RequestEditorFn
func (_e *MockClientInterface_Expecter) GetFieldExtractionByVersion(ctx interface{}, params interface{}, reqEditors ...interface{}) *MockClientInterface_GetFieldExtractionByVersion_Call {
return &MockClientInterface_GetFieldExtractionByVersion_Call{Call: _e.mock.On("GetFieldExtractionByVersion",
append([]interface{}{ctx, params}, reqEditors...)...)}
}
func (_c *MockClientInterface_GetFieldExtractionByVersion_Call) Run(run func(ctx context.Context, params *queryapi.GetFieldExtractionByVersionParams, reqEditors ...queryapi.RequestEditorFn)) *MockClientInterface_GetFieldExtractionByVersion_Call {
_c.Call.Run(func(args mock.Arguments) {
variadicArgs := make([]queryapi.RequestEditorFn, len(args)-2)
for i, a := range args[2:] {
if a != nil {
variadicArgs[i] = a.(queryapi.RequestEditorFn)
}
}
run(args[0].(context.Context), args[1].(*queryapi.GetFieldExtractionByVersionParams), variadicArgs...)
})
return _c
}
func (_c *MockClientInterface_GetFieldExtractionByVersion_Call) Return(_a0 *http.Response, _a1 error) *MockClientInterface_GetFieldExtractionByVersion_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockClientInterface_GetFieldExtractionByVersion_Call) RunAndReturn(run func(context.Context, *queryapi.GetFieldExtractionByVersionParams, ...queryapi.RequestEditorFn) (*http.Response, error)) *MockClientInterface_GetFieldExtractionByVersion_Call {
_c.Call.Return(run)
return _c
}
// GetFieldExtractionHistory provides a mock function with given fields: ctx, params, reqEditors
func (_m *MockClientInterface) GetFieldExtractionHistory(ctx context.Context, params *queryapi.GetFieldExtractionHistoryParams, reqEditors ...queryapi.RequestEditorFn) (*http.Response, error) {
_va := make([]interface{}, len(reqEditors))
@@ -2555,6 +2629,79 @@ func (_c *MockClientInterface_ListClientFolders_Call) RunAndReturn(run func(cont
return _c
}
// ListClients provides a mock function with given fields: ctx, reqEditors
func (_m *MockClientInterface) ListClients(ctx context.Context, reqEditors ...queryapi.RequestEditorFn) (*http.Response, error) {
_va := make([]interface{}, len(reqEditors))
for _i := range reqEditors {
_va[_i] = reqEditors[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
if len(ret) == 0 {
panic("no return value specified for ListClients")
}
var r0 *http.Response
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, ...queryapi.RequestEditorFn) (*http.Response, error)); ok {
return rf(ctx, reqEditors...)
}
if rf, ok := ret.Get(0).(func(context.Context, ...queryapi.RequestEditorFn) *http.Response); ok {
r0 = rf(ctx, reqEditors...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*http.Response)
}
}
if rf, ok := ret.Get(1).(func(context.Context, ...queryapi.RequestEditorFn) error); ok {
r1 = rf(ctx, reqEditors...)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockClientInterface_ListClients_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListClients'
type MockClientInterface_ListClients_Call struct {
*mock.Call
}
// ListClients is a helper method to define mock.On call
// - ctx context.Context
// - reqEditors ...queryapi.RequestEditorFn
func (_e *MockClientInterface_Expecter) ListClients(ctx interface{}, reqEditors ...interface{}) *MockClientInterface_ListClients_Call {
return &MockClientInterface_ListClients_Call{Call: _e.mock.On("ListClients",
append([]interface{}{ctx}, reqEditors...)...)}
}
func (_c *MockClientInterface_ListClients_Call) Run(run func(ctx context.Context, reqEditors ...queryapi.RequestEditorFn)) *MockClientInterface_ListClients_Call {
_c.Call.Run(func(args mock.Arguments) {
variadicArgs := make([]queryapi.RequestEditorFn, len(args)-1)
for i, a := range args[1:] {
if a != nil {
variadicArgs[i] = a.(queryapi.RequestEditorFn)
}
}
run(args[0].(context.Context), variadicArgs...)
})
return _c
}
func (_c *MockClientInterface_ListClients_Call) Return(_a0 *http.Response, _a1 error) *MockClientInterface_ListClients_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockClientInterface_ListClients_Call) RunAndReturn(run func(context.Context, ...queryapi.RequestEditorFn) (*http.Response, error)) *MockClientInterface_ListClients_Call {
_c.Call.Return(run)
return _c
}
// ListDocumentBatches provides a mock function with given fields: ctx, id, params, reqEditors
func (_m *MockClientInterface) ListDocumentBatches(ctx context.Context, id queryapi.ClientID, params *queryapi.ListDocumentBatchesParams, reqEditors ...queryapi.RequestEditorFn) (*http.Response, error) {
_va := make([]interface{}, len(reqEditors))
mocks/queryapi/mock_ClientWithResponsesInterface.go
+147
View File
@@ -1962,6 +1962,80 @@ func (_c *MockClientWithResponsesInterface_GetDocumentsByLabelWithResponse_Call)
return _c
}
// GetFieldExtractionByVersionWithResponse provides a mock function with given fields: ctx, params, reqEditors
func (_m *MockClientWithResponsesInterface) GetFieldExtractionByVersionWithResponse(ctx context.Context, params *queryapi.GetFieldExtractionByVersionParams, reqEditors ...queryapi.RequestEditorFn) (*queryapi.GetFieldExtractionByVersionResponse, error) {
_va := make([]interface{}, len(reqEditors))
for _i := range reqEditors {
_va[_i] = reqEditors[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, params)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
if len(ret) == 0 {
panic("no return value specified for GetFieldExtractionByVersionWithResponse")
}
var r0 *queryapi.GetFieldExtractionByVersionResponse
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, *queryapi.GetFieldExtractionByVersionParams, ...queryapi.RequestEditorFn) (*queryapi.GetFieldExtractionByVersionResponse, error)); ok {
return rf(ctx, params, reqEditors...)
}
if rf, ok := ret.Get(0).(func(context.Context, *queryapi.GetFieldExtractionByVersionParams, ...queryapi.RequestEditorFn) *queryapi.GetFieldExtractionByVersionResponse); ok {
r0 = rf(ctx, params, reqEditors...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*queryapi.GetFieldExtractionByVersionResponse)
}
}
if rf, ok := ret.Get(1).(func(context.Context, *queryapi.GetFieldExtractionByVersionParams, ...queryapi.RequestEditorFn) error); ok {
r1 = rf(ctx, params, reqEditors...)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetFieldExtractionByVersionWithResponse'
type MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call struct {
*mock.Call
}
// GetFieldExtractionByVersionWithResponse is a helper method to define mock.On call
// - ctx context.Context
// - params *queryapi.GetFieldExtractionByVersionParams
// - reqEditors ...queryapi.RequestEditorFn
func (_e *MockClientWithResponsesInterface_Expecter) GetFieldExtractionByVersionWithResponse(ctx interface{}, params interface{}, reqEditors ...interface{}) *MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call {
return &MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call{Call: _e.mock.On("GetFieldExtractionByVersionWithResponse",
append([]interface{}{ctx, params}, reqEditors...)...)}
}
func (_c *MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call) Run(run func(ctx context.Context, params *queryapi.GetFieldExtractionByVersionParams, reqEditors ...queryapi.RequestEditorFn)) *MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call {
_c.Call.Run(func(args mock.Arguments) {
variadicArgs := make([]queryapi.RequestEditorFn, len(args)-2)
for i, a := range args[2:] {
if a != nil {
variadicArgs[i] = a.(queryapi.RequestEditorFn)
}
}
run(args[0].(context.Context), args[1].(*queryapi.GetFieldExtractionByVersionParams), variadicArgs...)
})
return _c
}
func (_c *MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call) Return(_a0 *queryapi.GetFieldExtractionByVersionResponse, _a1 error) *MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call) RunAndReturn(run func(context.Context, *queryapi.GetFieldExtractionByVersionParams, ...queryapi.RequestEditorFn) (*queryapi.GetFieldExtractionByVersionResponse, error)) *MockClientWithResponsesInterface_GetFieldExtractionByVersionWithResponse_Call {
_c.Call.Return(run)
return _c
}
// GetFieldExtractionHistoryWithResponse provides a mock function with given fields: ctx, params, reqEditors
func (_m *MockClientWithResponsesInterface) GetFieldExtractionHistoryWithResponse(ctx context.Context, params *queryapi.GetFieldExtractionHistoryParams, reqEditors ...queryapi.RequestEditorFn) (*queryapi.GetFieldExtractionHistoryResponse, error) {
_va := make([]interface{}, len(reqEditors))
@@ -2553,6 +2627,79 @@ func (_c *MockClientWithResponsesInterface_ListClientFoldersWithResponse_Call) R
return _c
}
// ListClientsWithResponse provides a mock function with given fields: ctx, reqEditors
func (_m *MockClientWithResponsesInterface) ListClientsWithResponse(ctx context.Context, reqEditors ...queryapi.RequestEditorFn) (*queryapi.ListClientsResponse, error) {
_va := make([]interface{}, len(reqEditors))
for _i := range reqEditors {
_va[_i] = reqEditors[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
if len(ret) == 0 {
panic("no return value specified for ListClientsWithResponse")
}
var r0 *queryapi.ListClientsResponse
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, ...queryapi.RequestEditorFn) (*queryapi.ListClientsResponse, error)); ok {
return rf(ctx, reqEditors...)
}
if rf, ok := ret.Get(0).(func(context.Context, ...queryapi.RequestEditorFn) *queryapi.ListClientsResponse); ok {
r0 = rf(ctx, reqEditors...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*queryapi.ListClientsResponse)
}
}
if rf, ok := ret.Get(1).(func(context.Context, ...queryapi.RequestEditorFn) error); ok {
r1 = rf(ctx, reqEditors...)
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// MockClientWithResponsesInterface_ListClientsWithResponse_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListClientsWithResponse'
type MockClientWithResponsesInterface_ListClientsWithResponse_Call struct {
*mock.Call
}
// ListClientsWithResponse is a helper method to define mock.On call
// - ctx context.Context
// - reqEditors ...queryapi.RequestEditorFn
func (_e *MockClientWithResponsesInterface_Expecter) ListClientsWithResponse(ctx interface{}, reqEditors ...interface{}) *MockClientWithResponsesInterface_ListClientsWithResponse_Call {
return &MockClientWithResponsesInterface_ListClientsWithResponse_Call{Call: _e.mock.On("ListClientsWithResponse",
append([]interface{}{ctx}, reqEditors...)...)}
}
func (_c *MockClientWithResponsesInterface_ListClientsWithResponse_Call) Run(run func(ctx context.Context, reqEditors ...queryapi.RequestEditorFn)) *MockClientWithResponsesInterface_ListClientsWithResponse_Call {
_c.Call.Run(func(args mock.Arguments) {
variadicArgs := make([]queryapi.RequestEditorFn, len(args)-1)
for i, a := range args[1:] {
if a != nil {
variadicArgs[i] = a.(queryapi.RequestEditorFn)
}
}
run(args[0].(context.Context), variadicArgs...)
})
return _c
}
func (_c *MockClientWithResponsesInterface_ListClientsWithResponse_Call) Return(_a0 *queryapi.ListClientsResponse, _a1 error) *MockClientWithResponsesInterface_ListClientsWithResponse_Call {
_c.Call.Return(_a0, _a1)
return _c
}
func (_c *MockClientWithResponsesInterface_ListClientsWithResponse_Call) RunAndReturn(run func(context.Context, ...queryapi.RequestEditorFn) (*queryapi.ListClientsResponse, error)) *MockClientWithResponsesInterface_ListClientsWithResponse_Call {
_c.Call.Return(run)
return _c
}
// ListDocumentBatchesWithResponse provides a mock function with given fields: ctx, id, params, reqEditors
func (_m *MockClientWithResponsesInterface) ListDocumentBatchesWithResponse(ctx context.Context, id queryapi.ClientID, params *queryapi.ListDocumentBatchesParams, reqEditors ...queryapi.RequestEditorFn) (*queryapi.ListDocumentBatchesResponse, error) {
_va := make([]interface{}, len(reqEditors))