edsl 0.1.39.dev3__py3-none-any.whl → 0.1.39.dev4__py3-none-any.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 (344) hide show
  1. edsl/Base.py +413 -332
  2. edsl/BaseDiff.py +260 -260
  3. edsl/TemplateLoader.py +24 -24
  4. edsl/__init__.py +57 -49
  5. edsl/__version__.py +1 -1
  6. edsl/agents/Agent.py +1071 -867
  7. edsl/agents/AgentList.py +551 -413
  8. edsl/agents/Invigilator.py +284 -233
  9. edsl/agents/InvigilatorBase.py +257 -270
  10. edsl/agents/PromptConstructor.py +272 -354
  11. edsl/agents/QuestionInstructionPromptBuilder.py +128 -0
  12. edsl/agents/QuestionTemplateReplacementsBuilder.py +137 -0
  13. edsl/agents/__init__.py +2 -3
  14. edsl/agents/descriptors.py +99 -99
  15. edsl/agents/prompt_helpers.py +129 -129
  16. edsl/agents/question_option_processor.py +172 -0
  17. edsl/auto/AutoStudy.py +130 -117
  18. edsl/auto/StageBase.py +243 -230
  19. edsl/auto/StageGenerateSurvey.py +178 -178
  20. edsl/auto/StageLabelQuestions.py +125 -125
  21. edsl/auto/StagePersona.py +61 -61
  22. edsl/auto/StagePersonaDimensionValueRanges.py +88 -88
  23. edsl/auto/StagePersonaDimensionValues.py +74 -74
  24. edsl/auto/StagePersonaDimensions.py +69 -69
  25. edsl/auto/StageQuestions.py +74 -73
  26. edsl/auto/SurveyCreatorPipeline.py +21 -21
  27. edsl/auto/utilities.py +218 -224
  28. edsl/base/Base.py +279 -279
  29. edsl/config.py +177 -157
  30. edsl/conversation/Conversation.py +290 -290
  31. edsl/conversation/car_buying.py +59 -58
  32. edsl/conversation/chips.py +95 -95
  33. edsl/conversation/mug_negotiation.py +81 -81
  34. edsl/conversation/next_speaker_utilities.py +93 -93
  35. edsl/coop/CoopFunctionsMixin.py +15 -0
  36. edsl/coop/ExpectedParrotKeyHandler.py +125 -0
  37. edsl/coop/PriceFetcher.py +54 -54
  38. edsl/coop/__init__.py +2 -2
  39. edsl/coop/coop.py +1106 -1028
  40. edsl/coop/utils.py +131 -131
  41. edsl/data/Cache.py +573 -555
  42. edsl/data/CacheEntry.py +230 -233
  43. edsl/data/CacheHandler.py +168 -149
  44. edsl/data/RemoteCacheSync.py +186 -78
  45. edsl/data/SQLiteDict.py +292 -292
  46. edsl/data/__init__.py +5 -4
  47. edsl/data/hack.py +10 -0
  48. edsl/data/orm.py +10 -10
  49. edsl/data_transfer_models.py +74 -73
  50. edsl/enums.py +202 -175
  51. edsl/exceptions/BaseException.py +21 -21
  52. edsl/exceptions/__init__.py +54 -54
  53. edsl/exceptions/agents.py +54 -42
  54. edsl/exceptions/cache.py +5 -5
  55. edsl/exceptions/configuration.py +16 -16
  56. edsl/exceptions/coop.py +10 -10
  57. edsl/exceptions/data.py +14 -14
  58. edsl/exceptions/general.py +34 -34
  59. edsl/exceptions/inference_services.py +5 -0
  60. edsl/exceptions/jobs.py +33 -33
  61. edsl/exceptions/language_models.py +63 -63
  62. edsl/exceptions/prompts.py +15 -15
  63. edsl/exceptions/questions.py +109 -91
  64. edsl/exceptions/results.py +29 -29
  65. edsl/exceptions/scenarios.py +29 -22
  66. edsl/exceptions/surveys.py +37 -37
  67. edsl/inference_services/AnthropicService.py +106 -87
  68. edsl/inference_services/AvailableModelCacheHandler.py +184 -0
  69. edsl/inference_services/AvailableModelFetcher.py +215 -0
  70. edsl/inference_services/AwsBedrock.py +118 -120
  71. edsl/inference_services/AzureAI.py +215 -217
  72. edsl/inference_services/DeepInfraService.py +18 -18
  73. edsl/inference_services/GoogleService.py +143 -148
  74. edsl/inference_services/GroqService.py +20 -20
  75. edsl/inference_services/InferenceServiceABC.py +80 -147
  76. edsl/inference_services/InferenceServicesCollection.py +138 -97
  77. edsl/inference_services/MistralAIService.py +120 -123
  78. edsl/inference_services/OllamaService.py +18 -18
  79. edsl/inference_services/OpenAIService.py +236 -224
  80. edsl/inference_services/PerplexityService.py +160 -163
  81. edsl/inference_services/ServiceAvailability.py +135 -0
  82. edsl/inference_services/TestService.py +90 -89
  83. edsl/inference_services/TogetherAIService.py +172 -170
  84. edsl/inference_services/data_structures.py +134 -0
  85. edsl/inference_services/models_available_cache.py +118 -118
  86. edsl/inference_services/rate_limits_cache.py +25 -25
  87. edsl/inference_services/registry.py +41 -41
  88. edsl/inference_services/write_available.py +10 -10
  89. edsl/jobs/AnswerQuestionFunctionConstructor.py +223 -0
  90. edsl/jobs/Answers.py +43 -56
  91. edsl/jobs/FetchInvigilator.py +47 -0
  92. edsl/jobs/InterviewTaskManager.py +98 -0
  93. edsl/jobs/InterviewsConstructor.py +50 -0
  94. edsl/jobs/Jobs.py +823 -898
  95. edsl/jobs/JobsChecks.py +172 -147
  96. edsl/jobs/JobsComponentConstructor.py +189 -0
  97. edsl/jobs/JobsPrompts.py +270 -268
  98. edsl/jobs/JobsRemoteInferenceHandler.py +311 -239
  99. edsl/jobs/JobsRemoteInferenceLogger.py +239 -0
  100. edsl/jobs/RequestTokenEstimator.py +30 -0
  101. edsl/jobs/__init__.py +1 -1
  102. edsl/jobs/async_interview_runner.py +138 -0
  103. edsl/jobs/buckets/BucketCollection.py +104 -63
  104. edsl/jobs/buckets/ModelBuckets.py +65 -65
  105. edsl/jobs/buckets/TokenBucket.py +283 -251
  106. edsl/jobs/buckets/TokenBucketAPI.py +211 -0
  107. edsl/jobs/buckets/TokenBucketClient.py +191 -0
  108. edsl/jobs/check_survey_scenario_compatibility.py +85 -0
  109. edsl/jobs/data_structures.py +120 -0
  110. edsl/jobs/decorators.py +35 -0
  111. edsl/jobs/interviews/Interview.py +396 -661
  112. edsl/jobs/interviews/InterviewExceptionCollection.py +99 -99
  113. edsl/jobs/interviews/InterviewExceptionEntry.py +186 -186
  114. edsl/jobs/interviews/InterviewStatistic.py +63 -63
  115. edsl/jobs/interviews/InterviewStatisticsCollection.py +25 -25
  116. edsl/jobs/interviews/InterviewStatusDictionary.py +78 -78
  117. edsl/jobs/interviews/InterviewStatusLog.py +92 -92
  118. edsl/jobs/interviews/ReportErrors.py +66 -66
  119. edsl/jobs/interviews/interview_status_enum.py +9 -9
  120. edsl/jobs/jobs_status_enums.py +9 -0
  121. edsl/jobs/loggers/HTMLTableJobLogger.py +304 -0
  122. edsl/jobs/results_exceptions_handler.py +98 -0
  123. edsl/jobs/runners/JobsRunnerAsyncio.py +151 -466
  124. edsl/jobs/runners/JobsRunnerStatus.py +297 -330
  125. edsl/jobs/tasks/QuestionTaskCreator.py +244 -242
  126. edsl/jobs/tasks/TaskCreators.py +64 -64
  127. edsl/jobs/tasks/TaskHistory.py +470 -450
  128. edsl/jobs/tasks/TaskStatusLog.py +23 -23
  129. edsl/jobs/tasks/task_status_enum.py +161 -163
  130. edsl/jobs/tokens/InterviewTokenUsage.py +27 -27
  131. edsl/jobs/tokens/TokenUsage.py +34 -34
  132. edsl/language_models/ComputeCost.py +63 -0
  133. edsl/language_models/LanguageModel.py +626 -668
  134. edsl/language_models/ModelList.py +164 -155
  135. edsl/language_models/PriceManager.py +127 -0
  136. edsl/language_models/RawResponseHandler.py +106 -0
  137. edsl/language_models/RegisterLanguageModelsMeta.py +184 -184
  138. edsl/language_models/ServiceDataSources.py +0 -0
  139. edsl/language_models/__init__.py +2 -3
  140. edsl/language_models/fake_openai_call.py +15 -15
  141. edsl/language_models/fake_openai_service.py +61 -61
  142. edsl/language_models/key_management/KeyLookup.py +63 -0
  143. edsl/language_models/key_management/KeyLookupBuilder.py +273 -0
  144. edsl/language_models/key_management/KeyLookupCollection.py +38 -0
  145. edsl/language_models/key_management/__init__.py +0 -0
  146. edsl/language_models/key_management/models.py +131 -0
  147. edsl/language_models/model.py +256 -0
  148. edsl/language_models/repair.py +156 -156
  149. edsl/language_models/utilities.py +65 -64
  150. edsl/notebooks/Notebook.py +263 -258
  151. edsl/notebooks/NotebookToLaTeX.py +142 -0
  152. edsl/notebooks/__init__.py +1 -1
  153. edsl/prompts/Prompt.py +352 -362
  154. edsl/prompts/__init__.py +2 -2
  155. edsl/questions/ExceptionExplainer.py +77 -0
  156. edsl/questions/HTMLQuestion.py +103 -0
  157. edsl/questions/QuestionBase.py +518 -664
  158. edsl/questions/QuestionBasePromptsMixin.py +221 -217
  159. edsl/questions/QuestionBudget.py +227 -227
  160. edsl/questions/QuestionCheckBox.py +359 -359
  161. edsl/questions/QuestionExtract.py +180 -182
  162. edsl/questions/QuestionFreeText.py +113 -114
  163. edsl/questions/QuestionFunctional.py +166 -166
  164. edsl/questions/QuestionList.py +223 -231
  165. edsl/questions/QuestionMatrix.py +265 -0
  166. edsl/questions/QuestionMultipleChoice.py +330 -286
  167. edsl/questions/QuestionNumerical.py +151 -153
  168. edsl/questions/QuestionRank.py +314 -324
  169. edsl/questions/Quick.py +41 -41
  170. edsl/questions/SimpleAskMixin.py +74 -73
  171. edsl/questions/__init__.py +27 -26
  172. edsl/questions/{AnswerValidatorMixin.py → answer_validator_mixin.py} +334 -289
  173. edsl/questions/compose_questions.py +98 -98
  174. edsl/questions/data_structures.py +20 -0
  175. edsl/questions/decorators.py +21 -21
  176. edsl/questions/derived/QuestionLikertFive.py +76 -76
  177. edsl/questions/derived/QuestionLinearScale.py +90 -87
  178. edsl/questions/derived/QuestionTopK.py +93 -93
  179. edsl/questions/derived/QuestionYesNo.py +82 -82
  180. edsl/questions/descriptors.py +427 -413
  181. edsl/questions/loop_processor.py +149 -0
  182. edsl/questions/prompt_templates/question_budget.jinja +13 -13
  183. edsl/questions/prompt_templates/question_checkbox.jinja +32 -32
  184. edsl/questions/prompt_templates/question_extract.jinja +11 -11
  185. edsl/questions/prompt_templates/question_free_text.jinja +3 -3
  186. edsl/questions/prompt_templates/question_linear_scale.jinja +11 -11
  187. edsl/questions/prompt_templates/question_list.jinja +17 -17
  188. edsl/questions/prompt_templates/question_multiple_choice.jinja +33 -33
  189. edsl/questions/prompt_templates/question_numerical.jinja +36 -36
  190. edsl/questions/{QuestionBaseGenMixin.py → question_base_gen_mixin.py} +168 -161
  191. edsl/questions/question_registry.py +177 -177
  192. edsl/questions/{RegisterQuestionsMeta.py → register_questions_meta.py} +71 -71
  193. edsl/questions/{ResponseValidatorABC.py → response_validator_abc.py} +188 -174
  194. edsl/questions/response_validator_factory.py +34 -0
  195. edsl/questions/settings.py +12 -12
  196. edsl/questions/templates/budget/answering_instructions.jinja +7 -7
  197. edsl/questions/templates/budget/question_presentation.jinja +7 -7
  198. edsl/questions/templates/checkbox/answering_instructions.jinja +10 -10
  199. edsl/questions/templates/checkbox/question_presentation.jinja +22 -22
  200. edsl/questions/templates/extract/answering_instructions.jinja +7 -7
  201. edsl/questions/templates/likert_five/answering_instructions.jinja +10 -10
  202. edsl/questions/templates/likert_five/question_presentation.jinja +11 -11
  203. edsl/questions/templates/linear_scale/answering_instructions.jinja +5 -5
  204. edsl/questions/templates/linear_scale/question_presentation.jinja +5 -5
  205. edsl/questions/templates/list/answering_instructions.jinja +3 -3
  206. edsl/questions/templates/list/question_presentation.jinja +5 -5
  207. edsl/questions/templates/matrix/__init__.py +1 -0
  208. edsl/questions/templates/matrix/answering_instructions.jinja +5 -0
  209. edsl/questions/templates/matrix/question_presentation.jinja +20 -0
  210. edsl/questions/templates/multiple_choice/answering_instructions.jinja +9 -9
  211. edsl/questions/templates/multiple_choice/question_presentation.jinja +11 -11
  212. edsl/questions/templates/numerical/answering_instructions.jinja +6 -6
  213. edsl/questions/templates/numerical/question_presentation.jinja +6 -6
  214. edsl/questions/templates/rank/answering_instructions.jinja +11 -11
  215. edsl/questions/templates/rank/question_presentation.jinja +15 -15
  216. edsl/questions/templates/top_k/answering_instructions.jinja +8 -8
  217. edsl/questions/templates/top_k/question_presentation.jinja +22 -22
  218. edsl/questions/templates/yes_no/answering_instructions.jinja +6 -6
  219. edsl/questions/templates/yes_no/question_presentation.jinja +11 -11
  220. edsl/results/CSSParameterizer.py +108 -108
  221. edsl/results/Dataset.py +587 -424
  222. edsl/results/DatasetExportMixin.py +594 -731
  223. edsl/results/DatasetTree.py +295 -275
  224. edsl/results/MarkdownToDocx.py +122 -0
  225. edsl/results/MarkdownToPDF.py +111 -0
  226. edsl/results/Result.py +557 -465
  227. edsl/results/Results.py +1183 -1165
  228. edsl/results/ResultsExportMixin.py +45 -43
  229. edsl/results/ResultsGGMixin.py +121 -121
  230. edsl/results/TableDisplay.py +125 -198
  231. edsl/results/TextEditor.py +50 -0
  232. edsl/results/__init__.py +2 -2
  233. edsl/results/file_exports.py +252 -0
  234. edsl/results/{ResultsFetchMixin.py → results_fetch_mixin.py} +33 -33
  235. edsl/results/{Selector.py → results_selector.py} +145 -135
  236. edsl/results/{ResultsToolsMixin.py → results_tools_mixin.py} +98 -98
  237. edsl/results/smart_objects.py +96 -0
  238. edsl/results/table_data_class.py +12 -0
  239. edsl/results/table_display.css +77 -77
  240. edsl/results/table_renderers.py +118 -0
  241. edsl/results/tree_explore.py +115 -115
  242. edsl/scenarios/ConstructDownloadLink.py +109 -0
  243. edsl/scenarios/DocumentChunker.py +102 -0
  244. edsl/scenarios/DocxScenario.py +16 -0
  245. edsl/scenarios/FileStore.py +511 -632
  246. edsl/scenarios/PdfExtractor.py +40 -0
  247. edsl/scenarios/Scenario.py +498 -601
  248. edsl/scenarios/ScenarioHtmlMixin.py +65 -64
  249. edsl/scenarios/ScenarioList.py +1458 -1287
  250. edsl/scenarios/ScenarioListExportMixin.py +45 -52
  251. edsl/scenarios/ScenarioListPdfMixin.py +239 -261
  252. edsl/scenarios/__init__.py +3 -4
  253. edsl/scenarios/directory_scanner.py +96 -0
  254. edsl/scenarios/file_methods.py +85 -0
  255. edsl/scenarios/handlers/__init__.py +13 -0
  256. edsl/scenarios/handlers/csv.py +38 -0
  257. edsl/scenarios/handlers/docx.py +76 -0
  258. edsl/scenarios/handlers/html.py +37 -0
  259. edsl/scenarios/handlers/json.py +111 -0
  260. edsl/scenarios/handlers/latex.py +5 -0
  261. edsl/scenarios/handlers/md.py +51 -0
  262. edsl/scenarios/handlers/pdf.py +68 -0
  263. edsl/scenarios/handlers/png.py +39 -0
  264. edsl/scenarios/handlers/pptx.py +105 -0
  265. edsl/scenarios/handlers/py.py +294 -0
  266. edsl/scenarios/handlers/sql.py +313 -0
  267. edsl/scenarios/handlers/sqlite.py +149 -0
  268. edsl/scenarios/handlers/txt.py +33 -0
  269. edsl/scenarios/{ScenarioJoin.py → scenario_join.py} +131 -127
  270. edsl/scenarios/scenario_selector.py +156 -0
  271. edsl/shared.py +1 -1
  272. edsl/study/ObjectEntry.py +173 -173
  273. edsl/study/ProofOfWork.py +113 -113
  274. edsl/study/SnapShot.py +80 -80
  275. edsl/study/Study.py +521 -528
  276. edsl/study/__init__.py +4 -4
  277. edsl/surveys/ConstructDAG.py +92 -0
  278. edsl/surveys/DAG.py +148 -148
  279. edsl/surveys/EditSurvey.py +221 -0
  280. edsl/surveys/InstructionHandler.py +100 -0
  281. edsl/surveys/Memory.py +31 -31
  282. edsl/surveys/MemoryManagement.py +72 -0
  283. edsl/surveys/MemoryPlan.py +244 -244
  284. edsl/surveys/Rule.py +327 -326
  285. edsl/surveys/RuleCollection.py +385 -387
  286. edsl/surveys/RuleManager.py +172 -0
  287. edsl/surveys/Simulator.py +75 -0
  288. edsl/surveys/Survey.py +1280 -1801
  289. edsl/surveys/SurveyCSS.py +273 -261
  290. edsl/surveys/SurveyExportMixin.py +259 -259
  291. edsl/surveys/{SurveyFlowVisualizationMixin.py → SurveyFlowVisualization.py} +181 -179
  292. edsl/surveys/SurveyQualtricsImport.py +284 -284
  293. edsl/surveys/SurveyToApp.py +141 -0
  294. edsl/surveys/__init__.py +5 -3
  295. edsl/surveys/base.py +53 -53
  296. edsl/surveys/descriptors.py +60 -56
  297. edsl/surveys/instructions/ChangeInstruction.py +48 -49
  298. edsl/surveys/instructions/Instruction.py +56 -65
  299. edsl/surveys/instructions/InstructionCollection.py +82 -77
  300. edsl/templates/error_reporting/base.html +23 -23
  301. edsl/templates/error_reporting/exceptions_by_model.html +34 -34
  302. edsl/templates/error_reporting/exceptions_by_question_name.html +16 -16
  303. edsl/templates/error_reporting/exceptions_by_type.html +16 -16
  304. edsl/templates/error_reporting/interview_details.html +115 -115
  305. edsl/templates/error_reporting/interviews.html +19 -19
  306. edsl/templates/error_reporting/overview.html +4 -4
  307. edsl/templates/error_reporting/performance_plot.html +1 -1
  308. edsl/templates/error_reporting/report.css +73 -73
  309. edsl/templates/error_reporting/report.html +117 -117
  310. edsl/templates/error_reporting/report.js +25 -25
  311. edsl/test_h +1 -0
  312. edsl/tools/__init__.py +1 -1
  313. edsl/tools/clusters.py +192 -192
  314. edsl/tools/embeddings.py +27 -27
  315. edsl/tools/embeddings_plotting.py +118 -118
  316. edsl/tools/plotting.py +112 -112
  317. edsl/tools/summarize.py +18 -18
  318. edsl/utilities/PrettyList.py +56 -0
  319. edsl/utilities/SystemInfo.py +28 -28
  320. edsl/utilities/__init__.py +22 -22
  321. edsl/utilities/ast_utilities.py +25 -25
  322. edsl/utilities/data/Registry.py +6 -6
  323. edsl/utilities/data/__init__.py +1 -1
  324. edsl/utilities/data/scooter_results.json +1 -1
  325. edsl/utilities/decorators.py +77 -77
  326. edsl/utilities/gcp_bucket/cloud_storage.py +96 -96
  327. edsl/utilities/gcp_bucket/example.py +50 -0
  328. edsl/utilities/interface.py +627 -627
  329. edsl/utilities/is_notebook.py +18 -0
  330. edsl/utilities/is_valid_variable_name.py +11 -0
  331. edsl/utilities/naming_utilities.py +263 -263
  332. edsl/utilities/remove_edsl_version.py +24 -0
  333. edsl/utilities/repair_functions.py +28 -28
  334. edsl/utilities/restricted_python.py +70 -70
  335. edsl/utilities/utilities.py +436 -424
  336. {edsl-0.1.39.dev3.dist-info → edsl-0.1.39.dev4.dist-info}/LICENSE +21 -21
  337. {edsl-0.1.39.dev3.dist-info → edsl-0.1.39.dev4.dist-info}/METADATA +13 -11
  338. edsl-0.1.39.dev4.dist-info/RECORD +361 -0
  339. edsl/language_models/KeyLookup.py +0 -30
  340. edsl/language_models/registry.py +0 -190
  341. edsl/language_models/unused/ReplicateBase.py +0 -83
  342. edsl/results/ResultsDBMixin.py +0 -238
  343. edsl-0.1.39.dev3.dist-info/RECORD +0 -277
  344. {edsl-0.1.39.dev3.dist-info → edsl-0.1.39.dev4.dist-info}/WHEEL +0 -0
@@ -1,77 +1,82 @@
1
- from edsl.surveys.instructions.Instruction import Instruction
2
- from edsl.surveys.instructions.ChangeInstruction import ChangeInstruction
3
- from edsl.questions import QuestionBase
4
- from typing import Union, Optional, List, Generator, Dict
5
-
6
-
7
- from collections import UserDict
8
-
9
-
10
- class InstructionCollection(UserDict):
11
- def __init__(
12
- self,
13
- instruction_names_to_instruction: Dict[str, Instruction],
14
- questions: List[QuestionBase],
15
- ):
16
- self.instruction_names_to_instruction = instruction_names_to_instruction
17
- self.questions = questions
18
- data = {}
19
- for question in self.questions:
20
- # just add both the question and the question name
21
- data[question.name] = list(self.relevant_instructions(question))
22
- # data[question] = list(self.relevant_instructions(question))
23
- super().__init__(data)
24
-
25
- def __getitem__(self, key):
26
- # in case the person uses question instead of the name
27
- if isinstance(key, QuestionBase):
28
- key = key.name
29
- return self.data[key]
30
-
31
- @property
32
- def question_names(self):
33
- return [q.name for q in self.questions]
34
-
35
- def question_index(self, question_name):
36
- return self.question_names.index(question_name)
37
-
38
- def _entries_before(
39
- self, question_name
40
- ) -> tuple[List[Instruction], List[ChangeInstruction]]:
41
- if question_name not in self.question_names:
42
- raise ValueError(
43
- f"Question name not found in the list of questions: got '{question_name}'; list is {self.question_names}"
44
- )
45
- instructions, changes = [], []
46
-
47
- index = self.question_index(question_name)
48
- for instruction in self.instruction_names_to_instruction.values():
49
- if instruction.pseudo_index < index:
50
- if isinstance(instruction, Instruction):
51
- instructions.append(instruction)
52
- elif isinstance(instruction, ChangeInstruction):
53
- changes.append(instruction)
54
- return instructions, changes
55
-
56
- def relevant_instructions(
57
- self, question: Union[str, QuestionBase]
58
- ) -> Generator[Instruction, None, None]:
59
- ## Find all the questions that are after a given instruction
60
- if isinstance(question, str):
61
- question_name = question
62
- elif isinstance(question, QuestionBase):
63
- question_name = question.name
64
-
65
- instructions_before, changes_before = self._entries_before(question_name)
66
- keep_list = []
67
- drop_list = []
68
- for change in changes_before:
69
- keep_list.extend(change.keep)
70
- drop_list.extend(change.drop)
71
-
72
- for instruction in instructions_before:
73
- if instruction.name in keep_list or instruction.name not in drop_list:
74
- yield instruction
75
-
76
- def __len__(self):
77
- return len(self.instruction_names_to_instruction)
1
+ from typing import TYPE_CHECKING, Dict, List, Generator, Union
2
+ from collections import UserDict
3
+
4
+ from edsl.surveys.instructions.Instruction import Instruction
5
+ from edsl.surveys.instructions.ChangeInstruction import ChangeInstruction
6
+
7
+ if TYPE_CHECKING:
8
+ from edsl.questions.QuestionBase import QuestionBase
9
+
10
+
11
+ class InstructionCollection(UserDict):
12
+ def __init__(
13
+ self,
14
+ instruction_names_to_instruction: Dict[str, Instruction],
15
+ questions: List["QuestionBase"],
16
+ ):
17
+ self.instruction_names_to_instruction = instruction_names_to_instruction
18
+ self.questions = questions
19
+ data = {}
20
+ for question in self.questions:
21
+ # just add both the question and the question name
22
+ data[question.name] = list(self.relevant_instructions(question))
23
+ # data[question] = list(self.relevant_instructions(question))
24
+ super().__init__(data)
25
+
26
+ def __getitem__(self, key):
27
+ # in case the person uses question instead of the name
28
+ from edsl.questions.QuestionBase import QuestionBase
29
+
30
+ if isinstance(key, QuestionBase):
31
+ key = key.name
32
+ return self.data[key]
33
+
34
+ @property
35
+ def question_names(self):
36
+ return [q.name for q in self.questions]
37
+
38
+ def question_index(self, question_name):
39
+ return self.question_names.index(question_name)
40
+
41
+ def _entries_before(
42
+ self, question_name
43
+ ) -> tuple[List[Instruction], List[ChangeInstruction]]:
44
+ if question_name not in self.question_names:
45
+ raise ValueError(
46
+ f"Question name not found in the list of questions: got '{question_name}'; list is {self.question_names}"
47
+ )
48
+ instructions, changes = [], []
49
+
50
+ index = self.question_index(question_name)
51
+ for instruction in self.instruction_names_to_instruction.values():
52
+ if instruction.pseudo_index < index:
53
+ if isinstance(instruction, Instruction):
54
+ instructions.append(instruction)
55
+ elif isinstance(instruction, ChangeInstruction):
56
+ changes.append(instruction)
57
+ return instructions, changes
58
+
59
+ def relevant_instructions(
60
+ self, question: Union[str, "QuestionBase"]
61
+ ) -> Generator[Instruction, None, None]:
62
+ ## Find all the questions that are after a given instruction
63
+ from edsl.questions.QuestionBase import QuestionBase
64
+
65
+ if isinstance(question, str):
66
+ question_name = question
67
+ elif isinstance(question, QuestionBase):
68
+ question_name = question.name
69
+
70
+ instructions_before, changes_before = self._entries_before(question_name)
71
+ keep_list = []
72
+ drop_list = []
73
+ for change in changes_before:
74
+ keep_list.extend(change.keep)
75
+ drop_list.extend(change.drop)
76
+
77
+ for instruction in instructions_before:
78
+ if instruction.name in keep_list or instruction.name not in drop_list:
79
+ yield instruction
80
+
81
+ def __len__(self):
82
+ return len(self.instruction_names_to_instruction)
@@ -1,24 +1,24 @@
1
- <!DOCTYPE html>
2
- <html lang="en">
3
- <head>
4
- <meta charset="UTF-8">
5
- <meta name="viewport" content="width=device-width, initial-scale=1.0">
6
- <title>Exception Details</title>
7
- <style>
8
- {{ css }}
9
- </style>
10
-
11
- <script>
12
- {{ javascript }}
13
- </script>
14
-
15
- </head>
16
- <body>
17
- {% include 'overview.html' %}
18
- {% include 'exceptions_by_type.html' %}
19
- {% include 'exceptions_by_model.html' %}
20
- {% include 'exceptions_by_question_name.html' %}
21
- {% include 'interviews.html' %}
22
- {% include 'performance_plot.html' %}
23
- </body>
1
+ <!DOCTYPE html>
2
+ <html lang="en">
3
+ <head>
4
+ <meta charset="UTF-8">
5
+ <meta name="viewport" content="width=device-width, initial-scale=1.0">
6
+ <title>Exception Details</title>
7
+ <style>
8
+ {{ css }}
9
+ </style>
10
+
11
+ <script>
12
+ {{ javascript }}
13
+ </script>
14
+
15
+ </head>
16
+ <body>
17
+ {% include 'overview.html' %}
18
+ {% include 'exceptions_by_type.html' %}
19
+ {% include 'exceptions_by_model.html' %}
20
+ {% include 'exceptions_by_question_name.html' %}
21
+ {% include 'interviews.html' %}
22
+ {% include 'performance_plot.html' %}
23
+ </body>
24
24
  </html>
@@ -1,35 +1,35 @@
1
- <h2>Exceptions by Model</h2>
2
- <table border="1">
3
- <thead>
4
- <tr>
5
- <th>Model</th>
6
- <th>Number</th>
7
- </tr>
8
- </thead>
9
- <tbody>
10
- {% for model, exceptions in exceptions_by_model.items() %}
11
- <tr>
12
- <td>{{ model }}</td>
13
- <td>{{ exceptions }}</td>
14
- </tr>
15
- {% endfor %}
16
- </tbody>
17
- </table>
18
-
19
- <h2>Exceptions by Service</h2>
20
- <table border="1">
21
- <thead>
22
- <tr>
23
- <th>Service</th>
24
- <th>Number</th>
25
- </tr>
26
- </thead>
27
- <tbody>
28
- {% for service, exceptions in exceptions_by_service.items() %}
29
- <tr>
30
- <td>{{ service }}</td>
31
- <td>{{ exceptions }}</td>
32
- </tr>
33
- {% endfor %}
34
- </tbody>
1
+ <h2>Exceptions by Model</h2>
2
+ <table border="1">
3
+ <thead>
4
+ <tr>
5
+ <th>Model</th>
6
+ <th>Number</th>
7
+ </tr>
8
+ </thead>
9
+ <tbody>
10
+ {% for model, exceptions in exceptions_by_model.items() %}
11
+ <tr>
12
+ <td>{{ model }}</td>
13
+ <td>{{ exceptions }}</td>
14
+ </tr>
15
+ {% endfor %}
16
+ </tbody>
17
+ </table>
18
+
19
+ <h2>Exceptions by Service</h2>
20
+ <table border="1">
21
+ <thead>
22
+ <tr>
23
+ <th>Service</th>
24
+ <th>Number</th>
25
+ </tr>
26
+ </thead>
27
+ <tbody>
28
+ {% for service, exceptions in exceptions_by_service.items() %}
29
+ <tr>
30
+ <td>{{ service }}</td>
31
+ <td>{{ exceptions }}</td>
32
+ </tr>
33
+ {% endfor %}
34
+ </tbody>
35
35
  </table>
@@ -1,17 +1,17 @@
1
- <h2>Exceptions by Question Name</h2>
2
- <table border="1">
3
- <thead>
4
- <tr>
5
- <th>Question Name</th>
6
- <th>Number of Exceptions</th>
7
- </tr>
8
- </thead>
9
- <tbody>
10
- {% for question_name, exception_count in exceptions_by_question_name.items() %}
11
- <tr>
12
- <td>{{ question_name }}</td>
13
- <td>{{ exception_count }}</td>
14
- </tr>
15
- {% endfor %}
16
- </tbody>
1
+ <h2>Exceptions by Question Name</h2>
2
+ <table border="1">
3
+ <thead>
4
+ <tr>
5
+ <th>Question Name</th>
6
+ <th>Number of Exceptions</th>
7
+ </tr>
8
+ </thead>
9
+ <tbody>
10
+ {% for question_name, exception_count in exceptions_by_question_name.items() %}
11
+ <tr>
12
+ <td>{{ question_name }}</td>
13
+ <td>{{ exception_count }}</td>
14
+ </tr>
15
+ {% endfor %}
16
+ </tbody>
17
17
  </table>
@@ -1,17 +1,17 @@
1
- <h2>Exceptions by Type</h2>
2
- <table border="1">
3
- <thead>
4
- <tr>
5
- <th>Exception Type</th>
6
- <th>Number</th>
7
- </tr>
8
- </thead>
9
- <tbody>
10
- {% for exception_type, exceptions in exceptions_by_type.items() %}
11
- <tr>
12
- <td>{{ exception_type }}</td>
13
- <td>{{ exceptions }}</td>
14
- </tr>
15
- {% endfor %}
16
- </tbody>
1
+ <h2>Exceptions by Type</h2>
2
+ <table border="1">
3
+ <thead>
4
+ <tr>
5
+ <th>Exception Type</th>
6
+ <th>Number</th>
7
+ </tr>
8
+ </thead>
9
+ <tbody>
10
+ {% for exception_type, exceptions in exceptions_by_type.items() %}
11
+ <tr>
12
+ <td>{{ exception_type }}</td>
13
+ <td>{{ exceptions }}</td>
14
+ </tr>
15
+ {% endfor %}
16
+ </tbody>
17
17
  </table>
@@ -1,116 +1,116 @@
1
- <div class="question">question_name: {{ question }}</div>
2
-
3
-
4
- <h2>Exception details</h2>
5
-
6
- {% for exception_message in exceptions %}
7
- <div class="exception-detail">
8
- <div class="exception-header">
9
- <span class="exception-exception">Exception: {{ exception_message.name }}</span>
10
- <button class="toggle-btn">▼</button>
11
- </div>
12
- <div class="exception-content">
13
- <table border="1">
14
- <tr>
15
- <th>Key</th>
16
- <th>Value</th>
17
- </tr>
18
- <tr>
19
- <td>Interview ID (index in results)</td>
20
- <td>{{ index }}</td>
21
- </tr>
22
- <tr>
23
- <td>Question name (question_name)</td>
24
- <td>{{ question }}</td>
25
- </tr>
26
-
27
- <tr>
28
- <td>Question type (question_type)</td>
29
- <td>{{ exception_message.question_type }}</td>
30
- </tr>
31
-
32
- <tr>
33
- <td>Human-readable question</td>
34
- <td>{{ interview.survey.get_question(question).html(
35
- scenario = interview.scenario,
36
- agent = interview.agent,
37
- answers = exception_message.answers)
38
-
39
- }}</td>
40
- </tr>
41
- <tr>
42
- <td>Scenario</td>
43
- <td>{{ interview.scenario.__repr__() }}</td>
44
- </tr>
45
- <tr>
46
- <td>Agent</td>
47
- <td>{{ interview.agent.__repr__() }}</td>
48
- </tr>
49
- <tr>
50
- <td>Model name</td>
51
- <td>{{ interview.model.model }}</td>
52
- </tr>
53
- <tr>
54
- <td>Inference service</td>
55
- <td>{{ interview.model._inference_service_ }}</td>
56
- </tr>
57
- <tr>
58
- <td>Model parameters</td>
59
- <td>{{ interview.model.__repr__() }}</td>
60
- </tr>
61
- <tr>
62
- <td>User Prompt</td>
63
- <td><pre>{{ exception_message.rendered_prompts['user_prompt'] }}</pre></td>
64
- </tr>
65
- <tr>
66
- <td>System Prompt</td>
67
- <td><pre>{{ exception_message.rendered_prompts['system_prompt'] }}</pre></td>
68
- </tr>
69
- <tr>
70
- <td>Raw model response</td>
71
- <td><pre>{{ exception_message.raw_model_response }}</pre>
72
- </td>
73
- </tr>
74
- <tr>
75
- <td>Generated token string (at {{ exception_message.key_sequence }}) in raw response</td>
76
- <td><pre>{{ exception_message.generated_token_string }}</pre>
77
- </td>
78
- </tr>
79
- <tr>
80
- <td>Code to (likely) reproduce the error</td>
81
- <td>
82
- <textarea id="codeToCopy" rows="10" cols="90">{{ exception_message.code_to_reproduce }}</textarea>
83
- <button onclick="copyCode()">Copy</button>
84
- </td>
85
- </tr>
86
-
87
- </table>
88
-
89
-
90
- {% if exception_message.exception.__class__.__name__ == 'QuestionAnswerValidationError' %}
91
- <h3>Answer validation details</h3>
92
- <table border="1">
93
- <tr>
94
- <th>Field</th>
95
- <th>Value</th>
96
- </tr>
97
- {% for field, (explanation, open_tag, close_tag, value) in exception_message.exception.to_html_dict().items() %}
98
-
99
- <tr>
100
- <td>{{ field }}: ({{ explanation }})</td>
101
- <td><{{open_tag}}> {{ value | escape }} <{{close_tag}}></td>
102
- </tr>
103
- {% endfor %}
104
- </table>
105
- {% endif %}
106
-
107
- <div class="exception-time">Time: {{ exception_message.time }}</div>
108
- <div class="exception-traceback">Traceback:
109
- <text>
110
- <pre>{{ exception_message.traceback }}</pre>
111
- </text>
112
- </div>
113
- </div>
114
- </div>
115
-
1
+ <div class="question">question_name: {{ question }}</div>
2
+
3
+
4
+ <h2>Exception details</h2>
5
+
6
+ {% for exception_message in exceptions %}
7
+ <div class="exception-detail">
8
+ <div class="exception-header">
9
+ <span class="exception-exception">Exception: {{ exception_message.name }}</span>
10
+ <button class="toggle-btn">▼</button>
11
+ </div>
12
+ <div class="exception-content">
13
+ <table border="1">
14
+ <tr>
15
+ <th>Key</th>
16
+ <th>Value</th>
17
+ </tr>
18
+ <tr>
19
+ <td>Interview ID (index in results)</td>
20
+ <td>{{ index }}</td>
21
+ </tr>
22
+ <tr>
23
+ <td>Question name (question_name)</td>
24
+ <td>{{ question }}</td>
25
+ </tr>
26
+
27
+ <tr>
28
+ <td>Question type (question_type)</td>
29
+ <td>{{ exception_message.question_type }}</td>
30
+ </tr>
31
+
32
+ <tr>
33
+ <td>Human-readable question</td>
34
+ <td>{{ interview.survey._get_question_by_name(question).html(
35
+ scenario = interview.scenario,
36
+ agent = interview.agent,
37
+ answers = exception_message.answers)
38
+
39
+ }}</td>
40
+ </tr>
41
+ <tr>
42
+ <td>Scenario</td>
43
+ <td>{{ interview.scenario.__repr__() }}</td>
44
+ </tr>
45
+ <tr>
46
+ <td>Agent</td>
47
+ <td>{{ interview.agent.__repr__() }}</td>
48
+ </tr>
49
+ <tr>
50
+ <td>Model name</td>
51
+ <td>{{ interview.model.model }}</td>
52
+ </tr>
53
+ <tr>
54
+ <td>Inference service</td>
55
+ <td>{{ interview.model._inference_service_ }}</td>
56
+ </tr>
57
+ <tr>
58
+ <td>Model parameters</td>
59
+ <td>{{ interview.model.__repr__() }}</td>
60
+ </tr>
61
+ <tr>
62
+ <td>User Prompt</td>
63
+ <td><pre>{{ exception_message.rendered_prompts['user_prompt'] }}</pre></td>
64
+ </tr>
65
+ <tr>
66
+ <td>System Prompt</td>
67
+ <td><pre>{{ exception_message.rendered_prompts['system_prompt'] }}</pre></td>
68
+ </tr>
69
+ <tr>
70
+ <td>Raw model response</td>
71
+ <td><pre>{{ exception_message.raw_model_response }}</pre>
72
+ </td>
73
+ </tr>
74
+ <tr>
75
+ <td>Generated token string (at {{ exception_message.key_sequence }}) in raw response</td>
76
+ <td><pre>{{ exception_message.generated_token_string }}</pre>
77
+ </td>
78
+ </tr>
79
+ <tr>
80
+ <td>Code to (likely) reproduce the error</td>
81
+ <td>
82
+ <textarea id="codeToCopy" rows="10" cols="90">{{ exception_message.code_to_reproduce }}</textarea>
83
+ <button onclick="copyCode()">Copy</button>
84
+ </td>
85
+ </tr>
86
+
87
+ </table>
88
+
89
+
90
+ {% if exception_message.exception.__class__.__name__ == 'QuestionAnswerValidationError' %}
91
+ <h3>Answer validation details</h3>
92
+ <table border="1">
93
+ <tr>
94
+ <th>Field</th>
95
+ <th>Value</th>
96
+ </tr>
97
+ {% for field, (explanation, open_tag, close_tag, value) in exception_message.exception.to_html_dict().items() %}
98
+
99
+ <tr>
100
+ <td>{{ field }}: ({{ explanation }})</td>
101
+ <td><{{open_tag}}> {{ value | escape }} <{{close_tag}}></td>
102
+ </tr>
103
+ {% endfor %}
104
+ </table>
105
+ {% endif %}
106
+
107
+ <div class="exception-time">Time: {{ exception_message.time }}</div>
108
+ <div class="exception-traceback">Traceback:
109
+ <text>
110
+ <pre>{{ exception_message.traceback }}</pre>
111
+ </text>
112
+ </div>
113
+ </div>
114
+ </div>
115
+
116
116
  {% endfor %}
@@ -1,19 +1,19 @@
1
-
2
- {% if interviews|length > max_interviews %}
3
- <h1>Only showing the first {{ max_interviews }} interviews with errors</h1>
4
- {% else %}
5
- <h1>Showing all interviews</h1>
6
- {% endif %}
7
-
8
- {% for index, interview in interviews.items() %}
9
- {% if index < max_interviews %}
10
- {% if interview.exceptions != {} %}
11
- <div class="interview">Interview: {{ index }} </div>
12
- Model: {{ interview.model.model }}
13
- <h1>Failing questions</h1>
14
- {% endif %}
15
- {% for question, exceptions in interview.exceptions.items() %}
16
- {% include 'interview_details.html' %}
17
- {% endfor %}
18
- {% endif %}
19
- {% endfor %}
1
+
2
+ {% if interviews|length > max_interviews %}
3
+ <h1>Only showing the first {{ max_interviews }} interviews with errors</h1>
4
+ {% else %}
5
+ <h1>Showing all interviews</h1>
6
+ {% endif %}
7
+
8
+ {% for index, interview in interviews.items() %}
9
+ {% if index < max_interviews %}
10
+ {% if interview.exceptions != {} %}
11
+ <div class="interview">Interview: {{ index }} </div>
12
+ Model: {{ interview.model.model }}
13
+ <h1>Failing questions</h1>
14
+ {% endif %}
15
+ {% for question, exceptions in interview.exceptions.items() %}
16
+ {% include 'interview_details.html' %}
17
+ {% endfor %}
18
+ {% endif %}
19
+ {% endfor %}
@@ -1,5 +1,5 @@
1
- <h1>Overview</h1>
2
- <p>There were {{ interviews|length }} total interview(s). An 'interview' is the result of one survey, taken by one agent, with one model, with one scenario.</p>
3
- The number of interviews with any exceptions was {{ num_exceptions }}.</p>
4
- <p>For advice on dealing with exceptions on Expected Parrot,
1
+ <h1>Overview</h1>
2
+ <p>There were {{ interviews|length }} total interview(s). An 'interview' is the result of one survey, taken by one agent, with one model, with one scenario.</p>
3
+ The number of interviews with any exceptions was {{ num_exceptions }}.</p>
4
+ <p>For advice on dealing with exceptions on Expected Parrot,
5
5
  see <a href="https://docs.expectedparrot.com/en/latest/exceptions.html">here</a>.</p>
@@ -1,2 +1,2 @@
1
- <h1>Performance Plot</h1>
1
+ <h1>Performance Plot</h1>
2
2
  {{ performance_plot_html }}