corvic-engine 0.3.0rc58__cp38-abi3-win_amd64.whl → 0.3.0rc60__cp38-abi3-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (42) hide show
  1. corvic/engine/_native.pyd +0 -0
  2. corvic/model/__init__.py +0 -3
  3. corvic/model/_base_model.py +9 -6
  4. corvic/model/_completion_model.py +10 -4
  5. corvic/model/_proto_orm_convert.py +4 -91
  6. corvic/orm/__init__.py +5 -75
  7. corvic/orm/base.py +1 -1
  8. {corvic_engine-0.3.0rc58.dist-info → corvic_engine-0.3.0rc60.dist-info}/METADATA +1 -1
  9. {corvic_engine-0.3.0rc58.dist-info → corvic_engine-0.3.0rc60.dist-info}/RECORD +23 -42
  10. corvic_generated/feature/v2/feature_view_pb2.py +39 -47
  11. corvic_generated/feature/v2/feature_view_pb2.pyi +0 -14
  12. corvic_generated/feature/v2/space_pb2.py +77 -87
  13. corvic_generated/ingest/v2/pipeline_pb2.py +33 -27
  14. corvic_generated/ingest/v2/quick_mode_pb2.py +18 -13
  15. corvic_generated/ingest/v2/quick_mode_pb2.pyi +1 -0
  16. corvic_generated/ingest/v2/resource_pb2.py +64 -58
  17. corvic_generated/ingest/v2/room_pb2.py +32 -32
  18. corvic_generated/ingest/v2/source_pb2.py +43 -43
  19. corvic_generated/ingest/v2/table_pb2.py +23 -23
  20. corvic_generated/model/v1alpha/models_pb2.py +2 -2
  21. corvic_generated/model/v1alpha/models_pb2.pyi +4 -4
  22. corvic/model/_agent.py +0 -154
  23. corvic_generated/chat/__init__.py +0 -0
  24. corvic_generated/chat/v1/__init__.py +0 -0
  25. corvic_generated/chat/v1/service_pb2.py +0 -65
  26. corvic_generated/chat/v1/service_pb2.pyi +0 -132
  27. corvic_generated/chat/v1/service_pb2_grpc.py +0 -198
  28. corvic_generated/chat/v1/service_pb2_grpc.pyi +0 -97
  29. corvic_generated/completion/__init__.py +0 -0
  30. corvic_generated/completion/v1/__init__.py +0 -0
  31. corvic_generated/completion/v1/service_pb2.py +0 -69
  32. corvic_generated/completion/v1/service_pb2.pyi +0 -201
  33. corvic_generated/completion/v1/service_pb2_grpc.py +0 -198
  34. corvic_generated/completion/v1/service_pb2_grpc.pyi +0 -97
  35. corvic_generated/query/__init__.py +0 -0
  36. corvic_generated/query/v1/__init__.py +0 -0
  37. corvic_generated/query/v1/query_pb2.py +0 -36
  38. corvic_generated/query/v1/query_pb2.pyi +0 -63
  39. corvic_generated/query/v1/query_pb2_grpc.py +0 -66
  40. corvic_generated/query/v1/query_pb2_grpc.pyi +0 -41
  41. {corvic_engine-0.3.0rc58.dist-info → corvic_engine-0.3.0rc60.dist-info}/WHEEL +0 -0
  42. {corvic_engine-0.3.0rc58.dist-info → corvic_engine-0.3.0rc60.dist-info}/licenses/LICENSE +0 -0
@@ -1,97 +0,0 @@
1
- """
2
- @generated by mypy-protobuf. Do not edit manually!
3
- isort:skip_file
4
- """
5
- import abc
6
- import collections.abc
7
- import corvic_generated.completion.v1.service_pb2
8
- import grpc
9
- import grpc.aio
10
- import typing
11
-
12
- _T = typing.TypeVar('_T')
13
-
14
- class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta):
15
- ...
16
-
17
- class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext): # type: ignore
18
- ...
19
-
20
- class CompletionServiceStub:
21
- def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
22
- Complete: grpc.UnaryUnaryMultiCallable[
23
- corvic_generated.completion.v1.service_pb2.CompleteRequest,
24
- corvic_generated.completion.v1.service_pb2.CompleteResponse,
25
- ]
26
- CompleteRag: grpc.UnaryUnaryMultiCallable[
27
- corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
28
- corvic_generated.completion.v1.service_pb2.CompleteRagResponse,
29
- ]
30
- GetSha: grpc.UnaryUnaryMultiCallable[
31
- corvic_generated.completion.v1.service_pb2.GetShaRequest,
32
- corvic_generated.completion.v1.service_pb2.GetShaResponse,
33
- ]
34
- GetModels: grpc.UnaryUnaryMultiCallable[
35
- corvic_generated.completion.v1.service_pb2.GetModelsRequest,
36
- corvic_generated.completion.v1.service_pb2.GetModelsResponse,
37
- ]
38
- GetDefault: grpc.UnaryUnaryMultiCallable[
39
- corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
40
- corvic_generated.completion.v1.service_pb2.GetDefaultResponse,
41
- ]
42
-
43
- class CompletionServiceAsyncStub:
44
- Complete: grpc.aio.UnaryUnaryMultiCallable[
45
- corvic_generated.completion.v1.service_pb2.CompleteRequest,
46
- corvic_generated.completion.v1.service_pb2.CompleteResponse,
47
- ]
48
- CompleteRag: grpc.aio.UnaryUnaryMultiCallable[
49
- corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
50
- corvic_generated.completion.v1.service_pb2.CompleteRagResponse,
51
- ]
52
- GetSha: grpc.aio.UnaryUnaryMultiCallable[
53
- corvic_generated.completion.v1.service_pb2.GetShaRequest,
54
- corvic_generated.completion.v1.service_pb2.GetShaResponse,
55
- ]
56
- GetModels: grpc.aio.UnaryUnaryMultiCallable[
57
- corvic_generated.completion.v1.service_pb2.GetModelsRequest,
58
- corvic_generated.completion.v1.service_pb2.GetModelsResponse,
59
- ]
60
- GetDefault: grpc.aio.UnaryUnaryMultiCallable[
61
- corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
62
- corvic_generated.completion.v1.service_pb2.GetDefaultResponse,
63
- ]
64
-
65
- class CompletionServiceServicer(metaclass=abc.ABCMeta):
66
- @abc.abstractmethod
67
- def Complete(
68
- self,
69
- request: corvic_generated.completion.v1.service_pb2.CompleteRequest,
70
- context: _ServicerContext,
71
- ) -> typing.Union[corvic_generated.completion.v1.service_pb2.CompleteResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.CompleteResponse]]: ...
72
- @abc.abstractmethod
73
- def CompleteRag(
74
- self,
75
- request: corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
76
- context: _ServicerContext,
77
- ) -> typing.Union[corvic_generated.completion.v1.service_pb2.CompleteRagResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.CompleteRagResponse]]: ...
78
- @abc.abstractmethod
79
- def GetSha(
80
- self,
81
- request: corvic_generated.completion.v1.service_pb2.GetShaRequest,
82
- context: _ServicerContext,
83
- ) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetShaResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetShaResponse]]: ...
84
- @abc.abstractmethod
85
- def GetModels(
86
- self,
87
- request: corvic_generated.completion.v1.service_pb2.GetModelsRequest,
88
- context: _ServicerContext,
89
- ) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetModelsResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetModelsResponse]]: ...
90
- @abc.abstractmethod
91
- def GetDefault(
92
- self,
93
- request: corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
94
- context: _ServicerContext,
95
- ) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetDefaultResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetDefaultResponse]]: ...
96
-
97
- def add_CompletionServiceServicer_to_server(servicer: CompletionServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...
File without changes
File without changes
@@ -1,36 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: corvic/query/v1/query.proto
4
- # Protobuf Python Version: 4.25.3
5
- """Generated protocol buffer code."""
6
- from google.protobuf import descriptor as _descriptor
7
- from google.protobuf import descriptor_pool as _descriptor_pool
8
- from google.protobuf import symbol_database as _symbol_database
9
- from google.protobuf.internal import builder as _builder
10
- # @@protoc_insertion_point(imports)
11
-
12
- _sym_db = _symbol_database.Default()
13
-
14
-
15
-
16
-
17
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x63orvic/query/v1/query.proto\x12\x0f\x63orvic.query.v1\"P\n\nTokenUsage\x12\x1f\n\x0btoken_input\x18\x01 \x01(\x01R\ntokenInput\x12!\n\x0ctoken_output\x18\x02 \x01(\x01R\x0btokenOutput\"\xd4\x01\n\x0cQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query\x12\x39\n\nrag_method\x18\x02 \x01(\x0e\x32\x1a.corvic.query.v1.RagMethodR\tragMethod\x12\x38\n\x18model_system_instruction\x18\x03 \x01(\tR\x16modelSystemInstruction\x12\x39\n\nmodel_type\x18\x04 \x01(\x0e\x32\x1a.corvic.query.v1.ModelTypeR\tmodelType\"\x83\x01\n\rQueryResponse\x12<\n\x0btoken_usage\x18\x01 \x01(\x0b\x32\x1b.corvic.query.v1.TokenUsageR\ntokenUsage\x12\x1a\n\x08response\x18\x02 \x01(\tR\x08response\x12\x18\n\x07sources\x18\x03 \x03(\tR\x07sources*T\n\tRagMethod\x12\x1a\n\x16RAG_METHOD_UNSPECIFIED\x10\x00\x12\x14\n\x10RAG_METHOD_NAIVE\x10\x01\x12\x15\n\x11RAG_METHOD_CORVIC\x10\x02*\xed\x01\n\tModelType\x12\x1a\n\x16MODEL_TYPE_UNSPECIFIED\x10\x00\x12\x1d\n\x19MODEL_TYPE_GEMINI_1_5_PRO\x10\x01\x12%\n!MODEL_TYPE_GEMINI_1_5_PRO_PREVIEW\x10\x02\x12\x1f\n\x1bMODEL_TYPE_GEMINI_1_5_FLASH\x10\x03\x12\'\n#MODEL_TYPE_GEMINI_1_5_FLASH_PREVIEW\x10\x04\x12\x16\n\x12MODEL_TYPE_GPT_4_O\x10\x05\x12\x1c\n\x18MODEL_TYPE_GPT_3_5_TURBO\x10\x06\x32Z\n\x0cQueryService\x12J\n\x05Query\x12\x1d.corvic.query.v1.QueryRequest\x1a\x1e.corvic.query.v1.QueryResponse\"\x00\x30\x01\x62\x06proto3')
18
-
19
- _globals = globals()
20
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
21
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.query.v1.query_pb2', _globals)
22
- if _descriptor._USE_C_DESCRIPTORS == False:
23
- DESCRIPTOR._options = None
24
- _globals['_RAGMETHOD']._serialized_start=479
25
- _globals['_RAGMETHOD']._serialized_end=563
26
- _globals['_MODELTYPE']._serialized_start=566
27
- _globals['_MODELTYPE']._serialized_end=803
28
- _globals['_TOKENUSAGE']._serialized_start=48
29
- _globals['_TOKENUSAGE']._serialized_end=128
30
- _globals['_QUERYREQUEST']._serialized_start=131
31
- _globals['_QUERYREQUEST']._serialized_end=343
32
- _globals['_QUERYRESPONSE']._serialized_start=346
33
- _globals['_QUERYRESPONSE']._serialized_end=477
34
- _globals['_QUERYSERVICE']._serialized_start=805
35
- _globals['_QUERYSERVICE']._serialized_end=895
36
- # @@protoc_insertion_point(module_scope)
@@ -1,63 +0,0 @@
1
- from google.protobuf.internal import containers as _containers
2
- from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
3
- from google.protobuf import descriptor as _descriptor
4
- from google.protobuf import message as _message
5
- from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
6
-
7
- DESCRIPTOR: _descriptor.FileDescriptor
8
-
9
- class RagMethod(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
10
- __slots__ = ()
11
- RAG_METHOD_UNSPECIFIED: _ClassVar[RagMethod]
12
- RAG_METHOD_NAIVE: _ClassVar[RagMethod]
13
- RAG_METHOD_CORVIC: _ClassVar[RagMethod]
14
-
15
- class ModelType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
16
- __slots__ = ()
17
- MODEL_TYPE_UNSPECIFIED: _ClassVar[ModelType]
18
- MODEL_TYPE_GEMINI_1_5_PRO: _ClassVar[ModelType]
19
- MODEL_TYPE_GEMINI_1_5_PRO_PREVIEW: _ClassVar[ModelType]
20
- MODEL_TYPE_GEMINI_1_5_FLASH: _ClassVar[ModelType]
21
- MODEL_TYPE_GEMINI_1_5_FLASH_PREVIEW: _ClassVar[ModelType]
22
- MODEL_TYPE_GPT_4_O: _ClassVar[ModelType]
23
- MODEL_TYPE_GPT_3_5_TURBO: _ClassVar[ModelType]
24
- RAG_METHOD_UNSPECIFIED: RagMethod
25
- RAG_METHOD_NAIVE: RagMethod
26
- RAG_METHOD_CORVIC: RagMethod
27
- MODEL_TYPE_UNSPECIFIED: ModelType
28
- MODEL_TYPE_GEMINI_1_5_PRO: ModelType
29
- MODEL_TYPE_GEMINI_1_5_PRO_PREVIEW: ModelType
30
- MODEL_TYPE_GEMINI_1_5_FLASH: ModelType
31
- MODEL_TYPE_GEMINI_1_5_FLASH_PREVIEW: ModelType
32
- MODEL_TYPE_GPT_4_O: ModelType
33
- MODEL_TYPE_GPT_3_5_TURBO: ModelType
34
-
35
- class TokenUsage(_message.Message):
36
- __slots__ = ("token_input", "token_output")
37
- TOKEN_INPUT_FIELD_NUMBER: _ClassVar[int]
38
- TOKEN_OUTPUT_FIELD_NUMBER: _ClassVar[int]
39
- token_input: float
40
- token_output: float
41
- def __init__(self, token_input: _Optional[float] = ..., token_output: _Optional[float] = ...) -> None: ...
42
-
43
- class QueryRequest(_message.Message):
44
- __slots__ = ("query", "rag_method", "model_system_instruction", "model_type")
45
- QUERY_FIELD_NUMBER: _ClassVar[int]
46
- RAG_METHOD_FIELD_NUMBER: _ClassVar[int]
47
- MODEL_SYSTEM_INSTRUCTION_FIELD_NUMBER: _ClassVar[int]
48
- MODEL_TYPE_FIELD_NUMBER: _ClassVar[int]
49
- query: str
50
- rag_method: RagMethod
51
- model_system_instruction: str
52
- model_type: ModelType
53
- def __init__(self, query: _Optional[str] = ..., rag_method: _Optional[_Union[RagMethod, str]] = ..., model_system_instruction: _Optional[str] = ..., model_type: _Optional[_Union[ModelType, str]] = ...) -> None: ...
54
-
55
- class QueryResponse(_message.Message):
56
- __slots__ = ("token_usage", "response", "sources")
57
- TOKEN_USAGE_FIELD_NUMBER: _ClassVar[int]
58
- RESPONSE_FIELD_NUMBER: _ClassVar[int]
59
- SOURCES_FIELD_NUMBER: _ClassVar[int]
60
- token_usage: TokenUsage
61
- response: str
62
- sources: _containers.RepeatedScalarFieldContainer[str]
63
- def __init__(self, token_usage: _Optional[_Union[TokenUsage, _Mapping]] = ..., response: _Optional[str] = ..., sources: _Optional[_Iterable[str]] = ...) -> None: ...
@@ -1,66 +0,0 @@
1
- # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
- """Client and server classes corresponding to protobuf-defined services."""
3
- import grpc
4
-
5
- from corvic_generated.query.v1 import query_pb2 as corvic_dot_query_dot_v1_dot_query__pb2
6
-
7
-
8
- class QueryServiceStub(object):
9
- """Missing associated documentation comment in .proto file."""
10
-
11
- def __init__(self, channel):
12
- """Constructor.
13
-
14
- Args:
15
- channel: A grpc.Channel.
16
- """
17
- self.Query = channel.unary_stream(
18
- '/corvic.query.v1.QueryService/Query',
19
- request_serializer=corvic_dot_query_dot_v1_dot_query__pb2.QueryRequest.SerializeToString,
20
- response_deserializer=corvic_dot_query_dot_v1_dot_query__pb2.QueryResponse.FromString,
21
- )
22
-
23
-
24
- class QueryServiceServicer(object):
25
- """Missing associated documentation comment in .proto file."""
26
-
27
- def Query(self, request, context):
28
- """Missing associated documentation comment in .proto file."""
29
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
30
- context.set_details('Method not implemented!')
31
- raise NotImplementedError('Method not implemented!')
32
-
33
-
34
- def add_QueryServiceServicer_to_server(servicer, server):
35
- rpc_method_handlers = {
36
- 'Query': grpc.unary_stream_rpc_method_handler(
37
- servicer.Query,
38
- request_deserializer=corvic_dot_query_dot_v1_dot_query__pb2.QueryRequest.FromString,
39
- response_serializer=corvic_dot_query_dot_v1_dot_query__pb2.QueryResponse.SerializeToString,
40
- ),
41
- }
42
- generic_handler = grpc.method_handlers_generic_handler(
43
- 'corvic.query.v1.QueryService', rpc_method_handlers)
44
- server.add_generic_rpc_handlers((generic_handler,))
45
-
46
-
47
- # This class is part of an EXPERIMENTAL API.
48
- class QueryService(object):
49
- """Missing associated documentation comment in .proto file."""
50
-
51
- @staticmethod
52
- def Query(request,
53
- target,
54
- options=(),
55
- channel_credentials=None,
56
- call_credentials=None,
57
- insecure=False,
58
- compression=None,
59
- wait_for_ready=None,
60
- timeout=None,
61
- metadata=None):
62
- return grpc.experimental.unary_stream(request, target, '/corvic.query.v1.QueryService/Query',
63
- corvic_dot_query_dot_v1_dot_query__pb2.QueryRequest.SerializeToString,
64
- corvic_dot_query_dot_v1_dot_query__pb2.QueryResponse.FromString,
65
- options, channel_credentials,
66
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
@@ -1,41 +0,0 @@
1
- """
2
- @generated by mypy-protobuf. Do not edit manually!
3
- isort:skip_file
4
- """
5
- import abc
6
- import collections.abc
7
- import corvic_generated.query.v1.query_pb2
8
- import grpc
9
- import grpc.aio
10
- import typing
11
-
12
- _T = typing.TypeVar('_T')
13
-
14
- class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta):
15
- ...
16
-
17
- class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext): # type: ignore
18
- ...
19
-
20
- class QueryServiceStub:
21
- def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
22
- Query: grpc.UnaryStreamMultiCallable[
23
- corvic_generated.query.v1.query_pb2.QueryRequest,
24
- corvic_generated.query.v1.query_pb2.QueryResponse,
25
- ]
26
-
27
- class QueryServiceAsyncStub:
28
- Query: grpc.aio.UnaryStreamMultiCallable[
29
- corvic_generated.query.v1.query_pb2.QueryRequest,
30
- corvic_generated.query.v1.query_pb2.QueryResponse,
31
- ]
32
-
33
- class QueryServiceServicer(metaclass=abc.ABCMeta):
34
- @abc.abstractmethod
35
- def Query(
36
- self,
37
- request: corvic_generated.query.v1.query_pb2.QueryRequest,
38
- context: _ServicerContext,
39
- ) -> typing.Union[collections.abc.Iterator[corvic_generated.query.v1.query_pb2.QueryResponse], collections.abc.AsyncIterator[corvic_generated.query.v1.query_pb2.QueryResponse]]: ...
40
-
41
- def add_QueryServiceServicer_to_server(servicer: QueryServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...