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,664 +1,518 @@
1
- """This module contains the Question class, which is the base class for all questions in EDSL."""
2
-
3
- from __future__ import annotations
4
- from abc import ABC, abstractmethod
5
- from typing import Any, Type, Optional, List, Callable, Union, TypedDict
6
- import copy
7
-
8
- from edsl.exceptions import (
9
- QuestionResponseValidationError,
10
- QuestionAnswerValidationError,
11
- QuestionSerializationError,
12
- )
13
- from edsl.questions.descriptors import QuestionNameDescriptor, QuestionTextDescriptor
14
-
15
-
16
- from edsl.questions.AnswerValidatorMixin import AnswerValidatorMixin
17
- from edsl.questions.RegisterQuestionsMeta import RegisterQuestionsMeta
18
- from edsl.Base import PersistenceMixin, RichPrintingMixin
19
- from edsl.BaseDiff import BaseDiff, BaseDiffCollection
20
-
21
- from edsl.questions.SimpleAskMixin import SimpleAskMixin
22
- from edsl.questions.QuestionBasePromptsMixin import QuestionBasePromptsMixin
23
- from edsl.questions.QuestionBaseGenMixin import QuestionBaseGenMixin
24
- from edsl.utilities.decorators import add_edsl_version, remove_edsl_version
25
-
26
-
27
- class QuestionBase(
28
- PersistenceMixin,
29
- RichPrintingMixin,
30
- SimpleAskMixin,
31
- QuestionBasePromptsMixin,
32
- QuestionBaseGenMixin,
33
- ABC,
34
- AnswerValidatorMixin,
35
- metaclass=RegisterQuestionsMeta,
36
- ):
37
- """ABC for the Question class. All questions inherit from this class.
38
- Some of the constraints on child questions are defined in the RegisterQuestionsMeta metaclass.
39
- """
40
-
41
- question_name: str = QuestionNameDescriptor()
42
- question_text: str = QuestionTextDescriptor()
43
-
44
- _answering_instructions = None
45
- _question_presentation = None
46
-
47
- @property
48
- def response_model(self) -> type["BaseModel"]:
49
- if self._response_model is not None:
50
- return self._response_model
51
- else:
52
- return self.create_response_model()
53
-
54
- # region: Validation and simulation methods
55
- @property
56
- def response_validator(self) -> "ResponseValidatorBase":
57
- """Return the response validator."""
58
- params = (
59
- {
60
- "response_model": self.response_model,
61
- }
62
- | {k: getattr(self, k) for k in self.validator_parameters}
63
- | {"exception_to_throw": getattr(self, "exception_to_throw", None)}
64
- | {"override_answer": getattr(self, "override_answer", None)}
65
- )
66
- return self.response_validator_class(**params)
67
-
68
- @property
69
- def validator_parameters(self) -> list[str]:
70
- """Return the parameters required for the response validator.
71
-
72
- >>> from edsl import QuestionMultipleChoice as Q
73
- >>> Q.example().validator_parameters
74
- ['question_options', 'use_code']
75
-
76
- """
77
- return self.response_validator_class.required_params
78
-
79
- @property
80
- def fake_data_factory(self):
81
- """Return the fake data factory."""
82
- if not hasattr(self, "_fake_data_factory"):
83
- from polyfactory.factories.pydantic_factory import ModelFactory
84
-
85
- class FakeData(ModelFactory[self.response_model]):
86
- ...
87
-
88
- self._fake_data_factory = FakeData
89
- return self._fake_data_factory
90
-
91
- def _simulate_answer(self, human_readable: bool = False) -> dict:
92
- """Simulate a valid answer for debugging purposes (what the validator expects).
93
- >>> from edsl import QuestionFreeText as Q
94
- >>> Q.example()._simulate_answer()
95
- {'answer': '...', 'generated_tokens': ...}
96
- """
97
- simulated_answer = self.fake_data_factory.build().dict()
98
- if human_readable and hasattr(self, "question_options") and self.use_code:
99
- simulated_answer["answer"] = [
100
- self.question_options[index] for index in simulated_answer["answer"]
101
- ]
102
- return simulated_answer
103
-
104
- class ValidatedAnswer(TypedDict):
105
- answer: Any
106
- comment: Optional[str]
107
- generated_tokens: Optional[str]
108
-
109
- def _validate_answer(
110
- self, answer: dict, replacement_dict: dict = None
111
- ) -> ValidatedAnswer:
112
- """Validate the answer.
113
- >>> from edsl.exceptions import QuestionAnswerValidationError
114
- >>> from edsl import QuestionFreeText as Q
115
- >>> Q.example()._validate_answer({'answer': 'Hello', 'generated_tokens': 'Hello'})
116
- {'answer': 'Hello', 'generated_tokens': 'Hello'}
117
- """
118
-
119
- return self.response_validator.validate(answer, replacement_dict)
120
-
121
- # endregion
122
-
123
- # region: Serialization methods
124
- @property
125
- def name(self) -> str:
126
- "Helper function so questions and instructions can use the same access method"
127
- return self.question_name
128
-
129
- def __hash__(self) -> int:
130
- """Return a hash of the question.
131
-
132
- >>> from edsl import QuestionFreeText as Q
133
- >>> hash(Q.example())
134
- 1144312636257752766
135
- """
136
- from edsl.utilities.utilities import dict_hash
137
-
138
- return dict_hash(self.to_dict(add_edsl_version=False))
139
-
140
- @property
141
- def data(self) -> dict:
142
- """Return a dictionary of question attributes **except** for question_type.
143
-
144
- >>> from edsl import QuestionFreeText as Q
145
- >>> Q.example().data
146
- {'question_name': 'how_are_you', 'question_text': 'How are you?'}
147
- """
148
- exclude_list = [
149
- "question_type",
150
- # "_include_comment",
151
- "_fake_data_factory",
152
- # "_use_code",
153
- "_model_instructions",
154
- ]
155
- only_if_not_na_list = ["_answering_instructions", "_question_presentation"]
156
-
157
- only_if_not_default_list = {"_include_comment": True, "_use_code": False}
158
-
159
- def ok(key, value):
160
- if not key.startswith("_"):
161
- return False
162
- if key in exclude_list:
163
- return False
164
- if key in only_if_not_na_list and value is None:
165
- return False
166
- if (
167
- key in only_if_not_default_list
168
- and value == only_if_not_default_list[key]
169
- ):
170
- return False
171
-
172
- return True
173
-
174
- candidate_data = {
175
- k.replace("_", "", 1): v for k, v in self.__dict__.items() if ok(k, v)
176
- }
177
-
178
- if "func" in candidate_data:
179
- func = candidate_data.pop("func")
180
- import inspect
181
-
182
- candidate_data["function_source_code"] = inspect.getsource(func)
183
-
184
- return candidate_data
185
-
186
- def to_dict(self, add_edsl_version=True):
187
- """Convert the question to a dictionary that includes the question type (used in deserialization).
188
-
189
- >>> from edsl import QuestionFreeText as Q; Q.example().to_dict(add_edsl_version = False)
190
- {'question_name': 'how_are_you', 'question_text': 'How are you?', 'question_type': 'free_text'}
191
- """
192
- candidate_data = self.data.copy()
193
- candidate_data["question_type"] = self.question_type
194
- d = {key: value for key, value in candidate_data.items() if value is not None}
195
- if add_edsl_version:
196
- from edsl import __version__
197
-
198
- d["edsl_version"] = __version__
199
- d["edsl_class_name"] = "QuestionBase"
200
-
201
- return d
202
-
203
- @classmethod
204
- @remove_edsl_version
205
- def from_dict(cls, data: dict) -> Type[QuestionBase]:
206
- """Construct a question object from a dictionary created by that question's `to_dict` method."""
207
- local_data = data.copy()
208
-
209
- try:
210
- question_type = local_data.pop("question_type")
211
- if question_type == "linear_scale":
212
- # This is a fix for issue https://github.com/expectedparrot/edsl/issues/165
213
- options_labels = local_data.get("option_labels", None)
214
- if options_labels:
215
- options_labels = {
216
- int(key): value for key, value in options_labels.items()
217
- }
218
- local_data["option_labels"] = options_labels
219
- except:
220
- raise QuestionSerializationError(
221
- f"Data does not have a 'question_type' field (got {data})."
222
- )
223
- from edsl.questions.question_registry import get_question_class
224
-
225
- try:
226
- question_class = get_question_class(question_type)
227
- except ValueError:
228
- raise QuestionSerializationError(
229
- f"No question registered with question_type {question_type}"
230
- )
231
-
232
- if "model_instructions" in local_data:
233
- model_instructions = local_data.pop("model_instructions")
234
- new_q = question_class(**local_data)
235
- new_q.model_instructions = model_instructions
236
- return new_q
237
-
238
- return question_class(**local_data)
239
-
240
- # endregion
241
-
242
- # region: Running methods
243
- @classmethod
244
- def _get_test_model(self, canned_response: Optional[str] = None) -> "LanguageModel":
245
- """Get a test model for the question."""
246
- from edsl.language_models import LanguageModel
247
-
248
- return LanguageModel.example(canned_response=canned_response, test_model=True)
249
-
250
- @classmethod
251
- def run_example(
252
- cls,
253
- show_answer: bool = True,
254
- model: Optional["LanguageModel"] = None,
255
- cache=False,
256
- disable_remote_cache: bool = False,
257
- disable_remote_inference: bool = False,
258
- **kwargs,
259
- ):
260
- """Run an example of the question.
261
- >>> from edsl.language_models import LanguageModel
262
- >>> from edsl import QuestionFreeText as Q
263
- >>> m = Q._get_test_model(canned_response = "Yo, what's up?")
264
- >>> m.execute_model_call("", "")
265
- {'message': [{'text': "Yo, what's up?"}], 'usage': {'prompt_tokens': 1, 'completion_tokens': 1}}
266
- >>> Q.run_example(show_answer = True, model = m, disable_remote_cache = True, disable_remote_inference = True)
267
- answer.how_are_you
268
- --------------------
269
- Yo, what's up?
270
- """
271
- if model is None:
272
- from edsl import Model
273
-
274
- model = Model()
275
- results = (
276
- cls.example(**kwargs)
277
- .by(model)
278
- .run(
279
- cache=cache,
280
- disable_remote_cache=disable_remote_cache,
281
- disable_remote_inference=disable_remote_inference,
282
- )
283
- )
284
- if show_answer:
285
- return results.select("answer.*").print()
286
- else:
287
- return results
288
-
289
- def __call__(
290
- self,
291
- just_answer=True,
292
- model=None,
293
- agent=None,
294
- disable_remote_cache: bool = False,
295
- disable_remote_inference: bool = False,
296
- **kwargs,
297
- ):
298
- """Call the question.
299
-
300
-
301
- >>> from edsl import QuestionFreeText as Q
302
- >>> m = Q._get_test_model(canned_response = "Yo, what's up?")
303
- >>> q = Q(question_name = "color", question_text = "What is your favorite color?")
304
- >>> q(model = m, disable_remote_cache = True, disable_remote_inference = True)
305
- "Yo, what's up?"
306
-
307
- """
308
- survey = self.to_survey()
309
- results = survey(
310
- model=model,
311
- agent=agent,
312
- **kwargs,
313
- cache=False,
314
- disable_remote_cache=disable_remote_cache,
315
- disable_remote_inference=disable_remote_inference,
316
- )
317
- if just_answer:
318
- return results.select(f"answer.{self.question_name}").first()
319
- else:
320
- return results
321
-
322
- def run(self, *args, **kwargs) -> "Results":
323
- """Turn a single question into a survey and runs it."""
324
- from edsl.surveys.Survey import Survey
325
-
326
- s = self.to_survey()
327
- return s.run(*args, **kwargs)
328
-
329
- async def run_async(
330
- self,
331
- just_answer: bool = True,
332
- model: Optional["Model"] = None,
333
- agent: Optional["Agent"] = None,
334
- disable_remote_inference: bool = False,
335
- **kwargs,
336
- ) -> Union[Any, "Results"]:
337
- """Call the question asynchronously.
338
-
339
- >>> import asyncio
340
- >>> from edsl import QuestionFreeText as Q
341
- >>> m = Q._get_test_model(canned_response = "Blue")
342
- >>> q = Q(question_name = "color", question_text = "What is your favorite color?")
343
- >>> async def test_run_async(): result = await q.run_async(model=m, disable_remote_inference = True); print(result)
344
- >>> asyncio.run(test_run_async())
345
- Blue
346
- """
347
- survey = self.to_survey()
348
- results = await survey.run_async(
349
- model=model,
350
- agent=agent,
351
- disable_remote_inference=disable_remote_inference,
352
- **kwargs,
353
- )
354
- if just_answer:
355
- return results.select(f"answer.{self.question_name}").first()
356
- else:
357
- return results
358
-
359
- # endregion
360
-
361
- # region: Magic methods
362
- def _repr_html_(self):
363
- # from edsl.utilities.utilities import data_to_html
364
-
365
- data = self.to_dict(add_edsl_version=False)
366
- # keys = list(data.keys())
367
- # values = list(data.values())
368
- from tabulate import tabulate
369
-
370
- return tabulate(data.items(), headers=["keys", "values"], tablefmt="html")
371
-
372
- # try:
373
- # _ = data.pop("edsl_version")
374
- # _ = data.pop("edsl_class_name")
375
- # except KeyError:
376
- # print("Serialized question lacks edsl version, but is should have it.")
377
-
378
- # return data_to_html(data)
379
-
380
- def __getitem__(self, key: str) -> Any:
381
- """Get an attribute of the question so it can be treated like a dictionary.
382
-
383
- >>> from edsl.questions import QuestionFreeText as Q
384
- >>> Q.example()['question_text']
385
- 'How are you?'
386
- """
387
- return getattr(self, key)
388
-
389
- def __repr__(self) -> str:
390
- """Return a string representation of the question. Should be able to be used to reconstruct the question.
391
-
392
- >>> from edsl import QuestionFreeText as Q
393
- >>> repr(Q.example())
394
- 'Question(\\'free_text\\', question_name = \"""how_are_you\""", question_text = \"""How are you?\""")'
395
- """
396
- items = [
397
- f'{k} = """{v}"""' if isinstance(v, str) else f"{k} = {v}"
398
- for k, v in self.data.items()
399
- if k != "question_type"
400
- ]
401
- question_type = self.to_dict().get("question_type", "None")
402
- return f"Question('{question_type}', {', '.join(items)})"
403
-
404
- def __eq__(self, other: Union[Any, Type[QuestionBase]]) -> bool:
405
- """Check if two questions are equal. Equality is defined as having the .to_dict().
406
-
407
- >>> from edsl import QuestionFreeText as Q
408
- >>> q1 = Q.example()
409
- >>> q2 = Q.example()
410
- >>> q1 == q2
411
- True
412
- >>> q1.question_text = "How are you John?"
413
- >>> q1 == q2
414
- False
415
-
416
- """
417
- if not isinstance(other, QuestionBase):
418
- return False
419
- return self.to_dict() == other.to_dict()
420
-
421
- def __sub__(self, other) -> BaseDiff:
422
- """Return the difference between two objects.
423
- >>> from edsl import QuestionFreeText as Q
424
- >>> q1 = Q.example()
425
- >>> q2 = q1.copy()
426
- >>> q2.question_text = "How are you John?"
427
- >>> diff = q1 - q2
428
- """
429
-
430
- return BaseDiff(other, self)
431
-
432
- # TODO: Throws an error that should be addressed at QuestionFunctional
433
- def __add__(self, other_question_or_diff):
434
- """
435
- Compose two questions into a single question.
436
-
437
- TODO: Probably getting deprecated.
438
-
439
- """
440
- if isinstance(other_question_or_diff, BaseDiff) or isinstance(
441
- other_question_or_diff, BaseDiffCollection
442
- ):
443
- return other_question_or_diff.apply(self)
444
-
445
- # from edsl.questions import compose_questions
446
- # return compose_questions(self, other_question_or_diff)
447
-
448
- # def _validate_response(self, response):
449
- # """Validate the response from the LLM. Behavior depends on the question type."""
450
- # if "answer" not in response:
451
- # raise QuestionResponseValidationError(
452
- # "Response from LLM does not have an answer"
453
- # )
454
- # return response
455
-
456
- def _translate_answer_code_to_answer(
457
- self, answer, scenario: Optional["Scenario"] = None
458
- ):
459
- """There is over-ridden by child classes that ask for codes."""
460
- return answer
461
-
462
- # endregion
463
-
464
- # region: Forward methods
465
- def add_question(self, other: QuestionBase) -> "Survey":
466
- """Add a question to this question by turning them into a survey with two questions.
467
-
468
- >>> from edsl.questions import QuestionFreeText as Q
469
- >>> from edsl.questions import QuestionMultipleChoice as QMC
470
- >>> s = Q.example().add_question(QMC.example())
471
- >>> len(s.questions)
472
- 2
473
- """
474
- from edsl.surveys.Survey import Survey
475
-
476
- s = Survey([self, other])
477
- return s
478
-
479
- def to_survey(self) -> "Survey":
480
- """Turn a single question into a survey.
481
- >>> from edsl import QuestionFreeText as Q
482
- >>> Q.example().to_survey().questions[0].question_name
483
- 'how_are_you'
484
- """
485
- from edsl.surveys.Survey import Survey
486
-
487
- s = Survey([self])
488
- return s
489
-
490
- def by(self, *args) -> "Jobs":
491
- """Turn a single question into a survey and then a Job."""
492
- from edsl.surveys.Survey import Survey
493
-
494
- s = Survey([self])
495
- return s.by(*args)
496
-
497
- # endregion
498
-
499
- # region: Display methods
500
- def print(self):
501
- from rich import print_json
502
- import json
503
-
504
- print_json(json.dumps(self.to_dict()))
505
-
506
- def human_readable(self) -> str:
507
- """Print the question in a human readable format.
508
-
509
- >>> from edsl.questions import QuestionFreeText
510
- >>> QuestionFreeText.example().human_readable()
511
- 'Question Type: free_text\\nQuestion: How are you?'
512
- """
513
- lines = []
514
- lines.append(f"Question Type: {self.question_type}")
515
- lines.append(f"Question: {self.question_text}")
516
- if hasattr(self, "question_options"):
517
- lines.append("Please name the option you choose from the following.:")
518
- for index, option in enumerate(self.question_options):
519
- lines.append(f"{option}")
520
- return "\n".join(lines)
521
-
522
- def html(
523
- self,
524
- scenario: Optional[dict] = None,
525
- agent: Optional[dict] = {},
526
- answers: Optional[dict] = None,
527
- include_question_name: bool = False,
528
- height: Optional[int] = None,
529
- width: Optional[int] = None,
530
- iframe=False,
531
- ):
532
- """Return the question in HTML format."""
533
- from jinja2 import Template
534
-
535
- if scenario is None:
536
- scenario = {}
537
-
538
- prior_answers_dict = {}
539
-
540
- if isinstance(answers, dict):
541
- for key, value in answers.items():
542
- if not key.endswith("_comment") and not key.endswith(
543
- "_generated_tokens"
544
- ):
545
- prior_answers_dict[key] = {"answer": value}
546
-
547
- # breakpoint()
548
-
549
- base_template = """
550
- <div id="{{ question_name }}" class="survey_question" data-type="{{ question_type }}">
551
- {% if include_question_name %}
552
- <p>question_name: {{ question_name }}</p>
553
- {% endif %}
554
- <p class="question_text">{{ question_text }}</p>
555
- {{ question_content }}
556
- </div>
557
- """
558
- if not hasattr(self, "question_type"):
559
- self.question_type = "unknown"
560
-
561
- if hasattr(self, "question_html_content"):
562
- question_content = self.question_html_content
563
- else:
564
- question_content = Template("")
565
-
566
- base_template = Template(base_template)
567
-
568
- context = {
569
- "scenario": scenario,
570
- "agent": agent,
571
- } | prior_answers_dict
572
-
573
- # Render the question text
574
- try:
575
- question_text = Template(self.question_text).render(context)
576
- except Exception as e:
577
- print(
578
- f"Error rendering question: question_text = {self.question_text}, error = {e}"
579
- )
580
- question_text = self.question_text
581
-
582
- try:
583
- question_content = Template(question_content).render(context)
584
- except Exception as e:
585
- print(
586
- f"Error rendering question: question_content = {question_content}, error = {e}"
587
- )
588
- question_content = question_content
589
-
590
- try:
591
- params = {
592
- "question_name": self.question_name,
593
- "question_text": question_text,
594
- "question_type": self.question_type,
595
- "question_content": question_content,
596
- "include_question_name": include_question_name,
597
- }
598
- except Exception as e:
599
- raise ValueError(
600
- f"Error rendering question: params = {params}, error = {e}"
601
- )
602
- rendered_html = base_template.render(**params)
603
-
604
- if iframe:
605
- import html
606
- from IPython.display import display, HTML
607
-
608
- height = height or 200
609
- width = width or 600
610
- escaped_output = html.escape(rendered_html)
611
- # escaped_output = rendered_html
612
- iframe = f""""
613
- <iframe srcdoc="{ escaped_output }" style="width: {width}px; height: {height}px;"></iframe>
614
- """
615
- display(HTML(iframe))
616
- return None
617
-
618
- return rendered_html
619
-
620
- @classmethod
621
- def example_model(cls):
622
- from edsl import Model
623
-
624
- q = cls.example()
625
- m = Model("test", canned_response=cls._simulate_answer(q)["answer"])
626
-
627
- return m
628
-
629
- @classmethod
630
- def example_results(cls):
631
- m = cls.example_model()
632
- q = cls.example()
633
- return q.by(m).run(cache=False)
634
-
635
- def rich_print(self):
636
- """Print the question in a rich format."""
637
- from rich.table import Table
638
-
639
- table = Table(show_header=True, header_style="bold magenta")
640
- table.add_column("Question Name", style="dim")
641
- table.add_column("Question Type")
642
- table.add_column("Question Text")
643
- table.add_column("Options")
644
-
645
- question = self
646
- if hasattr(question, "question_options"):
647
- options = ", ".join([str(o) for o in question.question_options])
648
- else:
649
- options = "None"
650
- table.add_row(
651
- question.question_name,
652
- question.question_type,
653
- question.question_text,
654
- options,
655
- )
656
- return table
657
-
658
- # endregion
659
-
660
-
661
- if __name__ == "__main__":
662
- import doctest
663
-
664
- doctest.testmod(optionflags=doctest.ELLIPSIS)
1
+ """This module contains the Question class, which is the base class for all questions in EDSL."""
2
+
3
+ from __future__ import annotations
4
+ from abc import ABC, abstractmethod
5
+ from typing import Any, Type, Optional, List, Callable, Union, TypedDict, TYPE_CHECKING
6
+
7
+ from edsl.exceptions.questions import (
8
+ QuestionSerializationError,
9
+ )
10
+ from edsl.questions.descriptors import QuestionNameDescriptor, QuestionTextDescriptor
11
+
12
+ from edsl.questions.answer_validator_mixin import AnswerValidatorMixin
13
+ from edsl.questions.register_questions_meta import RegisterQuestionsMeta
14
+ from edsl.Base import PersistenceMixin, RepresentationMixin
15
+ from edsl.BaseDiff import BaseDiff, BaseDiffCollection
16
+
17
+ from edsl.questions.SimpleAskMixin import SimpleAskMixin
18
+ from edsl.questions.QuestionBasePromptsMixin import QuestionBasePromptsMixin
19
+ from edsl.questions.question_base_gen_mixin import QuestionBaseGenMixin
20
+ from edsl.utilities.remove_edsl_version import remove_edsl_version
21
+
22
+ if TYPE_CHECKING:
23
+ from edsl.questions.response_validator_abc import ResponseValidatorABC
24
+ from edsl.language_models.LanguageModel import LanguageModel
25
+ from edsl.results.Results import Results
26
+ from edsl.agents.Agent import Agent
27
+ from edsl.surveys.Survey import Survey
28
+ from edsl.jobs.Jobs import Jobs
29
+
30
+
31
+ class QuestionBase(
32
+ PersistenceMixin,
33
+ RepresentationMixin,
34
+ SimpleAskMixin,
35
+ QuestionBasePromptsMixin,
36
+ QuestionBaseGenMixin,
37
+ ABC,
38
+ AnswerValidatorMixin,
39
+ metaclass=RegisterQuestionsMeta,
40
+ ):
41
+ """ABC for the Question class. All questions inherit from this class.
42
+ Some of the constraints on child questions are defined in the RegisterQuestionsMeta metaclass.
43
+
44
+
45
+ Every child class wiill have class attributes of question_type, _response_model and response_validator_class e.g.,
46
+
47
+ question_type = "free_text"
48
+ _response_model = FreeTextResponse
49
+ response_validator_class = FreeTextResponseValidator
50
+
51
+ """
52
+
53
+ question_name: str = QuestionNameDescriptor()
54
+ question_text: str = QuestionTextDescriptor()
55
+
56
+ _answering_instructions = None
57
+ _question_presentation = None
58
+
59
+ @property
60
+ def response_validator(self) -> "ResponseValidatorABC":
61
+ """Return the response validator."""
62
+ from edsl.questions.response_validator_factory import ResponseValidatorFactory
63
+
64
+ rvf = ResponseValidatorFactory(self)
65
+ return rvf.response_validator
66
+
67
+ def duplicate(self):
68
+ """Return a duplicate of the question."""
69
+ return self.from_dict(self.to_dict())
70
+
71
+ @property
72
+ def fake_data_factory(self):
73
+ """Return the fake data factory."""
74
+ if not hasattr(self, "_fake_data_factory"):
75
+ from polyfactory.factories.pydantic_factory import ModelFactory
76
+
77
+ class FakeData(ModelFactory[self.response_model]): ...
78
+
79
+ self._fake_data_factory = FakeData
80
+ return self._fake_data_factory
81
+
82
+ def _simulate_answer(self, human_readable: bool = False) -> dict:
83
+ """Simulate a valid answer for debugging purposes (what the validator expects).
84
+ >>> from edsl import QuestionFreeText as Q
85
+ >>> Q.example()._simulate_answer()
86
+ {'answer': '...', 'generated_tokens': ...}
87
+ """
88
+ simulated_answer = self.fake_data_factory.build().dict()
89
+ if human_readable and hasattr(self, "question_options") and self.use_code:
90
+ simulated_answer["answer"] = [
91
+ self.question_options[index] for index in simulated_answer["answer"]
92
+ ]
93
+ return simulated_answer
94
+
95
+ class ValidatedAnswer(TypedDict):
96
+ answer: Any
97
+ comment: Optional[str]
98
+ generated_tokens: Optional[str]
99
+
100
+ def _validate_answer(
101
+ self, answer: dict, replacement_dict: dict = None
102
+ ) -> ValidatedAnswer:
103
+ """Validate the answer.
104
+ >>> from edsl.exceptions.questions import QuestionAnswerValidationError
105
+ >>> from edsl.questions import QuestionFreeText as Q
106
+ >>> Q.example()._validate_answer({'answer': 'Hello', 'generated_tokens': 'Hello'})
107
+ {'answer': 'Hello', 'generated_tokens': 'Hello'}
108
+ """
109
+
110
+ return self.response_validator.validate(answer, replacement_dict)
111
+
112
+ @property
113
+ def name(self) -> str:
114
+ "Helper function so questions and instructions can use the same access method"
115
+ return self.question_name
116
+
117
+ def __hash__(self) -> int:
118
+ """Return a hash of the question.
119
+
120
+ >>> from edsl import QuestionFreeText as Q
121
+ >>> hash(Q.example())
122
+ 1144312636257752766
123
+ """
124
+ from edsl.utilities.utilities import dict_hash
125
+
126
+ return dict_hash(self.to_dict(add_edsl_version=False))
127
+
128
+ @property
129
+ def data(self) -> dict:
130
+ """Return a dictionary of question attributes **except** for question_type.
131
+
132
+ >>> from edsl.questions import QuestionFreeText as Q
133
+ >>> Q.example().data
134
+ {'question_name': 'how_are_you', 'question_text': 'How are you?'}
135
+ """
136
+ exclude_list = [
137
+ "question_type",
138
+ # "_include_comment",
139
+ "_fake_data_factory",
140
+ # "_use_code",
141
+ "_model_instructions",
142
+ ]
143
+ only_if_not_na_list = ["_answering_instructions", "_question_presentation"]
144
+
145
+ only_if_not_default_list = {"_include_comment": True, "_use_code": False}
146
+
147
+ def ok(key, value):
148
+ if not key.startswith("_"):
149
+ return False
150
+ if key in exclude_list:
151
+ return False
152
+ if key in only_if_not_na_list and value is None:
153
+ return False
154
+ if (
155
+ key in only_if_not_default_list
156
+ and value == only_if_not_default_list[key]
157
+ ):
158
+ return False
159
+
160
+ return True
161
+
162
+ candidate_data = {
163
+ k.replace("_", "", 1): v for k, v in self.__dict__.items() if ok(k, v)
164
+ }
165
+
166
+ if "func" in candidate_data:
167
+ func = candidate_data.pop("func")
168
+ import inspect
169
+
170
+ candidate_data["function_source_code"] = inspect.getsource(func)
171
+
172
+ return candidate_data
173
+
174
+ def to_dict(self, add_edsl_version: bool = True):
175
+ """Convert the question to a dictionary that includes the question type (used in deserialization).
176
+
177
+ >>> from edsl.questions import QuestionFreeText as Q; Q.example().to_dict(add_edsl_version = False)
178
+ {'question_name': 'how_are_you', 'question_text': 'How are you?', 'question_type': 'free_text'}
179
+ """
180
+ candidate_data = self.data.copy()
181
+ candidate_data["question_type"] = self.question_type
182
+ d = {key: value for key, value in candidate_data.items() if value is not None}
183
+ if add_edsl_version:
184
+ from edsl import __version__
185
+
186
+ d["edsl_version"] = __version__
187
+ d["edsl_class_name"] = "QuestionBase"
188
+
189
+ return d
190
+
191
+ @classmethod
192
+ @remove_edsl_version
193
+ def from_dict(cls, data: dict) -> Type[QuestionBase]:
194
+ """Construct a question object from a dictionary created by that question's `to_dict` method."""
195
+ local_data = data.copy()
196
+
197
+ try:
198
+ question_type = local_data.pop("question_type")
199
+ if question_type == "linear_scale":
200
+ # This is a fix for issue https://github.com/expectedparrot/edsl/issues/165
201
+ options_labels = local_data.get("option_labels", None)
202
+ if options_labels:
203
+ options_labels = {
204
+ int(key): value for key, value in options_labels.items()
205
+ }
206
+ local_data["option_labels"] = options_labels
207
+ except:
208
+ raise QuestionSerializationError(
209
+ f"Data does not have a 'question_type' field (got {data})."
210
+ )
211
+ from edsl.questions.question_registry import get_question_class
212
+
213
+ try:
214
+ question_class = get_question_class(question_type)
215
+ except ValueError:
216
+ raise QuestionSerializationError(
217
+ f"No question registered with question_type {question_type}"
218
+ )
219
+
220
+ if "model_instructions" in local_data:
221
+ model_instructions = local_data.pop("model_instructions")
222
+ new_q = question_class(**local_data)
223
+ new_q.model_instructions = model_instructions
224
+ return new_q
225
+
226
+ return question_class(**local_data)
227
+
228
+ @classmethod
229
+ def _get_test_model(self, canned_response: Optional[str] = None) -> "LanguageModel":
230
+ """Get a test model for the question."""
231
+ from edsl.language_models import LanguageModel
232
+
233
+ return LanguageModel.example(canned_response=canned_response, test_model=True)
234
+
235
+ @classmethod
236
+ def run_example(
237
+ cls,
238
+ show_answer: bool = True,
239
+ model: Optional["LanguageModel"] = None,
240
+ cache=False,
241
+ disable_remote_cache: bool = False,
242
+ disable_remote_inference: bool = False,
243
+ **kwargs,
244
+ ):
245
+ """Run an example of the question.
246
+ >>> from edsl.language_models import LanguageModel
247
+ >>> from edsl import QuestionFreeText as Q
248
+ >>> m = Q._get_test_model(canned_response = "Yo, what's up?")
249
+ >>> m.execute_model_call("", "")
250
+ {'message': [{'text': "Yo, what's up?"}], 'usage': {'prompt_tokens': 1, 'completion_tokens': 1}}
251
+ >>> Q.run_example(show_answer = True, model = m, disable_remote_cache = True, disable_remote_inference = True)
252
+ Dataset([{'answer.how_are_you': ["Yo, what's up?"]}])
253
+ """
254
+ if model is None:
255
+ from edsl.language_models.model import Model
256
+
257
+ model = Model()
258
+ results = (
259
+ cls.example(**kwargs)
260
+ .by(model)
261
+ .run(
262
+ cache=cache,
263
+ disable_remote_cache=disable_remote_cache,
264
+ disable_remote_inference=disable_remote_inference,
265
+ )
266
+ )
267
+ if show_answer:
268
+ return results.select("answer.*")
269
+ else:
270
+ return results
271
+
272
+ def __call__(
273
+ self,
274
+ just_answer: bool = True,
275
+ model: Optional["LanguageModel"] = None,
276
+ agent: Optional["Agent"] = None,
277
+ disable_remote_cache: bool = False,
278
+ disable_remote_inference: bool = False,
279
+ verbose: bool = False,
280
+ **kwargs,
281
+ ) -> Union[Any, "Results"]:
282
+ """Call the question.
283
+
284
+
285
+ >>> from edsl import QuestionFreeText as Q
286
+ >>> from edsl import Model
287
+ >>> m = Model("test", canned_response = "Yo, what's up?")
288
+ >>> q = Q(question_name = "color", question_text = "What is your favorite color?")
289
+ >>> q(model = m, disable_remote_cache = True, disable_remote_inference = True, cache = False)
290
+ "Yo, what's up?"
291
+
292
+ """
293
+ survey = self.to_survey()
294
+ results = survey(
295
+ model=model,
296
+ agent=agent,
297
+ **kwargs,
298
+ verbose=verbose,
299
+ disable_remote_cache=disable_remote_cache,
300
+ disable_remote_inference=disable_remote_inference,
301
+ )
302
+ if just_answer:
303
+ return results.select(f"answer.{self.question_name}").first()
304
+ else:
305
+ return results
306
+
307
+ def run(self, *args, **kwargs) -> "Results":
308
+ """Turn a single question into a survey and runs it."""
309
+ return self.to_survey().run(*args, **kwargs)
310
+
311
+ def using(self, *args, **kwargs) -> "Jobs":
312
+ """Turn a single question into a survey and then a Job."""
313
+ return self.to_survey().to_jobs().using(*args, **kwargs)
314
+
315
+ async def run_async(
316
+ self,
317
+ just_answer: bool = True,
318
+ model: Optional["LanguageModel"] = None,
319
+ agent: Optional["Agent"] = None,
320
+ disable_remote_inference: bool = False,
321
+ **kwargs,
322
+ ) -> Union[Any, "Results"]:
323
+ """Call the question asynchronously.
324
+
325
+ >>> import asyncio
326
+ >>> from edsl.questions import QuestionFreeText as Q
327
+ >>> m = Q._get_test_model(canned_response = "Blue")
328
+ >>> q = Q(question_name = "color", question_text = "What is your favorite color?")
329
+ >>> async def test_run_async(): result = await q.run_async(model=m, disable_remote_inference = True, disable_remote_cache = True); print(result)
330
+ >>> asyncio.run(test_run_async())
331
+ Blue
332
+ """
333
+ survey = self.to_survey()
334
+ results = await survey.run_async(
335
+ model=model,
336
+ agent=agent,
337
+ disable_remote_inference=disable_remote_inference,
338
+ **kwargs,
339
+ )
340
+ if just_answer:
341
+ return results.select(f"answer.{self.question_name}").first()
342
+ else:
343
+ return results
344
+
345
+ def __getitem__(self, key: str) -> Any:
346
+ """Get an attribute of the question so it can be treated like a dictionary.
347
+
348
+ >>> from edsl.questions import QuestionFreeText as Q
349
+ >>> Q.example()['question_text']
350
+ 'How are you?'
351
+ """
352
+ try:
353
+ return getattr(self, key)
354
+ except TypeError:
355
+ raise KeyError(f"Question has no attribute {key} of type {type(key)}")
356
+
357
+ def __repr__(self) -> str:
358
+ """Return a string representation of the question. Should be able to be used to reconstruct the question.
359
+
360
+ >>> from edsl import QuestionFreeText as Q
361
+ >>> repr(Q.example())
362
+ 'Question(\\'free_text\\', question_name = \"""how_are_you\""", question_text = \"""How are you?\""")'
363
+ """
364
+ items = [
365
+ f'{k} = """{v}"""' if isinstance(v, str) else f"{k} = {v}"
366
+ for k, v in self.data.items()
367
+ if k != "question_type"
368
+ ]
369
+ question_type = self.to_dict().get("question_type", "None")
370
+ return f"Question('{question_type}', {', '.join(items)})"
371
+
372
+ def __eq__(self, other: Union[Any, Type[QuestionBase]]) -> bool:
373
+ """Check if two questions are equal. Equality is defined as having the .to_dict().
374
+
375
+ >>> from edsl import QuestionFreeText as Q
376
+ >>> q1 = Q.example()
377
+ >>> q2 = Q.example()
378
+ >>> q1 == q2
379
+ True
380
+ >>> q1.question_text = "How are you John?"
381
+ >>> q1 == q2
382
+ False
383
+
384
+ """
385
+ return hash(self) == hash(other)
386
+
387
+ def __sub__(self, other) -> BaseDiff:
388
+ """Return the difference between two objects.
389
+ >>> from edsl import QuestionFreeText as Q
390
+ >>> q1 = Q.example()
391
+ >>> q2 = q1.copy()
392
+ >>> q2.question_text = "How are you John?"
393
+ >>> diff = q1 - q2
394
+ """
395
+
396
+ return BaseDiff(other, self)
397
+
398
+ # TODO: Throws an error that should be addressed at QuestionFunctional
399
+ def __add__(self, other_question_or_diff):
400
+ """
401
+ Compose two questions into a single question.
402
+ """
403
+ if isinstance(other_question_or_diff, BaseDiff) or isinstance(
404
+ other_question_or_diff, BaseDiffCollection
405
+ ):
406
+ return other_question_or_diff.apply(self)
407
+
408
+ def _translate_answer_code_to_answer(
409
+ self, answer, scenario: Optional["Scenario"] = None
410
+ ):
411
+ """There is over-ridden by child classes that ask for codes."""
412
+ return answer
413
+
414
+ def add_question(self, other: QuestionBase) -> "Survey":
415
+ """Add a question to this question by turning them into a survey with two questions.
416
+
417
+ >>> from edsl.questions import QuestionFreeText as Q
418
+ >>> from edsl.questions import QuestionMultipleChoice as QMC
419
+ >>> s = Q.example().add_question(QMC.example())
420
+ >>> len(s.questions)
421
+ 2
422
+ """
423
+ return self.to_survey().add_question(other)
424
+
425
+ def to_survey(self) -> "Survey":
426
+ """Turn a single question into a survey.
427
+ >>> from edsl import QuestionFreeText as Q
428
+ >>> Q.example().to_survey().questions[0].question_name
429
+ 'how_are_you'
430
+ """
431
+ from edsl.surveys.Survey import Survey
432
+
433
+ return Survey([self])
434
+
435
+ def by(self, *args) -> "Jobs":
436
+ """Turn a single question into a survey and then a Job."""
437
+ from edsl.surveys.Survey import Survey
438
+
439
+ s = Survey([self])
440
+ return s.by(*args)
441
+
442
+ def human_readable(self) -> str:
443
+ """Print the question in a human readable format.
444
+
445
+ >>> from edsl.questions import QuestionFreeText
446
+ >>> QuestionFreeText.example().human_readable()
447
+ 'Question Type: free_text\\nQuestion: How are you?'
448
+ """
449
+ lines = []
450
+ lines.append(f"Question Type: {self.question_type}")
451
+ lines.append(f"Question: {self.question_text}")
452
+ if hasattr(self, "question_options"):
453
+ lines.append("Please name the option you choose from the following.:")
454
+ for index, option in enumerate(self.question_options):
455
+ lines.append(f"{option}")
456
+ return "\n".join(lines)
457
+
458
+ def html(
459
+ self,
460
+ scenario: Optional[dict] = None,
461
+ agent: Optional[dict] = {},
462
+ answers: Optional[dict] = None,
463
+ include_question_name: bool = False,
464
+ height: Optional[int] = None,
465
+ width: Optional[int] = None,
466
+ iframe=False,
467
+ ):
468
+ from edsl.questions.HTMLQuestion import HTMLQuestion
469
+
470
+ return HTMLQuestion(self).html(
471
+ scenario, agent, answers, include_question_name, height, width, iframe
472
+ )
473
+
474
+ @classmethod
475
+ def example_model(cls):
476
+ from edsl.language_models.model import Model
477
+
478
+ q = cls.example()
479
+ m = Model("test", canned_response=cls._simulate_answer(q)["answer"])
480
+
481
+ return m
482
+
483
+ @classmethod
484
+ def example_results(cls):
485
+ m = cls.example_model()
486
+ q = cls.example()
487
+ return q.by(m).run(cache=False)
488
+
489
+ def rich_print(self):
490
+ """Print the question in a rich format."""
491
+ from rich.table import Table
492
+
493
+ table = Table(show_header=True, header_style="bold magenta")
494
+ table.add_column("Question Name", style="dim")
495
+ table.add_column("Question Type")
496
+ table.add_column("Question Text")
497
+ table.add_column("Options")
498
+
499
+ question = self
500
+ if hasattr(question, "question_options"):
501
+ options = ", ".join([str(o) for o in question.question_options])
502
+ else:
503
+ options = "None"
504
+ table.add_row(
505
+ question.question_name,
506
+ question.question_type,
507
+ question.question_text,
508
+ options,
509
+ )
510
+ return table
511
+
512
+ # endregion
513
+
514
+
515
+ if __name__ == "__main__":
516
+ import doctest
517
+
518
+ doctest.testmod(optionflags=doctest.ELLIPSIS)