weave-python 0.38.0__tar.gz → 0.39.0__tar.gz

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 (136) hide show
  1. {weave_python-0.38.0 → weave_python-0.39.0}/PKG-INFO +1 -1
  2. weave_python-0.39.0/weave/weaveapi/chat/v1/chat_pb2.py +90 -0
  3. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/chat/v1/chat_pb2.pyi +71 -0
  4. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/chat.py +81 -3
  5. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/models.py +1 -0
  6. weave_python-0.38.0/weave/weaveapi/chat/v1/chat_pb2.py +0 -76
  7. {weave_python-0.38.0 → weave_python-0.39.0}/.github/runs-on.yml +0 -0
  8. {weave_python-0.38.0 → weave_python-0.39.0}/.github/workflows/format-lint.yaml +0 -0
  9. {weave_python-0.38.0 → weave_python-0.39.0}/.github/workflows/generate.yaml +0 -0
  10. {weave_python-0.38.0 → weave_python-0.39.0}/.github/workflows/release.yaml +0 -0
  11. {weave_python-0.38.0 → weave_python-0.39.0}/.gitignore +0 -0
  12. {weave_python-0.38.0 → weave_python-0.39.0}/.python-version +0 -0
  13. {weave_python-0.38.0 → weave_python-0.39.0}/LICENSE +0 -0
  14. {weave_python-0.38.0 → weave_python-0.39.0}/README.md +0 -0
  15. {weave_python-0.38.0 → weave_python-0.39.0}/Taskfile.yaml +0 -0
  16. {weave_python-0.38.0 → weave_python-0.39.0}/buf.gen.yaml +0 -0
  17. {weave_python-0.38.0 → weave_python-0.39.0}/pyproject.toml +0 -0
  18. {weave_python-0.38.0 → weave_python-0.39.0}/renovate.json +0 -0
  19. {weave_python-0.38.0 → weave_python-0.39.0}/tools/sqlcgen/README.md +0 -0
  20. {weave_python-0.38.0 → weave_python-0.39.0}/tools/sqlcgen/sqlcgen.py +0 -0
  21. {weave_python-0.38.0 → weave_python-0.39.0}/tools/sqlcgen/test_sqlcgen.py +0 -0
  22. {weave_python-0.38.0 → weave_python-0.39.0}/uv.lock +0 -0
  23. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/event_pb2.py +0 -0
  24. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/event_pb2.pyi +0 -0
  25. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/event_pb2_grpc.py +0 -0
  26. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/event_pb2_grpc.pyi +0 -0
  27. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/message_pb2.py +0 -0
  28. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/message_pb2.pyi +0 -0
  29. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/message_pb2_grpc.py +0 -0
  30. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/message_pb2_grpc.pyi +0 -0
  31. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/process_pb2.py +0 -0
  32. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/process_pb2.pyi +0 -0
  33. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/process_pb2_grpc.py +0 -0
  34. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/process_pb2_grpc.pyi +0 -0
  35. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/run_pb2.py +0 -0
  36. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/run_pb2.pyi +0 -0
  37. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/run_pb2_grpc.py +0 -0
  38. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/run_pb2_grpc.pyi +0 -0
  39. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/service_pb2.py +0 -0
  40. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/service_pb2.pyi +0 -0
  41. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/service_pb2_grpc.py +0 -0
  42. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/atc/v1/service_pb2_grpc.pyi +0 -0
  43. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/chat/v1/chat_pb2_grpc.py +0 -0
  44. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/chat/v1/chat_pb2_grpc.pyi +0 -0
  45. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/chat/v1/service_pb2.py +0 -0
  46. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/chat/v1/service_pb2.pyi +0 -0
  47. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/chat/v1/service_pb2_grpc.py +0 -0
  48. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/chat/v1/service_pb2_grpc.pyi +0 -0
  49. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/consolidation/v1/consolidation_pb2.py +0 -0
  50. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/consolidation/v1/consolidation_pb2.pyi +0 -0
  51. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/consolidation/v1/consolidation_pb2_grpc.py +0 -0
  52. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/consolidation/v1/consolidation_pb2_grpc.pyi +0 -0
  53. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/consolidation/v1/service_pb2.py +0 -0
  54. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/consolidation/v1/service_pb2.pyi +0 -0
  55. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/consolidation/v1/service_pb2_grpc.py +0 -0
  56. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/consolidation/v1/service_pb2_grpc.pyi +0 -0
  57. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/inference/v1/inference_pb2.py +0 -0
  58. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/inference/v1/inference_pb2.pyi +0 -0
  59. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/inference/v1/inference_pb2_grpc.py +0 -0
  60. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/inference/v1/inference_pb2_grpc.pyi +0 -0
  61. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/inference/v1/service_pb2.py +0 -0
  62. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/inference/v1/service_pb2.pyi +0 -0
  63. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/inference/v1/service_pb2_grpc.py +0 -0
  64. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/inference/v1/service_pb2_grpc.pyi +0 -0
  65. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/ingestion/v1/ingestion_pb2.py +0 -0
  66. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/ingestion/v1/ingestion_pb2.pyi +0 -0
  67. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/ingestion/v1/ingestion_pb2_grpc.py +0 -0
  68. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/ingestion/v1/ingestion_pb2_grpc.pyi +0 -0
  69. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/ingestion/v1/service_pb2.py +0 -0
  70. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/ingestion/v1/service_pb2.pyi +0 -0
  71. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/ingestion/v1/service_pb2_grpc.py +0 -0
  72. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/ingestion/v1/service_pb2_grpc.pyi +0 -0
  73. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/model/v1/model_pb2.py +0 -0
  74. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/model/v1/model_pb2.pyi +0 -0
  75. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/model/v1/model_pb2_grpc.py +0 -0
  76. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/model/v1/model_pb2_grpc.pyi +0 -0
  77. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/model/v1/service_pb2.py +0 -0
  78. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/model/v1/service_pb2.pyi +0 -0
  79. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/model/v1/service_pb2_grpc.py +0 -0
  80. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/model/v1/service_pb2_grpc.pyi +0 -0
  81. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/organization/v1/organization_pb2.py +0 -0
  82. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/organization/v1/organization_pb2.pyi +0 -0
  83. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/organization/v1/organization_pb2_grpc.py +0 -0
  84. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/organization/v1/organization_pb2_grpc.pyi +0 -0
  85. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/organization/v1/service_pb2.py +0 -0
  86. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/organization/v1/service_pb2.pyi +0 -0
  87. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/organization/v1/service_pb2_grpc.py +0 -0
  88. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/organization/v1/service_pb2_grpc.pyi +0 -0
  89. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/project/v1/project_pb2.py +0 -0
  90. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/project/v1/project_pb2.pyi +0 -0
  91. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/project/v1/project_pb2_grpc.py +0 -0
  92. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/project/v1/project_pb2_grpc.pyi +0 -0
  93. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/project/v1/service_pb2.py +0 -0
  94. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/project/v1/service_pb2.pyi +0 -0
  95. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/project/v1/service_pb2_grpc.py +0 -0
  96. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/project/v1/service_pb2_grpc.pyi +0 -0
  97. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/provider/v1/provider_pb2.py +0 -0
  98. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/provider/v1/provider_pb2.pyi +0 -0
  99. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/provider/v1/provider_pb2_grpc.py +0 -0
  100. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/provider/v1/provider_pb2_grpc.pyi +0 -0
  101. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/provider/v1/service_pb2.py +0 -0
  102. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/provider/v1/service_pb2.pyi +0 -0
  103. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/provider/v1/service_pb2_grpc.py +0 -0
  104. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/provider/v1/service_pb2_grpc.pyi +0 -0
  105. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/retrieval/v1/retrieval_pb2.py +0 -0
  106. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/retrieval/v1/retrieval_pb2.pyi +0 -0
  107. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/retrieval/v1/retrieval_pb2_grpc.py +0 -0
  108. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/retrieval/v1/retrieval_pb2_grpc.pyi +0 -0
  109. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/retrieval/v1/service_pb2.py +0 -0
  110. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/retrieval/v1/service_pb2.pyi +0 -0
  111. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/retrieval/v1/service_pb2_grpc.py +0 -0
  112. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/retrieval/v1/service_pb2_grpc.pyi +0 -0
  113. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/task/v1/service_pb2.py +0 -0
  114. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/task/v1/service_pb2.pyi +0 -0
  115. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/task/v1/service_pb2_grpc.py +0 -0
  116. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/task/v1/service_pb2_grpc.pyi +0 -0
  117. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/task/v1/task_pb2.py +0 -0
  118. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/task/v1/task_pb2.pyi +0 -0
  119. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/task/v1/task_pb2_grpc.py +0 -0
  120. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weaveapi/task/v1/task_pb2_grpc.pyi +0 -0
  121. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/atcdb/event.py +0 -0
  122. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/atcdb/inspection.py +0 -0
  123. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/atcdb/mailbox.py +0 -0
  124. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/atcdb/models.py +0 -0
  125. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/atcdb/run_template.py +0 -0
  126. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/atcdb/snapshot.py +0 -0
  127. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/atcdb/team.py +0 -0
  128. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/consolidation.py +0 -0
  129. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/document_security.py +0 -0
  130. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/ingestion.py +0 -0
  131. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/model_catalog.py +0 -0
  132. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/organization.py +0 -0
  133. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/project.py +0 -0
  134. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/provider.py +0 -0
  135. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/retrieval.py +0 -0
  136. {weave_python-0.38.0 → weave_python-0.39.0}/weave/weavesql/weavedb/task.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: weave-python
3
- Version: 0.38.0
3
+ Version: 0.39.0
4
4
  Author: Weave Labs
5
5
  License: FCL-1.0-ALv2
6
6
  License-File: LICENSE
@@ -0,0 +1,90 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # NO CHECKED-IN PROTOBUF GENCODE
4
+ # source: weaveapi/chat/v1/chat.proto
5
+ # Protobuf Python Version: 7.34.1
6
+ """Generated protocol buffer code."""
7
+
8
+ from google.protobuf import descriptor as _descriptor
9
+ from google.protobuf import descriptor_pool as _descriptor_pool
10
+ from google.protobuf import runtime_version as _runtime_version
11
+ from google.protobuf import symbol_database as _symbol_database
12
+ from google.protobuf.internal import builder as _builder
13
+
14
+ _runtime_version.ValidateProtobufRuntimeVersion(
15
+ _runtime_version.Domain.PUBLIC, 7, 34, 1, "", "weaveapi/chat/v1/chat.proto"
16
+ )
17
+ # @@protoc_insertion_point(imports)
18
+
19
+ _sym_db = _symbol_database.Default()
20
+
21
+
22
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
23
+ b'\n\x1bweaveapi/chat/v1/chat.proto\x12\x10weaveapi.chat.v1\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a%weaveapi/ingestion/v1/ingestion.proto"\xcc\x03\n\x0b\x43hatSession\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\'\n\x0forganization_id\x18\x02 \x01(\tR\x0eorganizationId\x12+\n\x12\x63reated_by_user_id\x18\x03 \x01(\tR\x0f\x63reatedByUserId\x12\x14\n\x05title\x18\x04 \x01(\tR\x05title\x12\x19\n\x08model_id\x18\x05 \x01(\tR\x07modelId\x12:\n\x19provider_configuration_id\x18\x06 \x01(\tR\x17providerConfigurationId\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x42\n\x0flast_message_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rlastMessageAt\x12\x30\n\x14\x61ttached_project_ids\x18\n \x03(\tR\x12\x61ttachedProjectIds"\xc2\x05\n\x0b\x43hatMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nsession_id\x18\x02 \x01(\tR\tsessionId\x12\x35\n\x04role\x18\x03 \x01(\x0e\x32!.weaveapi.chat.v1.ChatMessageRoleR\x04role\x12\x18\n\x07\x63ontent\x18\x04 \x01(\tR\x07\x63ontent\x12*\n\x11parent_message_id\x18\x05 \x01(\tR\x0fparentMessageId\x12 \n\x0ctool_call_id\x18\x06 \x01(\tR\ntoolCallId\x12K\n\x0ftool_call_state\x18\x07 \x01(\x0e\x32#.weaveapi.chat.v1.ChatToolCallStateR\rtoolCallState\x12\x19\n\x08model_id\x18\x08 \x01(\tR\x07modelId\x12<\n\x1a\x65phemeral_attachment_count\x18\t \x01(\x05R\x18\x65phemeralAttachmentCount\x12\x45\n\x1f\x65phemeral_attachment_mime_types\x18\n \x03(\tR\x1c\x65phemeralAttachmentMimeTypes\x12\x38\n\x0btoken_usage\x18\x0b \x01(\x0b\x32\x17.google.protobuf.StructR\ntokenUsage\x12\x44\n\x0cstream_state\x18\x0c \x01(\x0e\x32!.weaveapi.chat.v1.ChatStreamStateR\x0bstreamState\x12\x39\n\ncreated_at\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12=\n\x0c\x63ompleted_at\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0b\x63ompletedAt"\xc6\x02\n\x0c\x43hatArtifact\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nsession_id\x18\x02 \x01(\tR\tsessionId\x12\x14\n\x05title\x18\x03 \x01(\tR\x05title\x12\x36\n\x04kind\x18\x04 \x01(\x0e\x32".weaveapi.chat.v1.ChatArtifactKindR\x04kind\x12\x1a\n\x08language\x18\x05 \x01(\tR\x08language\x12\'\n\x0f\x63urrent_version\x18\x06 \x01(\x05R\x0e\x63urrentVersion\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt"\xc6\x02\n\x13\x43hatArtifactVersion\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1f\n\x0b\x61rtifact_id\x18\x02 \x01(\tR\nartifactId\x12\x18\n\x07version\x18\x03 \x01(\x05R\x07version\x12\x18\n\x07\x63ontent\x18\x04 \x01(\tR\x07\x63ontent\x12\x31\n\x15\x63reated_by_message_id\x18\x05 \x01(\tR\x12\x63reatedByMessageId\x12\x39\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\\\n\x13\x64ocument_references\x18\x07 \x03(\x0b\x32\'.weaveapi.chat.v1.ChatDocumentReferenceB\x02(\x01R\x12\x64ocumentReferences"\xef\x01\n\x15\x43hatDocumentReference\x12\x1f\n\x0b\x64ocument_id\x18\x01 \x01(\tR\ndocumentId\x12#\n\rdocument_name\x18\x02 \x01(\tR\x0c\x64ocumentName\x12\x1b\n\tmime_type\x18\x03 \x01(\tR\x08mimeType\x12\x1d\n\nthread_ids\x18\x04 \x03(\tR\tthreadIds\x12T\n\x10source_locations\x18\x05 \x03(\x0b\x32%.weaveapi.ingestion.v1.SourceLocationB\x02(\x01R\x0fsourceLocations"J\n\x17\x43hatEphemeralAttachment\x12\x1b\n\tmime_type\x18\x01 \x01(\tR\x08mimeType\x12\x12\n\x04\x64\x61ta\x18\x02 \x01(\x0cR\x04\x64\x61ta"\xdc\x10\n\x0f\x43hatStreamFrame\x12\x19\n\x08\x65vent_id\x18\x0b \x01(\tR\x07\x65ventId\x12[\n\x0fmessage_started\x18\x01 \x01(\x0b\x32\x30.weaveapi.chat.v1.ChatStreamFrame.MessageStartedH\x00R\x0emessageStarted\x12U\n\rmessage_delta\x18\x02 \x01(\x0b\x32..weaveapi.chat.v1.ChatStreamFrame.MessageDeltaH\x00R\x0cmessageDelta\x12\x62\n\x12tool_call_proposed\x18\x03 \x01(\x0b\x32\x32.weaveapi.chat.v1.ChatStreamFrame.ToolCallProposedH\x00R\x10toolCallProposed\x12\x65\n\x13tool_call_executing\x18\x04 \x01(\x0b\x32\x33.weaveapi.chat.v1.ChatStreamFrame.ToolCallExecutingH\x00R\x11toolCallExecuting\x12\\\n\x10tool_call_result\x18\x05 \x01(\x0b\x32\x30.weaveapi.chat.v1.ChatStreamFrame.ToolCallResultH\x00R\x0etoolCallResult\x12\x62\n\x12tool_call_rejected\x18\x06 \x01(\x0b\x32\x32.weaveapi.chat.v1.ChatStreamFrame.ToolCallRejectedH\x00R\x10toolCallRejected\x12^\n\x10\x61rtifact_created\x18\x07 \x01(\x0b\x32\x31.weaveapi.chat.v1.ChatStreamFrame.ArtifactCreatedH\x00R\x0f\x61rtifactCreated\x12^\n\x10\x61rtifact_updated\x18\x08 \x01(\x0b\x32\x31.weaveapi.chat.v1.ChatStreamFrame.ArtifactUpdatedH\x00R\x0f\x61rtifactUpdated\x12^\n\x10message_complete\x18\t \x01(\x0b\x32\x31.weaveapi.chat.v1.ChatStreamFrame.MessageCompleteH\x00R\x0fmessageComplete\x12U\n\rmessage_error\x18\n \x01(\x0b\x32..weaveapi.chat.v1.ChatStreamFrame.MessageErrorH\x00R\x0cmessageError\x1a\x66\n\x0eMessageStarted\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12\x35\n\x04role\x18\x02 \x01(\x0e\x32!.weaveapi.chat.v1.ChatMessageRoleR\x04role\x1aL\n\x0cMessageDelta\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12\x1d\n\ntext_delta\x18\x02 \x01(\tR\ttextDelta\x1a\xcb\x01\n\x10ToolCallProposed\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12 \n\x0ctool_call_id\x18\x02 \x01(\tR\ntoolCallId\x12\x12\n\x04name\x18\x03 \x01(\tR\x04name\x12\x35\n\targuments\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\targuments\x12+\n\x11requires_approval\x18\x05 \x01(\x08R\x10requiresApproval\x1a\x80\x01\n\x11ToolCallExecuting\x12 \n\x0ctool_call_id\x18\x01 \x01(\tR\ntoolCallId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x35\n\targuments\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\targuments\x1a\x64\n\x0eToolCallResult\x12 \n\x0ctool_call_id\x18\x01 \x01(\tR\ntoolCallId\x12\x16\n\x06status\x18\x02 \x01(\tR\x06status\x12\x18\n\x07summary\x18\x03 \x01(\tR\x07summary\x1aL\n\x10ToolCallRejected\x12 \n\x0ctool_call_id\x18\x01 \x01(\tR\ntoolCallId\x12\x16\n\x06reason\x18\x02 \x01(\tR\x06reason\x1a\x9a\x01\n\x0f\x41rtifactCreated\x12\x1f\n\x0b\x61rtifact_id\x18\x01 \x01(\tR\nartifactId\x12\x14\n\x05title\x18\x02 \x01(\tR\x05title\x12\x36\n\x04kind\x18\x03 \x01(\x0e\x32".weaveapi.chat.v1.ChatArtifactKindR\x04kind\x12\x18\n\x07version\x18\x04 \x01(\x05R\x07version\x1aL\n\x0f\x41rtifactUpdated\x12\x1f\n\x0b\x61rtifact_id\x18\x01 \x01(\tR\nartifactId\x12\x18\n\x07version\x18\x02 \x01(\x05R\x07version\x1aj\n\x0fMessageComplete\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12\x38\n\x0btoken_usage\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructR\ntokenUsage\x1a[\n\x0cMessageError\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12\x12\n\x04\x63ode\x18\x02 \x01(\tR\x04\x63ode\x12\x18\n\x07message\x18\x03 \x01(\tR\x07messageB\x07\n\x05\x66rame*\xab\x01\n\x0f\x43hatMessageRole\x12!\n\x1d\x43HAT_MESSAGE_ROLE_UNSPECIFIED\x10\x00\x12\x1a\n\x16\x43HAT_MESSAGE_ROLE_USER\x10\x01\x12\x1f\n\x1b\x43HAT_MESSAGE_ROLE_ASSISTANT\x10\x02\x12\x1a\n\x16\x43HAT_MESSAGE_ROLE_TOOL\x10\x03\x12\x1c\n\x18\x43HAT_MESSAGE_ROLE_SYSTEM\x10\x04*\x85\x02\n\x11\x43hatToolCallState\x12$\n CHAT_TOOL_CALL_STATE_UNSPECIFIED\x10\x00\x12\x1d\n\x19\x43HAT_TOOL_CALL_STATE_NONE\x10\x01\x12!\n\x1d\x43HAT_TOOL_CALL_STATE_PROPOSED\x10\x02\x12!\n\x1d\x43HAT_TOOL_CALL_STATE_APPROVED\x10\x03\x12!\n\x1d\x43HAT_TOOL_CALL_STATE_REJECTED\x10\x04\x12!\n\x1d\x43HAT_TOOL_CALL_STATE_EXECUTED\x10\x05\x12\x1f\n\x1b\x43HAT_TOOL_CALL_STATE_FAILED\x10\x06*\x92\x01\n\x0f\x43hatStreamState\x12!\n\x1d\x43HAT_STREAM_STATE_UNSPECIFIED\x10\x00\x12\x1f\n\x1b\x43HAT_STREAM_STATE_STREAMING\x10\x01\x12\x1e\n\x1a\x43HAT_STREAM_STATE_COMPLETE\x10\x02\x12\x1b\n\x17\x43HAT_STREAM_STATE_ERROR\x10\x03*\xde\x01\n\x10\x43hatArtifactKind\x12"\n\x1e\x43HAT_ARTIFACT_KIND_UNSPECIFIED\x10\x00\x12\x1f\n\x1b\x43HAT_ARTIFACT_KIND_MARKDOWN\x10\x01\x12\x1b\n\x17\x43HAT_ARTIFACT_KIND_HTML\x10\x02\x12 \n\x1c\x43HAT_ARTIFACT_KIND_PLAINTEXT\x10\x03\x12\x1b\n\x17\x43HAT_ARTIFACT_KIND_CODE\x10\x04\x12)\n%CHAT_ARTIFACT_KIND_DOCUMENT_REFERENCE\x10\x05\x42\xb9\x01\n\x14\x63om.weaveapi.chat.v1B\tChatProtoP\x01Z4github.com/weave-labs/weave-go/weaveapi/chat/v1;chat\xa2\x02\x03WCX\xaa\x02\x10Weaveapi.Chat.V1\xca\x02\x10Weaveapi\\Chat\\V1\xe2\x02\x1cWeaveapi\\Chat\\V1\\GPBMetadata\xea\x02\x12Weaveapi::Chat::V1b\x08\x65\x64itionsp\xe8\x07'
24
+ )
25
+
26
+ _globals = globals()
27
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
28
+ _builder.BuildTopDescriptorsAndMessages(
29
+ DESCRIPTOR, "weaveapi.chat.v1.chat_pb2", _globals
30
+ )
31
+ if not _descriptor._USE_C_DESCRIPTORS:
32
+ _globals["DESCRIPTOR"]._loaded_options = None
33
+ _globals[
34
+ "DESCRIPTOR"
35
+ ]._serialized_options = b"\n\024com.weaveapi.chat.v1B\tChatProtoP\001Z4github.com/weave-labs/weave-go/weaveapi/chat/v1;chat\242\002\003WCX\252\002\020Weaveapi.Chat.V1\312\002\020Weaveapi\\Chat\\V1\342\002\034Weaveapi\\Chat\\V1\\GPBMetadata\352\002\022Weaveapi::Chat::V1"
36
+ _globals["_CHATARTIFACTVERSION"].fields_by_name[
37
+ "document_references"
38
+ ]._loaded_options = None
39
+ _globals["_CHATARTIFACTVERSION"].fields_by_name[
40
+ "document_references"
41
+ ]._serialized_options = b"(\001"
42
+ _globals["_CHATDOCUMENTREFERENCE"].fields_by_name[
43
+ "source_locations"
44
+ ]._loaded_options = None
45
+ _globals["_CHATDOCUMENTREFERENCE"].fields_by_name[
46
+ "source_locations"
47
+ ]._serialized_options = b"(\001"
48
+ _globals["_CHATMESSAGEROLE"]._serialized_start = 4443
49
+ _globals["_CHATMESSAGEROLE"]._serialized_end = 4614
50
+ _globals["_CHATTOOLCALLSTATE"]._serialized_start = 4617
51
+ _globals["_CHATTOOLCALLSTATE"]._serialized_end = 4878
52
+ _globals["_CHATSTREAMSTATE"]._serialized_start = 4881
53
+ _globals["_CHATSTREAMSTATE"]._serialized_end = 5027
54
+ _globals["_CHATARTIFACTKIND"]._serialized_start = 5030
55
+ _globals["_CHATARTIFACTKIND"]._serialized_end = 5252
56
+ _globals["_CHATSESSION"]._serialized_start = 152
57
+ _globals["_CHATSESSION"]._serialized_end = 612
58
+ _globals["_CHATMESSAGE"]._serialized_start = 615
59
+ _globals["_CHATMESSAGE"]._serialized_end = 1321
60
+ _globals["_CHATARTIFACT"]._serialized_start = 1324
61
+ _globals["_CHATARTIFACT"]._serialized_end = 1650
62
+ _globals["_CHATARTIFACTVERSION"]._serialized_start = 1653
63
+ _globals["_CHATARTIFACTVERSION"]._serialized_end = 1979
64
+ _globals["_CHATDOCUMENTREFERENCE"]._serialized_start = 1982
65
+ _globals["_CHATDOCUMENTREFERENCE"]._serialized_end = 2221
66
+ _globals["_CHATEPHEMERALATTACHMENT"]._serialized_start = 2223
67
+ _globals["_CHATEPHEMERALATTACHMENT"]._serialized_end = 2297
68
+ _globals["_CHATSTREAMFRAME"]._serialized_start = 2300
69
+ _globals["_CHATSTREAMFRAME"]._serialized_end = 4440
70
+ _globals["_CHATSTREAMFRAME_MESSAGESTARTED"]._serialized_start = 3298
71
+ _globals["_CHATSTREAMFRAME_MESSAGESTARTED"]._serialized_end = 3400
72
+ _globals["_CHATSTREAMFRAME_MESSAGEDELTA"]._serialized_start = 3402
73
+ _globals["_CHATSTREAMFRAME_MESSAGEDELTA"]._serialized_end = 3478
74
+ _globals["_CHATSTREAMFRAME_TOOLCALLPROPOSED"]._serialized_start = 3481
75
+ _globals["_CHATSTREAMFRAME_TOOLCALLPROPOSED"]._serialized_end = 3684
76
+ _globals["_CHATSTREAMFRAME_TOOLCALLEXECUTING"]._serialized_start = 3687
77
+ _globals["_CHATSTREAMFRAME_TOOLCALLEXECUTING"]._serialized_end = 3815
78
+ _globals["_CHATSTREAMFRAME_TOOLCALLRESULT"]._serialized_start = 3817
79
+ _globals["_CHATSTREAMFRAME_TOOLCALLRESULT"]._serialized_end = 3917
80
+ _globals["_CHATSTREAMFRAME_TOOLCALLREJECTED"]._serialized_start = 3919
81
+ _globals["_CHATSTREAMFRAME_TOOLCALLREJECTED"]._serialized_end = 3995
82
+ _globals["_CHATSTREAMFRAME_ARTIFACTCREATED"]._serialized_start = 3998
83
+ _globals["_CHATSTREAMFRAME_ARTIFACTCREATED"]._serialized_end = 4152
84
+ _globals["_CHATSTREAMFRAME_ARTIFACTUPDATED"]._serialized_start = 4154
85
+ _globals["_CHATSTREAMFRAME_ARTIFACTUPDATED"]._serialized_end = 4230
86
+ _globals["_CHATSTREAMFRAME_MESSAGECOMPLETE"]._serialized_start = 4232
87
+ _globals["_CHATSTREAMFRAME_MESSAGECOMPLETE"]._serialized_end = 4338
88
+ _globals["_CHATSTREAMFRAME_MESSAGEERROR"]._serialized_start = 4340
89
+ _globals["_CHATSTREAMFRAME_MESSAGEERROR"]._serialized_end = 4431
90
+ # @@protoc_insertion_point(module_scope)
@@ -10,6 +10,7 @@ from google.protobuf import struct_pb2 as _struct_pb2
10
10
  from google.protobuf import timestamp_pb2 as _timestamp_pb2
11
11
  from google.protobuf.internal import containers as _containers
12
12
  from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
13
+ from weaveapi.ingestion.v1 import ingestion_pb2 as _ingestion_pb2
13
14
  import builtins as _builtins
14
15
  import sys
15
16
  import typing as _typing
@@ -107,6 +108,7 @@ class _ChatArtifactKindEnumTypeWrapper(
107
108
  CHAT_ARTIFACT_KIND_HTML: _ChatArtifactKind.ValueType # 2
108
109
  CHAT_ARTIFACT_KIND_PLAINTEXT: _ChatArtifactKind.ValueType # 3
109
110
  CHAT_ARTIFACT_KIND_CODE: _ChatArtifactKind.ValueType # 4
111
+ CHAT_ARTIFACT_KIND_DOCUMENT_REFERENCE: _ChatArtifactKind.ValueType # 5
110
112
 
111
113
  class ChatArtifactKind(
112
114
  _ChatArtifactKind, metaclass=_ChatArtifactKindEnumTypeWrapper
@@ -117,6 +119,7 @@ CHAT_ARTIFACT_KIND_MARKDOWN: ChatArtifactKind.ValueType # 1
117
119
  CHAT_ARTIFACT_KIND_HTML: ChatArtifactKind.ValueType # 2
118
120
  CHAT_ARTIFACT_KIND_PLAINTEXT: ChatArtifactKind.ValueType # 3
119
121
  CHAT_ARTIFACT_KIND_CODE: ChatArtifactKind.ValueType # 4
122
+ CHAT_ARTIFACT_KIND_DOCUMENT_REFERENCE: ChatArtifactKind.ValueType # 5
120
123
  Global___ChatArtifactKind: _TypeAlias = ChatArtifactKind # noqa: Y015
121
124
 
122
125
  @_typing.final
@@ -418,6 +421,7 @@ class ChatArtifactVersion(_message.Message):
418
421
  CONTENT_FIELD_NUMBER: _builtins.int
419
422
  CREATED_BY_MESSAGE_ID_FIELD_NUMBER: _builtins.int
420
423
  CREATED_AT_FIELD_NUMBER: _builtins.int
424
+ DOCUMENT_REFERENCES_FIELD_NUMBER: _builtins.int
421
425
  id: _builtins.str
422
426
  artifact_id: _builtins.str
423
427
  version: _builtins.int
@@ -425,6 +429,12 @@ class ChatArtifactVersion(_message.Message):
425
429
  created_by_message_id: _builtins.str
426
430
  @_builtins.property
427
431
  def created_at(self) -> _timestamp_pb2.Timestamp: ...
432
+ @_builtins.property
433
+ def document_references(
434
+ self,
435
+ ) -> _containers.RepeatedCompositeFieldContainer[
436
+ Global___ChatDocumentReference
437
+ ]: ...
428
438
  def __init__(
429
439
  self,
430
440
  *,
@@ -434,6 +444,7 @@ class ChatArtifactVersion(_message.Message):
434
444
  content: _builtins.str | None = ...,
435
445
  created_by_message_id: _builtins.str | None = ...,
436
446
  created_at: _timestamp_pb2.Timestamp | None = ...,
447
+ document_references: _abc.Iterable[Global___ChatDocumentReference] | None = ...,
437
448
  ) -> None: ...
438
449
  _HasFieldArgType: _TypeAlias = _typing.Literal[
439
450
  "artifact_id",
@@ -444,6 +455,8 @@ class ChatArtifactVersion(_message.Message):
444
455
  b"created_at",
445
456
  "created_by_message_id",
446
457
  b"created_by_message_id",
458
+ "document_references",
459
+ b"document_references",
447
460
  "id",
448
461
  b"id",
449
462
  "version",
@@ -459,6 +472,8 @@ class ChatArtifactVersion(_message.Message):
459
472
  b"created_at",
460
473
  "created_by_message_id",
461
474
  b"created_by_message_id",
475
+ "document_references",
476
+ b"document_references",
462
477
  "id",
463
478
  b"id",
464
479
  "version",
@@ -468,6 +483,62 @@ class ChatArtifactVersion(_message.Message):
468
483
 
469
484
  Global___ChatArtifactVersion: _TypeAlias = ChatArtifactVersion # noqa: Y015
470
485
 
486
+ @_typing.final
487
+ class ChatDocumentReference(_message.Message):
488
+ DESCRIPTOR: _descriptor.Descriptor
489
+
490
+ DOCUMENT_ID_FIELD_NUMBER: _builtins.int
491
+ DOCUMENT_NAME_FIELD_NUMBER: _builtins.int
492
+ MIME_TYPE_FIELD_NUMBER: _builtins.int
493
+ THREAD_IDS_FIELD_NUMBER: _builtins.int
494
+ SOURCE_LOCATIONS_FIELD_NUMBER: _builtins.int
495
+ document_id: _builtins.str
496
+ document_name: _builtins.str
497
+ mime_type: _builtins.str
498
+ @_builtins.property
499
+ def thread_ids(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
500
+ @_builtins.property
501
+ def source_locations(
502
+ self,
503
+ ) -> _containers.RepeatedCompositeFieldContainer[_ingestion_pb2.SourceLocation]: ...
504
+ def __init__(
505
+ self,
506
+ *,
507
+ document_id: _builtins.str | None = ...,
508
+ document_name: _builtins.str | None = ...,
509
+ mime_type: _builtins.str | None = ...,
510
+ thread_ids: _abc.Iterable[_builtins.str] | None = ...,
511
+ source_locations: _abc.Iterable[_ingestion_pb2.SourceLocation] | None = ...,
512
+ ) -> None: ...
513
+ _HasFieldArgType: _TypeAlias = _typing.Literal[
514
+ "document_id",
515
+ b"document_id",
516
+ "document_name",
517
+ b"document_name",
518
+ "mime_type",
519
+ b"mime_type",
520
+ "source_locations",
521
+ b"source_locations",
522
+ "thread_ids",
523
+ b"thread_ids",
524
+ ] # noqa: Y015
525
+ def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
526
+ _ClearFieldArgType: _TypeAlias = _typing.Literal[
527
+ "document_id",
528
+ b"document_id",
529
+ "document_name",
530
+ b"document_name",
531
+ "mime_type",
532
+ b"mime_type",
533
+ "source_locations",
534
+ b"source_locations",
535
+ "thread_ids",
536
+ b"thread_ids",
537
+ ] # noqa: Y015
538
+ def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
539
+
540
+ Global___ChatDocumentReference: _TypeAlias = ChatDocumentReference # noqa: Y015
541
+
471
542
  @_typing.final
472
543
  class ChatEphemeralAttachment(_message.Message):
473
544
  DESCRIPTOR: _descriptor.Descriptor
@@ -173,7 +173,7 @@ class GetChatArtifactParams:
173
173
 
174
174
 
175
175
  GET_CHAT_ARTIFACT_VERSION = """-- name: get_chat_artifact_version \\:one
176
- SELECT id, artifact_id, version, content, created_by_message_id, created_at FROM weave.chat_artifact_versions
176
+ SELECT id, artifact_id, version, content, created_by_message_id, created_at, document_references FROM weave.chat_artifact_versions
177
177
  WHERE artifact_id = :p1 AND version = :p2
178
178
  """
179
179
 
@@ -235,7 +235,7 @@ INSERT_CHAT_ARTIFACT_VERSION = """-- name: insert_chat_artifact_version \\:one
235
235
  INSERT INTO weave.chat_artifact_versions (
236
236
  id, artifact_id, version, content, created_by_message_id
237
237
  ) VALUES (:p1, :p2, :p3, :p4, :p5)
238
- RETURNING id, artifact_id, version, content, created_by_message_id, created_at
238
+ RETURNING id, artifact_id, version, content, created_by_message_id, created_at, document_references
239
239
  """
240
240
 
241
241
 
@@ -248,6 +248,24 @@ class InsertChatArtifactVersionParams:
248
248
  created_by_message_id: uuid.UUID
249
249
 
250
250
 
251
+ INSERT_CHAT_ARTIFACT_VERSION_WITH_DOCUMENT_REFERENCES = """-- name: insert_chat_artifact_version_with_document_references \\:one
252
+ INSERT INTO weave.chat_artifact_versions (
253
+ id, artifact_id, version, content, created_by_message_id, document_references
254
+ ) VALUES (:p1, :p2, :p3, :p4, :p5, :p6)
255
+ RETURNING id, artifact_id, version, content, created_by_message_id, created_at, document_references
256
+ """
257
+
258
+
259
+ @dataclasses.dataclass()
260
+ class InsertChatArtifactVersionWithDocumentReferencesParams:
261
+ id: uuid.UUID
262
+ artifact_id: uuid.UUID
263
+ version: int
264
+ content: str
265
+ created_by_message_id: uuid.UUID
266
+ document_references: Any
267
+
268
+
251
269
  INSERT_CHAT_MESSAGE = """-- name: insert_chat_message \\:one
252
270
  INSERT INTO weave.chat_messages (
253
271
  id, session_id, role, content,
@@ -280,7 +298,7 @@ class InsertChatMessageParams:
280
298
 
281
299
 
282
300
  LIST_CHAT_ARTIFACT_VERSIONS = """-- name: list_chat_artifact_versions \\:many
283
- SELECT id, artifact_id, version, content, created_by_message_id, created_at FROM weave.chat_artifact_versions
301
+ SELECT id, artifact_id, version, content, created_by_message_id, created_at, document_references FROM weave.chat_artifact_versions
284
302
  WHERE artifact_id = :p1
285
303
  ORDER BY version
286
304
  """
@@ -672,6 +690,7 @@ class Querier:
672
690
  content=row[3],
673
691
  created_by_message_id=row[4],
674
692
  created_at=row[5],
693
+ document_references=row[6],
675
694
  )
676
695
 
677
696
  def get_chat_message(
@@ -780,6 +799,33 @@ class Querier:
780
799
  content=row[3],
781
800
  created_by_message_id=row[4],
782
801
  created_at=row[5],
802
+ document_references=row[6],
803
+ )
804
+
805
+ def insert_chat_artifact_version_with_document_references(
806
+ self, arg: InsertChatArtifactVersionWithDocumentReferencesParams
807
+ ) -> Optional[models.WeaveChatArtifactVersion]:
808
+ row = self._conn.execute(
809
+ sqlalchemy.text(INSERT_CHAT_ARTIFACT_VERSION_WITH_DOCUMENT_REFERENCES),
810
+ {
811
+ "p1": arg.id,
812
+ "p2": arg.artifact_id,
813
+ "p3": arg.version,
814
+ "p4": arg.content,
815
+ "p5": arg.created_by_message_id,
816
+ "p6": arg.document_references,
817
+ },
818
+ ).first()
819
+ if row is None:
820
+ return None
821
+ return models.WeaveChatArtifactVersion(
822
+ id=row[0],
823
+ artifact_id=row[1],
824
+ version=row[2],
825
+ content=row[3],
826
+ created_by_message_id=row[4],
827
+ created_at=row[5],
828
+ document_references=row[6],
783
829
  )
784
830
 
785
831
  def insert_chat_message(
@@ -837,6 +883,7 @@ class Querier:
837
883
  content=row[3],
838
884
  created_by_message_id=row[4],
839
885
  created_at=row[5],
886
+ document_references=row[6],
840
887
  )
841
888
 
842
889
  def list_chat_artifacts_for_session(
@@ -1281,6 +1328,7 @@ class AsyncQuerier:
1281
1328
  content=row[3],
1282
1329
  created_by_message_id=row[4],
1283
1330
  created_at=row[5],
1331
+ document_references=row[6],
1284
1332
  )
1285
1333
 
1286
1334
  async def get_chat_message(
@@ -1396,6 +1444,35 @@ class AsyncQuerier:
1396
1444
  content=row[3],
1397
1445
  created_by_message_id=row[4],
1398
1446
  created_at=row[5],
1447
+ document_references=row[6],
1448
+ )
1449
+
1450
+ async def insert_chat_artifact_version_with_document_references(
1451
+ self, arg: InsertChatArtifactVersionWithDocumentReferencesParams
1452
+ ) -> Optional[models.WeaveChatArtifactVersion]:
1453
+ row = (
1454
+ await self._conn.execute(
1455
+ sqlalchemy.text(INSERT_CHAT_ARTIFACT_VERSION_WITH_DOCUMENT_REFERENCES),
1456
+ {
1457
+ "p1": arg.id,
1458
+ "p2": arg.artifact_id,
1459
+ "p3": arg.version,
1460
+ "p4": arg.content,
1461
+ "p5": arg.created_by_message_id,
1462
+ "p6": arg.document_references,
1463
+ },
1464
+ )
1465
+ ).first()
1466
+ if row is None:
1467
+ return None
1468
+ return models.WeaveChatArtifactVersion(
1469
+ id=row[0],
1470
+ artifact_id=row[1],
1471
+ version=row[2],
1472
+ content=row[3],
1473
+ created_by_message_id=row[4],
1474
+ created_at=row[5],
1475
+ document_references=row[6],
1399
1476
  )
1400
1477
 
1401
1478
  async def insert_chat_message(
@@ -1455,6 +1532,7 @@ class AsyncQuerier:
1455
1532
  content=row[3],
1456
1533
  created_by_message_id=row[4],
1457
1534
  created_at=row[5],
1535
+ document_references=row[6],
1458
1536
  )
1459
1537
 
1460
1538
  async def list_chat_artifacts_for_session(
@@ -27,6 +27,7 @@ class WeaveChatArtifactVersion:
27
27
  content: str
28
28
  created_by_message_id: uuid.UUID
29
29
  created_at: datetime.datetime
30
+ document_references: Any
30
31
 
31
32
 
32
33
  @dataclasses.dataclass()
@@ -1,76 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # NO CHECKED-IN PROTOBUF GENCODE
4
- # source: weaveapi/chat/v1/chat.proto
5
- # Protobuf Python Version: 7.34.1
6
- """Generated protocol buffer code."""
7
-
8
- from google.protobuf import descriptor as _descriptor
9
- from google.protobuf import descriptor_pool as _descriptor_pool
10
- from google.protobuf import runtime_version as _runtime_version
11
- from google.protobuf import symbol_database as _symbol_database
12
- from google.protobuf.internal import builder as _builder
13
-
14
- _runtime_version.ValidateProtobufRuntimeVersion(
15
- _runtime_version.Domain.PUBLIC, 7, 34, 1, "", "weaveapi/chat/v1/chat.proto"
16
- )
17
- # @@protoc_insertion_point(imports)
18
-
19
- _sym_db = _symbol_database.Default()
20
-
21
-
22
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
23
- b'\n\x1bweaveapi/chat/v1/chat.proto\x12\x10weaveapi.chat.v1\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xcc\x03\n\x0b\x43hatSession\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\'\n\x0forganization_id\x18\x02 \x01(\tR\x0eorganizationId\x12+\n\x12\x63reated_by_user_id\x18\x03 \x01(\tR\x0f\x63reatedByUserId\x12\x14\n\x05title\x18\x04 \x01(\tR\x05title\x12\x19\n\x08model_id\x18\x05 \x01(\tR\x07modelId\x12:\n\x19provider_configuration_id\x18\x06 \x01(\tR\x17providerConfigurationId\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x42\n\x0flast_message_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rlastMessageAt\x12\x30\n\x14\x61ttached_project_ids\x18\n \x03(\tR\x12\x61ttachedProjectIds"\xc2\x05\n\x0b\x43hatMessage\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nsession_id\x18\x02 \x01(\tR\tsessionId\x12\x35\n\x04role\x18\x03 \x01(\x0e\x32!.weaveapi.chat.v1.ChatMessageRoleR\x04role\x12\x18\n\x07\x63ontent\x18\x04 \x01(\tR\x07\x63ontent\x12*\n\x11parent_message_id\x18\x05 \x01(\tR\x0fparentMessageId\x12 \n\x0ctool_call_id\x18\x06 \x01(\tR\ntoolCallId\x12K\n\x0ftool_call_state\x18\x07 \x01(\x0e\x32#.weaveapi.chat.v1.ChatToolCallStateR\rtoolCallState\x12\x19\n\x08model_id\x18\x08 \x01(\tR\x07modelId\x12<\n\x1a\x65phemeral_attachment_count\x18\t \x01(\x05R\x18\x65phemeralAttachmentCount\x12\x45\n\x1f\x65phemeral_attachment_mime_types\x18\n \x03(\tR\x1c\x65phemeralAttachmentMimeTypes\x12\x38\n\x0btoken_usage\x18\x0b \x01(\x0b\x32\x17.google.protobuf.StructR\ntokenUsage\x12\x44\n\x0cstream_state\x18\x0c \x01(\x0e\x32!.weaveapi.chat.v1.ChatStreamStateR\x0bstreamState\x12\x39\n\ncreated_at\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12=\n\x0c\x63ompleted_at\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0b\x63ompletedAt"\xc6\x02\n\x0c\x43hatArtifact\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nsession_id\x18\x02 \x01(\tR\tsessionId\x12\x14\n\x05title\x18\x03 \x01(\tR\x05title\x12\x36\n\x04kind\x18\x04 \x01(\x0e\x32".weaveapi.chat.v1.ChatArtifactKindR\x04kind\x12\x1a\n\x08language\x18\x05 \x01(\tR\x08language\x12\'\n\x0f\x63urrent_version\x18\x06 \x01(\x05R\x0e\x63urrentVersion\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt"\xe8\x01\n\x13\x43hatArtifactVersion\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1f\n\x0b\x61rtifact_id\x18\x02 \x01(\tR\nartifactId\x12\x18\n\x07version\x18\x03 \x01(\x05R\x07version\x12\x18\n\x07\x63ontent\x18\x04 \x01(\tR\x07\x63ontent\x12\x31\n\x15\x63reated_by_message_id\x18\x05 \x01(\tR\x12\x63reatedByMessageId\x12\x39\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt"J\n\x17\x43hatEphemeralAttachment\x12\x1b\n\tmime_type\x18\x01 \x01(\tR\x08mimeType\x12\x12\n\x04\x64\x61ta\x18\x02 \x01(\x0cR\x04\x64\x61ta"\xdc\x10\n\x0f\x43hatStreamFrame\x12\x19\n\x08\x65vent_id\x18\x0b \x01(\tR\x07\x65ventId\x12[\n\x0fmessage_started\x18\x01 \x01(\x0b\x32\x30.weaveapi.chat.v1.ChatStreamFrame.MessageStartedH\x00R\x0emessageStarted\x12U\n\rmessage_delta\x18\x02 \x01(\x0b\x32..weaveapi.chat.v1.ChatStreamFrame.MessageDeltaH\x00R\x0cmessageDelta\x12\x62\n\x12tool_call_proposed\x18\x03 \x01(\x0b\x32\x32.weaveapi.chat.v1.ChatStreamFrame.ToolCallProposedH\x00R\x10toolCallProposed\x12\x65\n\x13tool_call_executing\x18\x04 \x01(\x0b\x32\x33.weaveapi.chat.v1.ChatStreamFrame.ToolCallExecutingH\x00R\x11toolCallExecuting\x12\\\n\x10tool_call_result\x18\x05 \x01(\x0b\x32\x30.weaveapi.chat.v1.ChatStreamFrame.ToolCallResultH\x00R\x0etoolCallResult\x12\x62\n\x12tool_call_rejected\x18\x06 \x01(\x0b\x32\x32.weaveapi.chat.v1.ChatStreamFrame.ToolCallRejectedH\x00R\x10toolCallRejected\x12^\n\x10\x61rtifact_created\x18\x07 \x01(\x0b\x32\x31.weaveapi.chat.v1.ChatStreamFrame.ArtifactCreatedH\x00R\x0f\x61rtifactCreated\x12^\n\x10\x61rtifact_updated\x18\x08 \x01(\x0b\x32\x31.weaveapi.chat.v1.ChatStreamFrame.ArtifactUpdatedH\x00R\x0f\x61rtifactUpdated\x12^\n\x10message_complete\x18\t \x01(\x0b\x32\x31.weaveapi.chat.v1.ChatStreamFrame.MessageCompleteH\x00R\x0fmessageComplete\x12U\n\rmessage_error\x18\n \x01(\x0b\x32..weaveapi.chat.v1.ChatStreamFrame.MessageErrorH\x00R\x0cmessageError\x1a\x66\n\x0eMessageStarted\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12\x35\n\x04role\x18\x02 \x01(\x0e\x32!.weaveapi.chat.v1.ChatMessageRoleR\x04role\x1aL\n\x0cMessageDelta\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12\x1d\n\ntext_delta\x18\x02 \x01(\tR\ttextDelta\x1a\xcb\x01\n\x10ToolCallProposed\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12 \n\x0ctool_call_id\x18\x02 \x01(\tR\ntoolCallId\x12\x12\n\x04name\x18\x03 \x01(\tR\x04name\x12\x35\n\targuments\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\targuments\x12+\n\x11requires_approval\x18\x05 \x01(\x08R\x10requiresApproval\x1a\x80\x01\n\x11ToolCallExecuting\x12 \n\x0ctool_call_id\x18\x01 \x01(\tR\ntoolCallId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x35\n\targuments\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\targuments\x1a\x64\n\x0eToolCallResult\x12 \n\x0ctool_call_id\x18\x01 \x01(\tR\ntoolCallId\x12\x16\n\x06status\x18\x02 \x01(\tR\x06status\x12\x18\n\x07summary\x18\x03 \x01(\tR\x07summary\x1aL\n\x10ToolCallRejected\x12 \n\x0ctool_call_id\x18\x01 \x01(\tR\ntoolCallId\x12\x16\n\x06reason\x18\x02 \x01(\tR\x06reason\x1a\x9a\x01\n\x0f\x41rtifactCreated\x12\x1f\n\x0b\x61rtifact_id\x18\x01 \x01(\tR\nartifactId\x12\x14\n\x05title\x18\x02 \x01(\tR\x05title\x12\x36\n\x04kind\x18\x03 \x01(\x0e\x32".weaveapi.chat.v1.ChatArtifactKindR\x04kind\x12\x18\n\x07version\x18\x04 \x01(\x05R\x07version\x1aL\n\x0f\x41rtifactUpdated\x12\x1f\n\x0b\x61rtifact_id\x18\x01 \x01(\tR\nartifactId\x12\x18\n\x07version\x18\x02 \x01(\x05R\x07version\x1aj\n\x0fMessageComplete\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12\x38\n\x0btoken_usage\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructR\ntokenUsage\x1a[\n\x0cMessageError\x12\x1d\n\nmessage_id\x18\x01 \x01(\tR\tmessageId\x12\x12\n\x04\x63ode\x18\x02 \x01(\tR\x04\x63ode\x12\x18\n\x07message\x18\x03 \x01(\tR\x07messageB\x07\n\x05\x66rame*\xab\x01\n\x0f\x43hatMessageRole\x12!\n\x1d\x43HAT_MESSAGE_ROLE_UNSPECIFIED\x10\x00\x12\x1a\n\x16\x43HAT_MESSAGE_ROLE_USER\x10\x01\x12\x1f\n\x1b\x43HAT_MESSAGE_ROLE_ASSISTANT\x10\x02\x12\x1a\n\x16\x43HAT_MESSAGE_ROLE_TOOL\x10\x03\x12\x1c\n\x18\x43HAT_MESSAGE_ROLE_SYSTEM\x10\x04*\x85\x02\n\x11\x43hatToolCallState\x12$\n CHAT_TOOL_CALL_STATE_UNSPECIFIED\x10\x00\x12\x1d\n\x19\x43HAT_TOOL_CALL_STATE_NONE\x10\x01\x12!\n\x1d\x43HAT_TOOL_CALL_STATE_PROPOSED\x10\x02\x12!\n\x1d\x43HAT_TOOL_CALL_STATE_APPROVED\x10\x03\x12!\n\x1d\x43HAT_TOOL_CALL_STATE_REJECTED\x10\x04\x12!\n\x1d\x43HAT_TOOL_CALL_STATE_EXECUTED\x10\x05\x12\x1f\n\x1b\x43HAT_TOOL_CALL_STATE_FAILED\x10\x06*\x92\x01\n\x0f\x43hatStreamState\x12!\n\x1d\x43HAT_STREAM_STATE_UNSPECIFIED\x10\x00\x12\x1f\n\x1b\x43HAT_STREAM_STATE_STREAMING\x10\x01\x12\x1e\n\x1a\x43HAT_STREAM_STATE_COMPLETE\x10\x02\x12\x1b\n\x17\x43HAT_STREAM_STATE_ERROR\x10\x03*\xb3\x01\n\x10\x43hatArtifactKind\x12"\n\x1e\x43HAT_ARTIFACT_KIND_UNSPECIFIED\x10\x00\x12\x1f\n\x1b\x43HAT_ARTIFACT_KIND_MARKDOWN\x10\x01\x12\x1b\n\x17\x43HAT_ARTIFACT_KIND_HTML\x10\x02\x12 \n\x1c\x43HAT_ARTIFACT_KIND_PLAINTEXT\x10\x03\x12\x1b\n\x17\x43HAT_ARTIFACT_KIND_CODE\x10\x04\x42\xb9\x01\n\x14\x63om.weaveapi.chat.v1B\tChatProtoP\x01Z4github.com/weave-labs/weave-go/weaveapi/chat/v1;chat\xa2\x02\x03WCX\xaa\x02\x10Weaveapi.Chat.V1\xca\x02\x10Weaveapi\\Chat\\V1\xe2\x02\x1cWeaveapi\\Chat\\V1\\GPBMetadata\xea\x02\x12Weaveapi::Chat::V1b\x08\x65\x64itionsp\xe8\x07'
24
- )
25
-
26
- _globals = globals()
27
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
28
- _builder.BuildTopDescriptorsAndMessages(
29
- DESCRIPTOR, "weaveapi.chat.v1.chat_pb2", _globals
30
- )
31
- if not _descriptor._USE_C_DESCRIPTORS:
32
- _globals["DESCRIPTOR"]._loaded_options = None
33
- _globals[
34
- "DESCRIPTOR"
35
- ]._serialized_options = b"\n\024com.weaveapi.chat.v1B\tChatProtoP\001Z4github.com/weave-labs/weave-go/weaveapi/chat/v1;chat\242\002\003WCX\252\002\020Weaveapi.Chat.V1\312\002\020Weaveapi\\Chat\\V1\342\002\034Weaveapi\\Chat\\V1\\GPBMetadata\352\002\022Weaveapi::Chat::V1"
36
- _globals["_CHATMESSAGEROLE"]._serialized_start = 4068
37
- _globals["_CHATMESSAGEROLE"]._serialized_end = 4239
38
- _globals["_CHATTOOLCALLSTATE"]._serialized_start = 4242
39
- _globals["_CHATTOOLCALLSTATE"]._serialized_end = 4503
40
- _globals["_CHATSTREAMSTATE"]._serialized_start = 4506
41
- _globals["_CHATSTREAMSTATE"]._serialized_end = 4652
42
- _globals["_CHATARTIFACTKIND"]._serialized_start = 4655
43
- _globals["_CHATARTIFACTKIND"]._serialized_end = 4834
44
- _globals["_CHATSESSION"]._serialized_start = 113
45
- _globals["_CHATSESSION"]._serialized_end = 573
46
- _globals["_CHATMESSAGE"]._serialized_start = 576
47
- _globals["_CHATMESSAGE"]._serialized_end = 1282
48
- _globals["_CHATARTIFACT"]._serialized_start = 1285
49
- _globals["_CHATARTIFACT"]._serialized_end = 1611
50
- _globals["_CHATARTIFACTVERSION"]._serialized_start = 1614
51
- _globals["_CHATARTIFACTVERSION"]._serialized_end = 1846
52
- _globals["_CHATEPHEMERALATTACHMENT"]._serialized_start = 1848
53
- _globals["_CHATEPHEMERALATTACHMENT"]._serialized_end = 1922
54
- _globals["_CHATSTREAMFRAME"]._serialized_start = 1925
55
- _globals["_CHATSTREAMFRAME"]._serialized_end = 4065
56
- _globals["_CHATSTREAMFRAME_MESSAGESTARTED"]._serialized_start = 2923
57
- _globals["_CHATSTREAMFRAME_MESSAGESTARTED"]._serialized_end = 3025
58
- _globals["_CHATSTREAMFRAME_MESSAGEDELTA"]._serialized_start = 3027
59
- _globals["_CHATSTREAMFRAME_MESSAGEDELTA"]._serialized_end = 3103
60
- _globals["_CHATSTREAMFRAME_TOOLCALLPROPOSED"]._serialized_start = 3106
61
- _globals["_CHATSTREAMFRAME_TOOLCALLPROPOSED"]._serialized_end = 3309
62
- _globals["_CHATSTREAMFRAME_TOOLCALLEXECUTING"]._serialized_start = 3312
63
- _globals["_CHATSTREAMFRAME_TOOLCALLEXECUTING"]._serialized_end = 3440
64
- _globals["_CHATSTREAMFRAME_TOOLCALLRESULT"]._serialized_start = 3442
65
- _globals["_CHATSTREAMFRAME_TOOLCALLRESULT"]._serialized_end = 3542
66
- _globals["_CHATSTREAMFRAME_TOOLCALLREJECTED"]._serialized_start = 3544
67
- _globals["_CHATSTREAMFRAME_TOOLCALLREJECTED"]._serialized_end = 3620
68
- _globals["_CHATSTREAMFRAME_ARTIFACTCREATED"]._serialized_start = 3623
69
- _globals["_CHATSTREAMFRAME_ARTIFACTCREATED"]._serialized_end = 3777
70
- _globals["_CHATSTREAMFRAME_ARTIFACTUPDATED"]._serialized_start = 3779
71
- _globals["_CHATSTREAMFRAME_ARTIFACTUPDATED"]._serialized_end = 3855
72
- _globals["_CHATSTREAMFRAME_MESSAGECOMPLETE"]._serialized_start = 3857
73
- _globals["_CHATSTREAMFRAME_MESSAGECOMPLETE"]._serialized_end = 3963
74
- _globals["_CHATSTREAMFRAME_MESSAGEERROR"]._serialized_start = 3965
75
- _globals["_CHATSTREAMFRAME_MESSAGEERROR"]._serialized_end = 4056
76
- # @@protoc_insertion_point(module_scope)
File without changes
File without changes
File without changes
File without changes