edsl 0.1.38.dev2__py3-none-any.whl → 0.1.38.dev3__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 (248) hide show
  1. edsl/Base.py +303 -303
  2. edsl/BaseDiff.py +260 -260
  3. edsl/TemplateLoader.py +24 -24
  4. edsl/__init__.py +49 -49
  5. edsl/__version__.py +1 -1
  6. edsl/agents/Agent.py +858 -858
  7. edsl/agents/AgentList.py +362 -362
  8. edsl/agents/Invigilator.py +222 -222
  9. edsl/agents/InvigilatorBase.py +284 -284
  10. edsl/agents/PromptConstructor.py +353 -353
  11. edsl/agents/__init__.py +3 -3
  12. edsl/agents/descriptors.py +99 -99
  13. edsl/agents/prompt_helpers.py +129 -129
  14. edsl/auto/AutoStudy.py +117 -117
  15. edsl/auto/StageBase.py +230 -230
  16. edsl/auto/StageGenerateSurvey.py +178 -178
  17. edsl/auto/StageLabelQuestions.py +125 -125
  18. edsl/auto/StagePersona.py +61 -61
  19. edsl/auto/StagePersonaDimensionValueRanges.py +88 -88
  20. edsl/auto/StagePersonaDimensionValues.py +74 -74
  21. edsl/auto/StagePersonaDimensions.py +69 -69
  22. edsl/auto/StageQuestions.py +73 -73
  23. edsl/auto/SurveyCreatorPipeline.py +21 -21
  24. edsl/auto/utilities.py +224 -224
  25. edsl/base/Base.py +279 -279
  26. edsl/config.py +149 -149
  27. edsl/conversation/Conversation.py +290 -290
  28. edsl/conversation/car_buying.py +58 -58
  29. edsl/conversation/chips.py +95 -95
  30. edsl/conversation/mug_negotiation.py +81 -81
  31. edsl/conversation/next_speaker_utilities.py +93 -93
  32. edsl/coop/PriceFetcher.py +54 -54
  33. edsl/coop/__init__.py +2 -2
  34. edsl/coop/coop.py +961 -961
  35. edsl/coop/utils.py +131 -131
  36. edsl/data/Cache.py +530 -530
  37. edsl/data/CacheEntry.py +228 -228
  38. edsl/data/CacheHandler.py +149 -149
  39. edsl/data/RemoteCacheSync.py +97 -97
  40. edsl/data/SQLiteDict.py +292 -292
  41. edsl/data/__init__.py +4 -4
  42. edsl/data/orm.py +10 -10
  43. edsl/data_transfer_models.py +73 -73
  44. edsl/enums.py +173 -173
  45. edsl/exceptions/BaseException.py +21 -21
  46. edsl/exceptions/__init__.py +54 -54
  47. edsl/exceptions/agents.py +42 -42
  48. edsl/exceptions/cache.py +5 -5
  49. edsl/exceptions/configuration.py +16 -16
  50. edsl/exceptions/coop.py +10 -10
  51. edsl/exceptions/data.py +14 -14
  52. edsl/exceptions/general.py +34 -34
  53. edsl/exceptions/jobs.py +33 -33
  54. edsl/exceptions/language_models.py +63 -63
  55. edsl/exceptions/prompts.py +15 -15
  56. edsl/exceptions/questions.py +91 -91
  57. edsl/exceptions/results.py +29 -29
  58. edsl/exceptions/scenarios.py +22 -22
  59. edsl/exceptions/surveys.py +37 -37
  60. edsl/inference_services/AnthropicService.py +87 -87
  61. edsl/inference_services/AwsBedrock.py +120 -120
  62. edsl/inference_services/AzureAI.py +217 -217
  63. edsl/inference_services/DeepInfraService.py +18 -18
  64. edsl/inference_services/GoogleService.py +156 -156
  65. edsl/inference_services/GroqService.py +20 -20
  66. edsl/inference_services/InferenceServiceABC.py +147 -147
  67. edsl/inference_services/InferenceServicesCollection.py +97 -97
  68. edsl/inference_services/MistralAIService.py +123 -123
  69. edsl/inference_services/OllamaService.py +18 -18
  70. edsl/inference_services/OpenAIService.py +224 -224
  71. edsl/inference_services/TestService.py +89 -89
  72. edsl/inference_services/TogetherAIService.py +170 -170
  73. edsl/inference_services/models_available_cache.py +118 -118
  74. edsl/inference_services/rate_limits_cache.py +25 -25
  75. edsl/inference_services/registry.py +39 -39
  76. edsl/inference_services/write_available.py +10 -10
  77. edsl/jobs/Answers.py +56 -56
  78. edsl/jobs/Jobs.py +1358 -1358
  79. edsl/jobs/__init__.py +1 -1
  80. edsl/jobs/buckets/BucketCollection.py +63 -63
  81. edsl/jobs/buckets/ModelBuckets.py +65 -65
  82. edsl/jobs/buckets/TokenBucket.py +251 -251
  83. edsl/jobs/interviews/Interview.py +661 -661
  84. edsl/jobs/interviews/InterviewExceptionCollection.py +99 -99
  85. edsl/jobs/interviews/InterviewExceptionEntry.py +186 -186
  86. edsl/jobs/interviews/InterviewStatistic.py +63 -63
  87. edsl/jobs/interviews/InterviewStatisticsCollection.py +25 -25
  88. edsl/jobs/interviews/InterviewStatusDictionary.py +78 -78
  89. edsl/jobs/interviews/InterviewStatusLog.py +92 -92
  90. edsl/jobs/interviews/ReportErrors.py +66 -66
  91. edsl/jobs/interviews/interview_status_enum.py +9 -9
  92. edsl/jobs/runners/JobsRunnerAsyncio.py +361 -361
  93. edsl/jobs/runners/JobsRunnerStatus.py +332 -332
  94. edsl/jobs/tasks/QuestionTaskCreator.py +242 -242
  95. edsl/jobs/tasks/TaskCreators.py +64 -64
  96. edsl/jobs/tasks/TaskHistory.py +451 -451
  97. edsl/jobs/tasks/TaskStatusLog.py +23 -23
  98. edsl/jobs/tasks/task_status_enum.py +163 -163
  99. edsl/jobs/tokens/InterviewTokenUsage.py +27 -27
  100. edsl/jobs/tokens/TokenUsage.py +34 -34
  101. edsl/language_models/KeyLookup.py +30 -30
  102. edsl/language_models/LanguageModel.py +708 -708
  103. edsl/language_models/ModelList.py +109 -109
  104. edsl/language_models/RegisterLanguageModelsMeta.py +184 -184
  105. edsl/language_models/__init__.py +3 -3
  106. edsl/language_models/fake_openai_call.py +15 -15
  107. edsl/language_models/fake_openai_service.py +61 -61
  108. edsl/language_models/registry.py +137 -137
  109. edsl/language_models/repair.py +156 -156
  110. edsl/language_models/unused/ReplicateBase.py +83 -83
  111. edsl/language_models/utilities.py +64 -64
  112. edsl/notebooks/Notebook.py +258 -258
  113. edsl/notebooks/__init__.py +1 -1
  114. edsl/prompts/Prompt.py +357 -357
  115. edsl/prompts/__init__.py +2 -2
  116. edsl/questions/AnswerValidatorMixin.py +289 -289
  117. edsl/questions/QuestionBase.py +660 -660
  118. edsl/questions/QuestionBaseGenMixin.py +161 -161
  119. edsl/questions/QuestionBasePromptsMixin.py +217 -217
  120. edsl/questions/QuestionBudget.py +227 -227
  121. edsl/questions/QuestionCheckBox.py +359 -359
  122. edsl/questions/QuestionExtract.py +183 -183
  123. edsl/questions/QuestionFreeText.py +114 -114
  124. edsl/questions/QuestionFunctional.py +166 -166
  125. edsl/questions/QuestionList.py +231 -231
  126. edsl/questions/QuestionMultipleChoice.py +286 -286
  127. edsl/questions/QuestionNumerical.py +153 -153
  128. edsl/questions/QuestionRank.py +324 -324
  129. edsl/questions/Quick.py +41 -41
  130. edsl/questions/RegisterQuestionsMeta.py +71 -71
  131. edsl/questions/ResponseValidatorABC.py +174 -174
  132. edsl/questions/SimpleAskMixin.py +73 -73
  133. edsl/questions/__init__.py +26 -26
  134. edsl/questions/compose_questions.py +98 -98
  135. edsl/questions/decorators.py +21 -21
  136. edsl/questions/derived/QuestionLikertFive.py +76 -76
  137. edsl/questions/derived/QuestionLinearScale.py +87 -87
  138. edsl/questions/derived/QuestionTopK.py +93 -93
  139. edsl/questions/derived/QuestionYesNo.py +82 -82
  140. edsl/questions/descriptors.py +413 -413
  141. edsl/questions/prompt_templates/question_budget.jinja +13 -13
  142. edsl/questions/prompt_templates/question_checkbox.jinja +32 -32
  143. edsl/questions/prompt_templates/question_extract.jinja +11 -11
  144. edsl/questions/prompt_templates/question_free_text.jinja +3 -3
  145. edsl/questions/prompt_templates/question_linear_scale.jinja +11 -11
  146. edsl/questions/prompt_templates/question_list.jinja +17 -17
  147. edsl/questions/prompt_templates/question_multiple_choice.jinja +33 -33
  148. edsl/questions/prompt_templates/question_numerical.jinja +36 -36
  149. edsl/questions/question_registry.py +147 -147
  150. edsl/questions/settings.py +12 -12
  151. edsl/questions/templates/budget/answering_instructions.jinja +7 -7
  152. edsl/questions/templates/budget/question_presentation.jinja +7 -7
  153. edsl/questions/templates/checkbox/answering_instructions.jinja +10 -10
  154. edsl/questions/templates/checkbox/question_presentation.jinja +22 -22
  155. edsl/questions/templates/extract/answering_instructions.jinja +7 -7
  156. edsl/questions/templates/likert_five/answering_instructions.jinja +10 -10
  157. edsl/questions/templates/likert_five/question_presentation.jinja +11 -11
  158. edsl/questions/templates/linear_scale/answering_instructions.jinja +5 -5
  159. edsl/questions/templates/linear_scale/question_presentation.jinja +5 -5
  160. edsl/questions/templates/list/answering_instructions.jinja +3 -3
  161. edsl/questions/templates/list/question_presentation.jinja +5 -5
  162. edsl/questions/templates/multiple_choice/answering_instructions.jinja +9 -9
  163. edsl/questions/templates/multiple_choice/question_presentation.jinja +11 -11
  164. edsl/questions/templates/numerical/answering_instructions.jinja +6 -6
  165. edsl/questions/templates/numerical/question_presentation.jinja +6 -6
  166. edsl/questions/templates/rank/answering_instructions.jinja +11 -11
  167. edsl/questions/templates/rank/question_presentation.jinja +15 -15
  168. edsl/questions/templates/top_k/answering_instructions.jinja +8 -8
  169. edsl/questions/templates/top_k/question_presentation.jinja +22 -22
  170. edsl/questions/templates/yes_no/answering_instructions.jinja +6 -6
  171. edsl/questions/templates/yes_no/question_presentation.jinja +11 -11
  172. edsl/results/Dataset.py +293 -293
  173. edsl/results/DatasetExportMixin.py +717 -717
  174. edsl/results/DatasetTree.py +145 -145
  175. edsl/results/Result.py +456 -456
  176. edsl/results/Results.py +1071 -1071
  177. edsl/results/ResultsDBMixin.py +238 -238
  178. edsl/results/ResultsExportMixin.py +43 -43
  179. edsl/results/ResultsFetchMixin.py +33 -33
  180. edsl/results/ResultsGGMixin.py +121 -121
  181. edsl/results/ResultsToolsMixin.py +98 -98
  182. edsl/results/Selector.py +135 -135
  183. edsl/results/__init__.py +2 -2
  184. edsl/results/tree_explore.py +115 -115
  185. edsl/scenarios/FileStore.py +458 -458
  186. edsl/scenarios/Scenario.py +544 -544
  187. edsl/scenarios/ScenarioHtmlMixin.py +64 -64
  188. edsl/scenarios/ScenarioList.py +1112 -1112
  189. edsl/scenarios/ScenarioListExportMixin.py +52 -52
  190. edsl/scenarios/ScenarioListPdfMixin.py +261 -261
  191. edsl/scenarios/__init__.py +4 -4
  192. edsl/shared.py +1 -1
  193. edsl/study/ObjectEntry.py +173 -173
  194. edsl/study/ProofOfWork.py +113 -113
  195. edsl/study/SnapShot.py +80 -80
  196. edsl/study/Study.py +528 -528
  197. edsl/study/__init__.py +4 -4
  198. edsl/surveys/DAG.py +148 -148
  199. edsl/surveys/Memory.py +31 -31
  200. edsl/surveys/MemoryPlan.py +244 -244
  201. edsl/surveys/Rule.py +326 -326
  202. edsl/surveys/RuleCollection.py +387 -387
  203. edsl/surveys/Survey.py +1787 -1787
  204. edsl/surveys/SurveyCSS.py +261 -261
  205. edsl/surveys/SurveyExportMixin.py +259 -259
  206. edsl/surveys/SurveyFlowVisualizationMixin.py +121 -121
  207. edsl/surveys/SurveyQualtricsImport.py +284 -284
  208. edsl/surveys/__init__.py +3 -3
  209. edsl/surveys/base.py +53 -53
  210. edsl/surveys/descriptors.py +56 -56
  211. edsl/surveys/instructions/ChangeInstruction.py +49 -49
  212. edsl/surveys/instructions/Instruction.py +53 -53
  213. edsl/surveys/instructions/InstructionCollection.py +77 -77
  214. edsl/templates/error_reporting/base.html +23 -23
  215. edsl/templates/error_reporting/exceptions_by_model.html +34 -34
  216. edsl/templates/error_reporting/exceptions_by_question_name.html +16 -16
  217. edsl/templates/error_reporting/exceptions_by_type.html +16 -16
  218. edsl/templates/error_reporting/interview_details.html +115 -115
  219. edsl/templates/error_reporting/interviews.html +9 -9
  220. edsl/templates/error_reporting/overview.html +4 -4
  221. edsl/templates/error_reporting/performance_plot.html +1 -1
  222. edsl/templates/error_reporting/report.css +73 -73
  223. edsl/templates/error_reporting/report.html +117 -117
  224. edsl/templates/error_reporting/report.js +25 -25
  225. edsl/tools/__init__.py +1 -1
  226. edsl/tools/clusters.py +192 -192
  227. edsl/tools/embeddings.py +27 -27
  228. edsl/tools/embeddings_plotting.py +118 -118
  229. edsl/tools/plotting.py +112 -112
  230. edsl/tools/summarize.py +18 -18
  231. edsl/utilities/SystemInfo.py +28 -28
  232. edsl/utilities/__init__.py +22 -22
  233. edsl/utilities/ast_utilities.py +25 -25
  234. edsl/utilities/data/Registry.py +6 -6
  235. edsl/utilities/data/__init__.py +1 -1
  236. edsl/utilities/data/scooter_results.json +1 -1
  237. edsl/utilities/decorators.py +77 -77
  238. edsl/utilities/gcp_bucket/cloud_storage.py +96 -96
  239. edsl/utilities/interface.py +627 -627
  240. edsl/utilities/naming_utilities.py +263 -263
  241. edsl/utilities/repair_functions.py +28 -28
  242. edsl/utilities/restricted_python.py +70 -70
  243. edsl/utilities/utilities.py +409 -409
  244. {edsl-0.1.38.dev2.dist-info → edsl-0.1.38.dev3.dist-info}/LICENSE +21 -21
  245. {edsl-0.1.38.dev2.dist-info → edsl-0.1.38.dev3.dist-info}/METADATA +1 -1
  246. edsl-0.1.38.dev3.dist-info/RECORD +269 -0
  247. edsl-0.1.38.dev2.dist-info/RECORD +0 -269
  248. {edsl-0.1.38.dev2.dist-info → edsl-0.1.38.dev3.dist-info}/WHEEL +0 -0
@@ -1,660 +1,660 @@
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
- self._fake_data_factory = FakeData
88
- return self._fake_data_factory
89
-
90
- def _simulate_answer(self, human_readable: bool = False) -> dict:
91
- """Simulate a valid answer for debugging purposes (what the validator expects).
92
- >>> from edsl import QuestionFreeText as Q
93
- >>> Q.example()._simulate_answer()
94
- {'answer': '...', 'generated_tokens': ...}
95
- """
96
- simulated_answer = self.fake_data_factory.build().dict()
97
- if human_readable and hasattr(self, "question_options") and self.use_code:
98
- simulated_answer["answer"] = [
99
- self.question_options[index] for index in simulated_answer["answer"]
100
- ]
101
- return simulated_answer
102
-
103
- class ValidatedAnswer(TypedDict):
104
- answer: Any
105
- comment: Optional[str]
106
- generated_tokens: Optional[str]
107
-
108
- def _validate_answer(
109
- self, answer: dict, replacement_dict: dict = None
110
- ) -> ValidatedAnswer:
111
- """Validate the answer.
112
- >>> from edsl.exceptions import QuestionAnswerValidationError
113
- >>> from edsl import QuestionFreeText as Q
114
- >>> Q.example()._validate_answer({'answer': 'Hello', 'generated_tokens': 'Hello'})
115
- {'answer': 'Hello', 'generated_tokens': 'Hello'}
116
- """
117
-
118
- return self.response_validator.validate(answer, replacement_dict)
119
-
120
- # endregion
121
-
122
- # region: Serialization methods
123
- @property
124
- def name(self) -> str:
125
- "Helper function so questions and instructions can use the same access method"
126
- return self.question_name
127
-
128
- def __hash__(self) -> int:
129
- """Return a hash of the question.
130
-
131
- >>> from edsl import QuestionFreeText as Q
132
- >>> hash(Q.example())
133
- 1144312636257752766
134
- """
135
- from edsl.utilities.utilities import dict_hash
136
-
137
- return dict_hash(self.to_dict(add_edsl_version=False))
138
-
139
- @property
140
- def data(self) -> dict:
141
- """Return a dictionary of question attributes **except** for question_type.
142
-
143
- >>> from edsl import QuestionFreeText as Q
144
- >>> Q.example().data
145
- {'question_name': 'how_are_you', 'question_text': 'How are you?'}
146
- """
147
- exclude_list = [
148
- "question_type",
149
- # "_include_comment",
150
- "_fake_data_factory",
151
- # "_use_code",
152
- "_model_instructions",
153
- ]
154
- only_if_not_na_list = ["_answering_instructions", "_question_presentation"]
155
-
156
- only_if_not_default_list = {"_include_comment": True, "_use_code": False}
157
-
158
- def ok(key, value):
159
- if not key.startswith("_"):
160
- return False
161
- if key in exclude_list:
162
- return False
163
- if key in only_if_not_na_list and value is None:
164
- return False
165
- if (
166
- key in only_if_not_default_list
167
- and value == only_if_not_default_list[key]
168
- ):
169
- return False
170
-
171
- return True
172
-
173
- candidate_data = {
174
- k.replace("_", "", 1): v for k, v in self.__dict__.items() if ok(k, v)
175
- }
176
-
177
- if "func" in candidate_data:
178
- func = candidate_data.pop("func")
179
- import inspect
180
-
181
- candidate_data["function_source_code"] = inspect.getsource(func)
182
-
183
- return candidate_data
184
-
185
- def to_dict(self, add_edsl_version=True):
186
- """Convert the question to a dictionary that includes the question type (used in deserialization).
187
-
188
- >>> from edsl import QuestionFreeText as Q; Q.example().to_dict(add_edsl_version = False)
189
- {'question_name': 'how_are_you', 'question_text': 'How are you?', 'question_type': 'free_text'}
190
- """
191
- candidate_data = self.data.copy()
192
- candidate_data["question_type"] = self.question_type
193
- d = {key: value for key, value in candidate_data.items() if value is not None}
194
- if add_edsl_version:
195
- from edsl import __version__
196
-
197
- d["edsl_version"] = __version__
198
- d["edsl_class_name"] = "QuestionBase"
199
-
200
- return d
201
-
202
- @classmethod
203
- @remove_edsl_version
204
- def from_dict(cls, data: dict) -> Type[QuestionBase]:
205
- """Construct a question object from a dictionary created by that question's `to_dict` method."""
206
- local_data = data.copy()
207
-
208
- try:
209
- question_type = local_data.pop("question_type")
210
- if question_type == "linear_scale":
211
- # This is a fix for issue https://github.com/expectedparrot/edsl/issues/165
212
- options_labels = local_data.get("option_labels", None)
213
- if options_labels:
214
- options_labels = {
215
- int(key): value for key, value in options_labels.items()
216
- }
217
- local_data["option_labels"] = options_labels
218
- except:
219
- raise QuestionSerializationError(
220
- f"Data does not have a 'question_type' field (got {data})."
221
- )
222
- from edsl.questions.question_registry import get_question_class
223
-
224
- try:
225
- question_class = get_question_class(question_type)
226
- except ValueError:
227
- raise QuestionSerializationError(
228
- f"No question registered with question_type {question_type}"
229
- )
230
-
231
- if "model_instructions" in local_data:
232
- model_instructions = local_data.pop("model_instructions")
233
- new_q = question_class(**local_data)
234
- new_q.model_instructions = model_instructions
235
- return new_q
236
-
237
- return question_class(**local_data)
238
-
239
- # endregion
240
-
241
- # region: Running methods
242
- @classmethod
243
- def _get_test_model(self, canned_response: Optional[str] = None) -> "LanguageModel":
244
- """Get a test model for the question."""
245
- from edsl.language_models import LanguageModel
246
-
247
- return LanguageModel.example(canned_response=canned_response, test_model=True)
248
-
249
- @classmethod
250
- def run_example(
251
- cls,
252
- show_answer: bool = True,
253
- model: Optional["LanguageModel"] = None,
254
- cache=False,
255
- disable_remote_cache: bool = False,
256
- disable_remote_inference: bool = False,
257
- **kwargs,
258
- ):
259
- """Run an example of the question.
260
- >>> from edsl.language_models import LanguageModel
261
- >>> from edsl import QuestionFreeText as Q
262
- >>> m = Q._get_test_model(canned_response = "Yo, what's up?")
263
- >>> m.execute_model_call("", "")
264
- {'message': [{'text': "Yo, what's up?"}], 'usage': {'prompt_tokens': 1, 'completion_tokens': 1}}
265
- >>> Q.run_example(show_answer = True, model = m, disable_remote_cache = True, disable_remote_inference = True)
266
- ┏━━━━━━━━━━━━━━━━┓
267
- ┃ answer ┃
268
- ┃ .how_are_you ┃
269
- ┡━━━━━━━━━━━━━━━━┩
270
- │ Yo, what's up? │
271
- └────────────────┘
272
- """
273
- if model is None:
274
- from edsl import Model
275
-
276
- model = Model()
277
- results = (
278
- cls.example(**kwargs)
279
- .by(model)
280
- .run(
281
- cache=cache,
282
- disable_remote_cache=disable_remote_cache,
283
- disable_remote_inference=disable_remote_inference,
284
- )
285
- )
286
- if show_answer:
287
- results.select("answer.*").print()
288
- else:
289
- return results
290
-
291
- def __call__(
292
- self,
293
- just_answer=True,
294
- model=None,
295
- agent=None,
296
- disable_remote_cache: bool = False,
297
- disable_remote_inference: bool = False,
298
- **kwargs,
299
- ):
300
- """Call the question.
301
-
302
-
303
- >>> from edsl import QuestionFreeText as Q
304
- >>> m = Q._get_test_model(canned_response = "Yo, what's up?")
305
- >>> q = Q(question_name = "color", question_text = "What is your favorite color?")
306
- >>> q(model = m, disable_remote_cache = True, disable_remote_inference = True)
307
- "Yo, what's up?"
308
-
309
- """
310
- survey = self.to_survey()
311
- results = survey(
312
- model=model,
313
- agent=agent,
314
- **kwargs,
315
- cache=False,
316
- disable_remote_cache=disable_remote_cache,
317
- disable_remote_inference=disable_remote_inference,
318
- )
319
- if just_answer:
320
- return results.select(f"answer.{self.question_name}").first()
321
- else:
322
- return results
323
-
324
- def run(self, *args, **kwargs) -> "Results":
325
- """Turn a single question into a survey and runs it."""
326
- from edsl.surveys.Survey import Survey
327
-
328
- s = self.to_survey()
329
- return s.run(*args, **kwargs)
330
-
331
- async def run_async(
332
- self,
333
- just_answer: bool = True,
334
- model: Optional["Model"] = None,
335
- agent: Optional["Agent"] = None,
336
- disable_remote_inference: bool = False,
337
- **kwargs,
338
- ) -> Union[Any, "Results"]:
339
- """Call the question asynchronously.
340
-
341
- >>> import asyncio
342
- >>> from edsl import QuestionFreeText as Q
343
- >>> m = Q._get_test_model(canned_response = "Blue")
344
- >>> q = Q(question_name = "color", question_text = "What is your favorite color?")
345
- >>> async def test_run_async(): result = await q.run_async(model=m, disable_remote_inference = True); print(result)
346
- >>> asyncio.run(test_run_async())
347
- Blue
348
- """
349
- survey = self.to_survey()
350
- results = await survey.run_async(
351
- model=model,
352
- agent=agent,
353
- disable_remote_inference=disable_remote_inference,
354
- **kwargs,
355
- )
356
- if just_answer:
357
- return results.select(f"answer.{self.question_name}").first()
358
- else:
359
- return results
360
-
361
- # endregion
362
-
363
- # region: Magic methods
364
- def _repr_html_(self):
365
- from edsl.utilities.utilities import data_to_html
366
-
367
- data = self.to_dict()
368
- try:
369
- _ = data.pop("edsl_version")
370
- _ = data.pop("edsl_class_name")
371
- except KeyError:
372
- print("Serialized question lacks edsl version, but is should have it.")
373
-
374
- return data_to_html(data)
375
-
376
- def __getitem__(self, key: str) -> Any:
377
- """Get an attribute of the question so it can be treated like a dictionary.
378
-
379
- >>> from edsl.questions import QuestionFreeText as Q
380
- >>> Q.example()['question_text']
381
- 'How are you?'
382
- """
383
- return getattr(self, key)
384
-
385
- def __repr__(self) -> str:
386
- """Return a string representation of the question. Should be able to be used to reconstruct the question.
387
-
388
- >>> from edsl import QuestionFreeText as Q
389
- >>> repr(Q.example())
390
- 'Question(\\'free_text\\', question_name = \"""how_are_you\""", question_text = \"""How are you?\""")'
391
- """
392
- items = [
393
- f'{k} = """{v}"""' if isinstance(v, str) else f"{k} = {v}"
394
- for k, v in self.data.items()
395
- if k != "question_type"
396
- ]
397
- question_type = self.to_dict().get("question_type", "None")
398
- return f"Question('{question_type}', {', '.join(items)})"
399
-
400
- def __eq__(self, other: Union[Any, Type[QuestionBase]]) -> bool:
401
- """Check if two questions are equal. Equality is defined as having the .to_dict().
402
-
403
- >>> from edsl import QuestionFreeText as Q
404
- >>> q1 = Q.example()
405
- >>> q2 = Q.example()
406
- >>> q1 == q2
407
- True
408
- >>> q1.question_text = "How are you John?"
409
- >>> q1 == q2
410
- False
411
-
412
- """
413
- if not isinstance(other, QuestionBase):
414
- return False
415
- return self.to_dict() == other.to_dict()
416
-
417
- def __sub__(self, other) -> BaseDiff:
418
- """Return the difference between two objects.
419
- >>> from edsl import QuestionFreeText as Q
420
- >>> q1 = Q.example()
421
- >>> q2 = q1.copy()
422
- >>> q2.question_text = "How are you John?"
423
- >>> diff = q1 - q2
424
- """
425
-
426
- return BaseDiff(other, self)
427
-
428
- # TODO: Throws an error that should be addressed at QuestionFunctional
429
- def __add__(self, other_question_or_diff):
430
- """
431
- Compose two questions into a single question.
432
-
433
- TODO: Probably getting deprecated.
434
-
435
- """
436
- if isinstance(other_question_or_diff, BaseDiff) or isinstance(
437
- other_question_or_diff, BaseDiffCollection
438
- ):
439
- return other_question_or_diff.apply(self)
440
-
441
- # from edsl.questions import compose_questions
442
- # return compose_questions(self, other_question_or_diff)
443
-
444
- # def _validate_response(self, response):
445
- # """Validate the response from the LLM. Behavior depends on the question type."""
446
- # if "answer" not in response:
447
- # raise QuestionResponseValidationError(
448
- # "Response from LLM does not have an answer"
449
- # )
450
- # return response
451
-
452
- def _translate_answer_code_to_answer(
453
- self, answer, scenario: Optional["Scenario"] = None
454
- ):
455
- """There is over-ridden by child classes that ask for codes."""
456
- return answer
457
-
458
- # endregion
459
-
460
- # region: Forward methods
461
- def add_question(self, other: QuestionBase) -> "Survey":
462
- """Add a question to this question by turning them into a survey with two questions.
463
-
464
- >>> from edsl.questions import QuestionFreeText as Q
465
- >>> from edsl.questions import QuestionMultipleChoice as QMC
466
- >>> s = Q.example().add_question(QMC.example())
467
- >>> len(s.questions)
468
- 2
469
- """
470
- from edsl.surveys.Survey import Survey
471
-
472
- s = Survey([self, other])
473
- return s
474
-
475
- def to_survey(self) -> "Survey":
476
- """Turn a single question into a survey.
477
- >>> from edsl import QuestionFreeText as Q
478
- >>> Q.example().to_survey().questions[0].question_name
479
- 'how_are_you'
480
- """
481
- from edsl.surveys.Survey import Survey
482
-
483
- s = Survey([self])
484
- return s
485
-
486
- def by(self, *args) -> "Jobs":
487
- """Turn a single question into a survey and then a Job."""
488
- from edsl.surveys.Survey import Survey
489
-
490
- s = Survey([self])
491
- return s.by(*args)
492
-
493
- # endregion
494
-
495
- # region: Display methods
496
- def print(self):
497
- from rich import print_json
498
- import json
499
-
500
- print_json(json.dumps(self.to_dict()))
501
-
502
- def human_readable(self) -> str:
503
- """Print the question in a human readable format.
504
-
505
- >>> from edsl.questions import QuestionFreeText
506
- >>> QuestionFreeText.example().human_readable()
507
- 'Question Type: free_text\\nQuestion: How are you?'
508
- """
509
- lines = []
510
- lines.append(f"Question Type: {self.question_type}")
511
- lines.append(f"Question: {self.question_text}")
512
- if hasattr(self, "question_options"):
513
- lines.append("Please name the option you choose from the following.:")
514
- for index, option in enumerate(self.question_options):
515
- lines.append(f"{option}")
516
- return "\n".join(lines)
517
-
518
- def html(
519
- self,
520
- scenario: Optional[dict] = None,
521
- agent: Optional[dict] = {},
522
- answers: Optional[dict] = None,
523
- include_question_name: bool = False,
524
- height: Optional[int] = None,
525
- width: Optional[int] = None,
526
- iframe=False,
527
- ):
528
- """Return the question in HTML format."""
529
- from jinja2 import Template
530
-
531
- if scenario is None:
532
- scenario = {}
533
-
534
- prior_answers_dict = {}
535
-
536
- if isinstance(answers, dict):
537
- for key, value in answers.items():
538
- if not key.endswith("_comment") and not key.endswith(
539
- "_generated_tokens"
540
- ):
541
- prior_answers_dict[key] = {"answer": value}
542
-
543
- # breakpoint()
544
-
545
- base_template = """
546
- <div id="{{ question_name }}" class="survey_question" data-type="{{ question_type }}">
547
- {% if include_question_name %}
548
- <p>question_name: {{ question_name }}</p>
549
- {% endif %}
550
- <p class="question_text">{{ question_text }}</p>
551
- {{ question_content }}
552
- </div>
553
- """
554
- if not hasattr(self, "question_type"):
555
- self.question_type = "unknown"
556
-
557
- if hasattr(self, "question_html_content"):
558
- question_content = self.question_html_content
559
- else:
560
- question_content = Template("")
561
-
562
- base_template = Template(base_template)
563
-
564
- context = {
565
- "scenario": scenario,
566
- "agent": agent,
567
- } | prior_answers_dict
568
-
569
- # Render the question text
570
- try:
571
- question_text = Template(self.question_text).render(context)
572
- except Exception as e:
573
- print(
574
- f"Error rendering question: question_text = {self.question_text}, error = {e}"
575
- )
576
- question_text = self.question_text
577
-
578
- try:
579
- question_content = Template(question_content).render(context)
580
- except Exception as e:
581
- print(
582
- f"Error rendering question: question_content = {question_content}, error = {e}"
583
- )
584
- question_content = question_content
585
-
586
- try:
587
- params = {
588
- "question_name": self.question_name,
589
- "question_text": question_text,
590
- "question_type": self.question_type,
591
- "question_content": question_content,
592
- "include_question_name": include_question_name,
593
- }
594
- except Exception as e:
595
- raise ValueError(
596
- f"Error rendering question: params = {params}, error = {e}"
597
- )
598
- rendered_html = base_template.render(**params)
599
-
600
- if iframe:
601
- import html
602
- from IPython.display import display, HTML
603
-
604
- height = height or 200
605
- width = width or 600
606
- escaped_output = html.escape(rendered_html)
607
- # escaped_output = rendered_html
608
- iframe = f""""
609
- <iframe srcdoc="{ escaped_output }" style="width: {width}px; height: {height}px;"></iframe>
610
- """
611
- display(HTML(iframe))
612
- return None
613
-
614
- return rendered_html
615
-
616
- @classmethod
617
- def example_model(cls):
618
- from edsl import Model
619
-
620
- q = cls.example()
621
- m = Model("test", canned_response=cls._simulate_answer(q)["answer"])
622
-
623
- return m
624
-
625
- @classmethod
626
- def example_results(cls):
627
- m = cls.example_model()
628
- q = cls.example()
629
- return q.by(m).run(cache=False)
630
-
631
- def rich_print(self):
632
- """Print the question in a rich format."""
633
- from rich.table import Table
634
-
635
- table = Table(show_header=True, header_style="bold magenta")
636
- table.add_column("Question Name", style="dim")
637
- table.add_column("Question Type")
638
- table.add_column("Question Text")
639
- table.add_column("Options")
640
-
641
- question = self
642
- if hasattr(question, "question_options"):
643
- options = ", ".join([str(o) for o in question.question_options])
644
- else:
645
- options = "None"
646
- table.add_row(
647
- question.question_name,
648
- question.question_type,
649
- question.question_text,
650
- options,
651
- )
652
- return table
653
-
654
- # endregion
655
-
656
-
657
- if __name__ == "__main__":
658
- import doctest
659
-
660
- 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
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
+ self._fake_data_factory = FakeData
88
+ return self._fake_data_factory
89
+
90
+ def _simulate_answer(self, human_readable: bool = False) -> dict:
91
+ """Simulate a valid answer for debugging purposes (what the validator expects).
92
+ >>> from edsl import QuestionFreeText as Q
93
+ >>> Q.example()._simulate_answer()
94
+ {'answer': '...', 'generated_tokens': ...}
95
+ """
96
+ simulated_answer = self.fake_data_factory.build().dict()
97
+ if human_readable and hasattr(self, "question_options") and self.use_code:
98
+ simulated_answer["answer"] = [
99
+ self.question_options[index] for index in simulated_answer["answer"]
100
+ ]
101
+ return simulated_answer
102
+
103
+ class ValidatedAnswer(TypedDict):
104
+ answer: Any
105
+ comment: Optional[str]
106
+ generated_tokens: Optional[str]
107
+
108
+ def _validate_answer(
109
+ self, answer: dict, replacement_dict: dict = None
110
+ ) -> ValidatedAnswer:
111
+ """Validate the answer.
112
+ >>> from edsl.exceptions import QuestionAnswerValidationError
113
+ >>> from edsl import QuestionFreeText as Q
114
+ >>> Q.example()._validate_answer({'answer': 'Hello', 'generated_tokens': 'Hello'})
115
+ {'answer': 'Hello', 'generated_tokens': 'Hello'}
116
+ """
117
+
118
+ return self.response_validator.validate(answer, replacement_dict)
119
+
120
+ # endregion
121
+
122
+ # region: Serialization methods
123
+ @property
124
+ def name(self) -> str:
125
+ "Helper function so questions and instructions can use the same access method"
126
+ return self.question_name
127
+
128
+ def __hash__(self) -> int:
129
+ """Return a hash of the question.
130
+
131
+ >>> from edsl import QuestionFreeText as Q
132
+ >>> hash(Q.example())
133
+ 1144312636257752766
134
+ """
135
+ from edsl.utilities.utilities import dict_hash
136
+
137
+ return dict_hash(self.to_dict(add_edsl_version=False))
138
+
139
+ @property
140
+ def data(self) -> dict:
141
+ """Return a dictionary of question attributes **except** for question_type.
142
+
143
+ >>> from edsl import QuestionFreeText as Q
144
+ >>> Q.example().data
145
+ {'question_name': 'how_are_you', 'question_text': 'How are you?'}
146
+ """
147
+ exclude_list = [
148
+ "question_type",
149
+ # "_include_comment",
150
+ "_fake_data_factory",
151
+ # "_use_code",
152
+ "_model_instructions",
153
+ ]
154
+ only_if_not_na_list = ["_answering_instructions", "_question_presentation"]
155
+
156
+ only_if_not_default_list = {"_include_comment": True, "_use_code": False}
157
+
158
+ def ok(key, value):
159
+ if not key.startswith("_"):
160
+ return False
161
+ if key in exclude_list:
162
+ return False
163
+ if key in only_if_not_na_list and value is None:
164
+ return False
165
+ if (
166
+ key in only_if_not_default_list
167
+ and value == only_if_not_default_list[key]
168
+ ):
169
+ return False
170
+
171
+ return True
172
+
173
+ candidate_data = {
174
+ k.replace("_", "", 1): v for k, v in self.__dict__.items() if ok(k, v)
175
+ }
176
+
177
+ if "func" in candidate_data:
178
+ func = candidate_data.pop("func")
179
+ import inspect
180
+
181
+ candidate_data["function_source_code"] = inspect.getsource(func)
182
+
183
+ return candidate_data
184
+
185
+ def to_dict(self, add_edsl_version=True):
186
+ """Convert the question to a dictionary that includes the question type (used in deserialization).
187
+
188
+ >>> from edsl import QuestionFreeText as Q; Q.example().to_dict(add_edsl_version = False)
189
+ {'question_name': 'how_are_you', 'question_text': 'How are you?', 'question_type': 'free_text'}
190
+ """
191
+ candidate_data = self.data.copy()
192
+ candidate_data["question_type"] = self.question_type
193
+ d = {key: value for key, value in candidate_data.items() if value is not None}
194
+ if add_edsl_version:
195
+ from edsl import __version__
196
+
197
+ d["edsl_version"] = __version__
198
+ d["edsl_class_name"] = "QuestionBase"
199
+
200
+ return d
201
+
202
+ @classmethod
203
+ @remove_edsl_version
204
+ def from_dict(cls, data: dict) -> Type[QuestionBase]:
205
+ """Construct a question object from a dictionary created by that question's `to_dict` method."""
206
+ local_data = data.copy()
207
+
208
+ try:
209
+ question_type = local_data.pop("question_type")
210
+ if question_type == "linear_scale":
211
+ # This is a fix for issue https://github.com/expectedparrot/edsl/issues/165
212
+ options_labels = local_data.get("option_labels", None)
213
+ if options_labels:
214
+ options_labels = {
215
+ int(key): value for key, value in options_labels.items()
216
+ }
217
+ local_data["option_labels"] = options_labels
218
+ except:
219
+ raise QuestionSerializationError(
220
+ f"Data does not have a 'question_type' field (got {data})."
221
+ )
222
+ from edsl.questions.question_registry import get_question_class
223
+
224
+ try:
225
+ question_class = get_question_class(question_type)
226
+ except ValueError:
227
+ raise QuestionSerializationError(
228
+ f"No question registered with question_type {question_type}"
229
+ )
230
+
231
+ if "model_instructions" in local_data:
232
+ model_instructions = local_data.pop("model_instructions")
233
+ new_q = question_class(**local_data)
234
+ new_q.model_instructions = model_instructions
235
+ return new_q
236
+
237
+ return question_class(**local_data)
238
+
239
+ # endregion
240
+
241
+ # region: Running methods
242
+ @classmethod
243
+ def _get_test_model(self, canned_response: Optional[str] = None) -> "LanguageModel":
244
+ """Get a test model for the question."""
245
+ from edsl.language_models import LanguageModel
246
+
247
+ return LanguageModel.example(canned_response=canned_response, test_model=True)
248
+
249
+ @classmethod
250
+ def run_example(
251
+ cls,
252
+ show_answer: bool = True,
253
+ model: Optional["LanguageModel"] = None,
254
+ cache=False,
255
+ disable_remote_cache: bool = False,
256
+ disable_remote_inference: bool = False,
257
+ **kwargs,
258
+ ):
259
+ """Run an example of the question.
260
+ >>> from edsl.language_models import LanguageModel
261
+ >>> from edsl import QuestionFreeText as Q
262
+ >>> m = Q._get_test_model(canned_response = "Yo, what's up?")
263
+ >>> m.execute_model_call("", "")
264
+ {'message': [{'text': "Yo, what's up?"}], 'usage': {'prompt_tokens': 1, 'completion_tokens': 1}}
265
+ >>> Q.run_example(show_answer = True, model = m, disable_remote_cache = True, disable_remote_inference = True)
266
+ ┏━━━━━━━━━━━━━━━━┓
267
+ ┃ answer ┃
268
+ ┃ .how_are_you ┃
269
+ ┡━━━━━━━━━━━━━━━━┩
270
+ │ Yo, what's up? │
271
+ └────────────────┘
272
+ """
273
+ if model is None:
274
+ from edsl import Model
275
+
276
+ model = Model()
277
+ results = (
278
+ cls.example(**kwargs)
279
+ .by(model)
280
+ .run(
281
+ cache=cache,
282
+ disable_remote_cache=disable_remote_cache,
283
+ disable_remote_inference=disable_remote_inference,
284
+ )
285
+ )
286
+ if show_answer:
287
+ results.select("answer.*").print()
288
+ else:
289
+ return results
290
+
291
+ def __call__(
292
+ self,
293
+ just_answer=True,
294
+ model=None,
295
+ agent=None,
296
+ disable_remote_cache: bool = False,
297
+ disable_remote_inference: bool = False,
298
+ **kwargs,
299
+ ):
300
+ """Call the question.
301
+
302
+
303
+ >>> from edsl import QuestionFreeText as Q
304
+ >>> m = Q._get_test_model(canned_response = "Yo, what's up?")
305
+ >>> q = Q(question_name = "color", question_text = "What is your favorite color?")
306
+ >>> q(model = m, disable_remote_cache = True, disable_remote_inference = True)
307
+ "Yo, what's up?"
308
+
309
+ """
310
+ survey = self.to_survey()
311
+ results = survey(
312
+ model=model,
313
+ agent=agent,
314
+ **kwargs,
315
+ cache=False,
316
+ disable_remote_cache=disable_remote_cache,
317
+ disable_remote_inference=disable_remote_inference,
318
+ )
319
+ if just_answer:
320
+ return results.select(f"answer.{self.question_name}").first()
321
+ else:
322
+ return results
323
+
324
+ def run(self, *args, **kwargs) -> "Results":
325
+ """Turn a single question into a survey and runs it."""
326
+ from edsl.surveys.Survey import Survey
327
+
328
+ s = self.to_survey()
329
+ return s.run(*args, **kwargs)
330
+
331
+ async def run_async(
332
+ self,
333
+ just_answer: bool = True,
334
+ model: Optional["Model"] = None,
335
+ agent: Optional["Agent"] = None,
336
+ disable_remote_inference: bool = False,
337
+ **kwargs,
338
+ ) -> Union[Any, "Results"]:
339
+ """Call the question asynchronously.
340
+
341
+ >>> import asyncio
342
+ >>> from edsl import QuestionFreeText as Q
343
+ >>> m = Q._get_test_model(canned_response = "Blue")
344
+ >>> q = Q(question_name = "color", question_text = "What is your favorite color?")
345
+ >>> async def test_run_async(): result = await q.run_async(model=m, disable_remote_inference = True); print(result)
346
+ >>> asyncio.run(test_run_async())
347
+ Blue
348
+ """
349
+ survey = self.to_survey()
350
+ results = await survey.run_async(
351
+ model=model,
352
+ agent=agent,
353
+ disable_remote_inference=disable_remote_inference,
354
+ **kwargs,
355
+ )
356
+ if just_answer:
357
+ return results.select(f"answer.{self.question_name}").first()
358
+ else:
359
+ return results
360
+
361
+ # endregion
362
+
363
+ # region: Magic methods
364
+ def _repr_html_(self):
365
+ from edsl.utilities.utilities import data_to_html
366
+
367
+ data = self.to_dict()
368
+ try:
369
+ _ = data.pop("edsl_version")
370
+ _ = data.pop("edsl_class_name")
371
+ except KeyError:
372
+ print("Serialized question lacks edsl version, but is should have it.")
373
+
374
+ return data_to_html(data)
375
+
376
+ def __getitem__(self, key: str) -> Any:
377
+ """Get an attribute of the question so it can be treated like a dictionary.
378
+
379
+ >>> from edsl.questions import QuestionFreeText as Q
380
+ >>> Q.example()['question_text']
381
+ 'How are you?'
382
+ """
383
+ return getattr(self, key)
384
+
385
+ def __repr__(self) -> str:
386
+ """Return a string representation of the question. Should be able to be used to reconstruct the question.
387
+
388
+ >>> from edsl import QuestionFreeText as Q
389
+ >>> repr(Q.example())
390
+ 'Question(\\'free_text\\', question_name = \"""how_are_you\""", question_text = \"""How are you?\""")'
391
+ """
392
+ items = [
393
+ f'{k} = """{v}"""' if isinstance(v, str) else f"{k} = {v}"
394
+ for k, v in self.data.items()
395
+ if k != "question_type"
396
+ ]
397
+ question_type = self.to_dict().get("question_type", "None")
398
+ return f"Question('{question_type}', {', '.join(items)})"
399
+
400
+ def __eq__(self, other: Union[Any, Type[QuestionBase]]) -> bool:
401
+ """Check if two questions are equal. Equality is defined as having the .to_dict().
402
+
403
+ >>> from edsl import QuestionFreeText as Q
404
+ >>> q1 = Q.example()
405
+ >>> q2 = Q.example()
406
+ >>> q1 == q2
407
+ True
408
+ >>> q1.question_text = "How are you John?"
409
+ >>> q1 == q2
410
+ False
411
+
412
+ """
413
+ if not isinstance(other, QuestionBase):
414
+ return False
415
+ return self.to_dict() == other.to_dict()
416
+
417
+ def __sub__(self, other) -> BaseDiff:
418
+ """Return the difference between two objects.
419
+ >>> from edsl import QuestionFreeText as Q
420
+ >>> q1 = Q.example()
421
+ >>> q2 = q1.copy()
422
+ >>> q2.question_text = "How are you John?"
423
+ >>> diff = q1 - q2
424
+ """
425
+
426
+ return BaseDiff(other, self)
427
+
428
+ # TODO: Throws an error that should be addressed at QuestionFunctional
429
+ def __add__(self, other_question_or_diff):
430
+ """
431
+ Compose two questions into a single question.
432
+
433
+ TODO: Probably getting deprecated.
434
+
435
+ """
436
+ if isinstance(other_question_or_diff, BaseDiff) or isinstance(
437
+ other_question_or_diff, BaseDiffCollection
438
+ ):
439
+ return other_question_or_diff.apply(self)
440
+
441
+ # from edsl.questions import compose_questions
442
+ # return compose_questions(self, other_question_or_diff)
443
+
444
+ # def _validate_response(self, response):
445
+ # """Validate the response from the LLM. Behavior depends on the question type."""
446
+ # if "answer" not in response:
447
+ # raise QuestionResponseValidationError(
448
+ # "Response from LLM does not have an answer"
449
+ # )
450
+ # return response
451
+
452
+ def _translate_answer_code_to_answer(
453
+ self, answer, scenario: Optional["Scenario"] = None
454
+ ):
455
+ """There is over-ridden by child classes that ask for codes."""
456
+ return answer
457
+
458
+ # endregion
459
+
460
+ # region: Forward methods
461
+ def add_question(self, other: QuestionBase) -> "Survey":
462
+ """Add a question to this question by turning them into a survey with two questions.
463
+
464
+ >>> from edsl.questions import QuestionFreeText as Q
465
+ >>> from edsl.questions import QuestionMultipleChoice as QMC
466
+ >>> s = Q.example().add_question(QMC.example())
467
+ >>> len(s.questions)
468
+ 2
469
+ """
470
+ from edsl.surveys.Survey import Survey
471
+
472
+ s = Survey([self, other])
473
+ return s
474
+
475
+ def to_survey(self) -> "Survey":
476
+ """Turn a single question into a survey.
477
+ >>> from edsl import QuestionFreeText as Q
478
+ >>> Q.example().to_survey().questions[0].question_name
479
+ 'how_are_you'
480
+ """
481
+ from edsl.surveys.Survey import Survey
482
+
483
+ s = Survey([self])
484
+ return s
485
+
486
+ def by(self, *args) -> "Jobs":
487
+ """Turn a single question into a survey and then a Job."""
488
+ from edsl.surveys.Survey import Survey
489
+
490
+ s = Survey([self])
491
+ return s.by(*args)
492
+
493
+ # endregion
494
+
495
+ # region: Display methods
496
+ def print(self):
497
+ from rich import print_json
498
+ import json
499
+
500
+ print_json(json.dumps(self.to_dict()))
501
+
502
+ def human_readable(self) -> str:
503
+ """Print the question in a human readable format.
504
+
505
+ >>> from edsl.questions import QuestionFreeText
506
+ >>> QuestionFreeText.example().human_readable()
507
+ 'Question Type: free_text\\nQuestion: How are you?'
508
+ """
509
+ lines = []
510
+ lines.append(f"Question Type: {self.question_type}")
511
+ lines.append(f"Question: {self.question_text}")
512
+ if hasattr(self, "question_options"):
513
+ lines.append("Please name the option you choose from the following.:")
514
+ for index, option in enumerate(self.question_options):
515
+ lines.append(f"{option}")
516
+ return "\n".join(lines)
517
+
518
+ def html(
519
+ self,
520
+ scenario: Optional[dict] = None,
521
+ agent: Optional[dict] = {},
522
+ answers: Optional[dict] = None,
523
+ include_question_name: bool = False,
524
+ height: Optional[int] = None,
525
+ width: Optional[int] = None,
526
+ iframe=False,
527
+ ):
528
+ """Return the question in HTML format."""
529
+ from jinja2 import Template
530
+
531
+ if scenario is None:
532
+ scenario = {}
533
+
534
+ prior_answers_dict = {}
535
+
536
+ if isinstance(answers, dict):
537
+ for key, value in answers.items():
538
+ if not key.endswith("_comment") and not key.endswith(
539
+ "_generated_tokens"
540
+ ):
541
+ prior_answers_dict[key] = {"answer": value}
542
+
543
+ # breakpoint()
544
+
545
+ base_template = """
546
+ <div id="{{ question_name }}" class="survey_question" data-type="{{ question_type }}">
547
+ {% if include_question_name %}
548
+ <p>question_name: {{ question_name }}</p>
549
+ {% endif %}
550
+ <p class="question_text">{{ question_text }}</p>
551
+ {{ question_content }}
552
+ </div>
553
+ """
554
+ if not hasattr(self, "question_type"):
555
+ self.question_type = "unknown"
556
+
557
+ if hasattr(self, "question_html_content"):
558
+ question_content = self.question_html_content
559
+ else:
560
+ question_content = Template("")
561
+
562
+ base_template = Template(base_template)
563
+
564
+ context = {
565
+ "scenario": scenario,
566
+ "agent": agent,
567
+ } | prior_answers_dict
568
+
569
+ # Render the question text
570
+ try:
571
+ question_text = Template(self.question_text).render(context)
572
+ except Exception as e:
573
+ print(
574
+ f"Error rendering question: question_text = {self.question_text}, error = {e}"
575
+ )
576
+ question_text = self.question_text
577
+
578
+ try:
579
+ question_content = Template(question_content).render(context)
580
+ except Exception as e:
581
+ print(
582
+ f"Error rendering question: question_content = {question_content}, error = {e}"
583
+ )
584
+ question_content = question_content
585
+
586
+ try:
587
+ params = {
588
+ "question_name": self.question_name,
589
+ "question_text": question_text,
590
+ "question_type": self.question_type,
591
+ "question_content": question_content,
592
+ "include_question_name": include_question_name,
593
+ }
594
+ except Exception as e:
595
+ raise ValueError(
596
+ f"Error rendering question: params = {params}, error = {e}"
597
+ )
598
+ rendered_html = base_template.render(**params)
599
+
600
+ if iframe:
601
+ import html
602
+ from IPython.display import display, HTML
603
+
604
+ height = height or 200
605
+ width = width or 600
606
+ escaped_output = html.escape(rendered_html)
607
+ # escaped_output = rendered_html
608
+ iframe = f""""
609
+ <iframe srcdoc="{ escaped_output }" style="width: {width}px; height: {height}px;"></iframe>
610
+ """
611
+ display(HTML(iframe))
612
+ return None
613
+
614
+ return rendered_html
615
+
616
+ @classmethod
617
+ def example_model(cls):
618
+ from edsl import Model
619
+
620
+ q = cls.example()
621
+ m = Model("test", canned_response=cls._simulate_answer(q)["answer"])
622
+
623
+ return m
624
+
625
+ @classmethod
626
+ def example_results(cls):
627
+ m = cls.example_model()
628
+ q = cls.example()
629
+ return q.by(m).run(cache=False)
630
+
631
+ def rich_print(self):
632
+ """Print the question in a rich format."""
633
+ from rich.table import Table
634
+
635
+ table = Table(show_header=True, header_style="bold magenta")
636
+ table.add_column("Question Name", style="dim")
637
+ table.add_column("Question Type")
638
+ table.add_column("Question Text")
639
+ table.add_column("Options")
640
+
641
+ question = self
642
+ if hasattr(question, "question_options"):
643
+ options = ", ".join([str(o) for o in question.question_options])
644
+ else:
645
+ options = "None"
646
+ table.add_row(
647
+ question.question_name,
648
+ question.question_type,
649
+ question.question_text,
650
+ options,
651
+ )
652
+ return table
653
+
654
+ # endregion
655
+
656
+
657
+ if __name__ == "__main__":
658
+ import doctest
659
+
660
+ doctest.testmod(optionflags=doctest.ELLIPSIS)