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
edsl/results/Result.py CHANGED
@@ -1,456 +1,456 @@
1
- # """This module contains the Result class, which captures the result of one interview."""
2
- from __future__ import annotations
3
- from collections import UserDict
4
- from typing import Any, Type, Callable, Optional
5
- from collections import UserDict
6
- from edsl.Base import Base
7
- from edsl.utilities.decorators import add_edsl_version, remove_edsl_version
8
-
9
-
10
- class PromptDict(UserDict):
11
- """A dictionary that is used to store the prompt for a given result."""
12
-
13
- def rich_print(self):
14
- """Display an object as a table."""
15
- from rich.table import Table
16
-
17
- table = Table(title="")
18
- table.add_column("Attribute", style="bold")
19
- table.add_column("Value")
20
-
21
- to_display = self
22
- for attr_name, attr_value in to_display.items():
23
- table.add_row(attr_name, repr(attr_value))
24
-
25
- return table
26
-
27
-
28
- def agent_namer_closure():
29
- """Return a function that can be used to name an agent."""
30
- agent_dict = {}
31
-
32
- def agent_namer(agent):
33
- """Return a name for an agent. If the agent has been named before, return the same name. Otherwise, return a new name."""
34
- nonlocal agent_dict
35
- agent_count = len(agent_dict)
36
- if id(agent) in agent_dict:
37
- return agent_dict[id(agent)]
38
- else:
39
- agent_dict[id(agent)] = f"Agent_{agent_count}"
40
- return agent_dict[id(agent)]
41
-
42
- return agent_namer
43
-
44
-
45
- agent_namer = agent_namer_closure()
46
-
47
-
48
- class Result(Base, UserDict):
49
- """
50
- This class captures the result of one interview.
51
-
52
- The answer dictionary has the structure:
53
-
54
- >>> import warnings
55
- >>> warnings.simplefilter("ignore", UserWarning)
56
- >>> Result.example().answer == {'how_feeling_yesterday': 'Great', 'how_feeling': 'OK'}
57
- True
58
-
59
- Its main data is an Agent, a Scenario, a Model, an Iteration, and an Answer.
60
- These are stored both in the UserDict and as attributes.
61
-
62
-
63
- """
64
-
65
- def __init__(
66
- self,
67
- agent: "Agent",
68
- scenario: "Scenario",
69
- model: Type["LanguageModel"],
70
- iteration: int,
71
- answer: str,
72
- prompt: dict[str, str] = None,
73
- raw_model_response=None,
74
- survey: Optional["Survey"] = None,
75
- question_to_attributes: Optional[dict] = None,
76
- generated_tokens: Optional[dict] = None,
77
- comments_dict: Optional[dict] = None,
78
- cache_used_dict: Optional[dict] = None,
79
- ):
80
- """Initialize a Result object.
81
-
82
- :param agent: The Agent object.
83
- :param scenario: The Scenario object.
84
- :param model: The LanguageModel object.
85
- :param iteration: The iteration number.
86
- :param answer: The answer string.
87
- :param prompt: A dictionary of prompts.
88
- :param raw_model_response: The raw model response.
89
-
90
- """
91
- if question_to_attributes is not None:
92
- question_to_attributes = question_to_attributes
93
- else:
94
- question_to_attributes = {}
95
-
96
- if survey is not None:
97
- question_to_attributes = {
98
- q.question_name: {
99
- "question_text": q.question_text,
100
- "question_type": q.question_type,
101
- "question_options": (
102
- None
103
- if not hasattr(q, "question_options")
104
- else q.question_options
105
- ),
106
- }
107
- for q in survey.questions
108
- }
109
-
110
- data = {
111
- "agent": agent,
112
- "scenario": scenario,
113
- "model": model,
114
- "iteration": iteration,
115
- "answer": answer,
116
- "prompt": prompt or {},
117
- "raw_model_response": raw_model_response or {},
118
- "question_to_attributes": question_to_attributes,
119
- "generated_tokens": generated_tokens or {},
120
- "comments_dict": comments_dict or {},
121
- }
122
- super().__init__(**data)
123
- # but also store the data as attributes
124
- self.agent = agent
125
- self.scenario = scenario
126
- self.model = model
127
- self.iteration = iteration
128
- self.answer = answer
129
- self.prompt = prompt or {}
130
- self.raw_model_response = raw_model_response or {}
131
- self.survey = survey
132
- self.question_to_attributes = question_to_attributes
133
- self.generated_tokens = generated_tokens
134
- self.comments_dict = comments_dict or {}
135
- self.cache_used_dict = cache_used_dict or {}
136
-
137
- self._combined_dict = None
138
- self._problem_keys = None
139
-
140
- ###############
141
- # Used in Results
142
- ###############
143
- @property
144
- def sub_dicts(self) -> dict[str, dict]:
145
- """Return a dictionary where keys are strings for each of the main class attributes/objects."""
146
- if self.agent.name is None:
147
- agent_name = agent_namer(self.agent)
148
- else:
149
- agent_name = self.agent.name
150
-
151
- # comments_dict = {k: v for k, v in self.answer.items() if k.endswith("_comment")}
152
- question_text_dict = {}
153
- question_options_dict = {}
154
- question_type_dict = {}
155
- for key, _ in self.answer.items():
156
- if key in self.question_to_attributes:
157
- # You might be tempted to just use the naked key
158
- # but this is a bad idea because it pollutes the namespace
159
- question_text_dict[key + "_question_text"] = (
160
- self.question_to_attributes[key]["question_text"]
161
- )
162
- question_options_dict[key + "_question_options"] = (
163
- self.question_to_attributes[key]["question_options"]
164
- )
165
- question_type_dict[key + "_question_type"] = (
166
- self.question_to_attributes[key]["question_type"]
167
- )
168
-
169
- return {
170
- "agent": self.agent.traits
171
- | {"agent_name": agent_name}
172
- | {"agent_instruction": self.agent.instruction},
173
- "scenario": self.scenario,
174
- "model": self.model.parameters | {"model": self.model.model},
175
- "answer": self.answer,
176
- "prompt": self.prompt,
177
- "raw_model_response": self.raw_model_response,
178
- "iteration": {"iteration": self.iteration},
179
- "question_text": question_text_dict,
180
- "question_options": question_options_dict,
181
- "question_type": question_type_dict,
182
- "comment": self.comments_dict,
183
- "generated_tokens": self.generated_tokens,
184
- }
185
-
186
- def check_expression(self, expression) -> None:
187
- for key in self.problem_keys:
188
- if key in expression and not key + "." in expression:
189
- raise ValueError(
190
- f"Key by iself {key} is problematic. Use the full key {key + '.' + key} name instead."
191
- )
192
- return None
193
-
194
- def code(self):
195
- """Return a string of code that can be used to recreate the Result object."""
196
- raise NotImplementedError
197
-
198
- @property
199
- def problem_keys(self):
200
- """Return a list of keys that are problematic."""
201
- return self._problem_keys
202
-
203
- def _compute_combined_dict_and_problem_keys(self) -> None:
204
- combined = {}
205
- problem_keys = []
206
- for key, sub_dict in self.sub_dicts.items():
207
- combined.update(sub_dict)
208
- # in some cases, the sub_dict might have keys that conflict with the main dict
209
- if key in combined:
210
- # The key is already in the combined dict
211
- problem_keys = problem_keys + [key]
212
-
213
- combined.update({key: sub_dict})
214
- # I *think* this allows us to do do things like "answer.how_feelling" i.e., that the evaluator can use
215
- # dot notation to access the subdicts.
216
- self._combined_dict = combined
217
- self._problem_keys = problem_keys
218
-
219
- @property
220
- def combined_dict(self) -> dict[str, Any]:
221
- """Return a dictionary that includes all sub_dicts, but also puts the key-value pairs in each sub_dict as a key_value pair in the combined dictionary.
222
-
223
- >>> r = Result.example()
224
- >>> r.combined_dict['how_feeling']
225
- 'OK'
226
- """
227
- if self._combined_dict is None or self._problem_keys is None:
228
- self._compute_combined_dict_and_problem_keys()
229
- return self._combined_dict
230
-
231
- @property
232
- def problem_keys(self):
233
- """Return a list of keys that are problematic."""
234
- if self._combined_dict is None or self._problem_keys is None:
235
- self._compute_combined_dict_and_problem_keys()
236
- return self._problem_keys
237
-
238
- def get_value(self, data_type: str, key: str) -> Any:
239
- """Return the value for a given data type and key.
240
-
241
- >>> r = Result.example()
242
- >>> r.get_value("answer", "how_feeling")
243
- 'OK'
244
-
245
- - data types can be "agent", "scenario", "model", or "answer"
246
- - keys are relevant attributes of the Objects the data types represent
247
- """
248
- return self.sub_dicts[data_type][key]
249
-
250
- @property
251
- def key_to_data_type(self) -> dict[str, str]:
252
- """Return a dictionary where keys are object attributes and values are the data type (object) that the attribute is associated with.
253
-
254
- >>> r = Result.example()
255
- >>> r.key_to_data_type["how_feeling"]
256
- 'answer'
257
-
258
- """
259
- d = {}
260
- problem_keys = []
261
- data_types = sorted(self.sub_dicts.keys())
262
- for data_type in data_types:
263
- for key in self.sub_dicts[data_type]:
264
- if key in d:
265
- import warnings
266
-
267
- warnings.warn(
268
- f"Key '{key}' of data type '{data_type}' is already in use. Renaming to {key}_{data_type}"
269
- )
270
- problem_keys.append((key, data_type))
271
- key = f"{key}_{data_type}"
272
- # raise ValueError(f"Key '{key}' is already in the dictionary")
273
- d[key] = data_type
274
-
275
- for key, data_type in problem_keys:
276
- self.sub_dicts[data_type][f"{key}_{data_type}"] = self.sub_dicts[
277
- data_type
278
- ].pop(key)
279
- return d
280
-
281
- def rows(self, index) -> tuple[int, str, str, str]:
282
- """Return a generator of rows for the Result object."""
283
- for data_type, subdict in self.sub_dicts.items():
284
- for key, value in subdict.items():
285
- yield (index, data_type, key, str(value))
286
-
287
- def leaves(self):
288
- leaves = []
289
- for question_name, answer in self.answer.items():
290
- if not question_name.endswith("_comment"):
291
- leaves.append(
292
- {
293
- "question": f"({question_name}): "
294
- + str(
295
- self.question_to_attributes[question_name]["question_text"]
296
- ),
297
- "answer": answer,
298
- "comment": self.answer.get(question_name + "_comment", ""),
299
- "scenario": repr(self.scenario),
300
- "agent": repr(self.agent),
301
- "model": repr(self.model),
302
- "iteration": self.iteration,
303
- }
304
- )
305
- return leaves
306
-
307
- ###############
308
- # Useful
309
- ###############
310
- def copy(self) -> Result:
311
- """Return a copy of the Result object."""
312
- return Result.from_dict(self.to_dict())
313
-
314
- def __eq__(self, other) -> bool:
315
- """Return True if the Result object is equal to another Result object.
316
-
317
- >>> r = Result.example()
318
- >>> r == r
319
- True
320
-
321
- """
322
- return self.to_dict() == other.to_dict()
323
-
324
- ###############
325
- # Serialization
326
- ###############
327
- def to_dict(self, add_edsl_version=True) -> dict[str, Any]:
328
- """Return a dictionary representation of the Result object.
329
-
330
- >>> r = Result.example()
331
- >>> r.to_dict()['scenario']
332
- {'period': 'morning', 'edsl_version': '...', 'edsl_class_name': 'Scenario'}
333
- """
334
-
335
- def convert_value(value, add_edsl_version=True):
336
- if hasattr(value, "to_dict"):
337
- return value.to_dict(add_edsl_version=add_edsl_version)
338
- else:
339
- return value
340
-
341
- d = {}
342
- for key, value in self.items():
343
- d[key] = convert_value(value, add_edsl_version=add_edsl_version)
344
-
345
- if key == "prompt":
346
- new_prompt_dict = {}
347
- for prompt_name, prompt_obj in value.items():
348
- new_prompt_dict[prompt_name] = (
349
- prompt_obj
350
- if not hasattr(prompt_obj, "to_dict")
351
- else prompt_obj.to_dict()
352
- )
353
- d[key] = new_prompt_dict
354
- if add_edsl_version:
355
- from edsl import __version__
356
-
357
- d["edsl_version"] = __version__
358
- d["edsl_class_name"] = "Result"
359
-
360
- return d
361
-
362
- def __hash__(self):
363
- """Return a hash of the Result object."""
364
- from edsl.utilities.utilities import dict_hash
365
-
366
- return dict_hash(self.to_dict(add_edsl_version=False))
367
-
368
- @classmethod
369
- @remove_edsl_version
370
- def from_dict(self, json_dict: dict) -> Result:
371
- """Return a Result object from a dictionary representation."""
372
-
373
- from edsl import Agent
374
- from edsl import Scenario
375
- from edsl.language_models.LanguageModel import LanguageModel
376
- from edsl.prompts.Prompt import Prompt
377
-
378
- prompt_data = json_dict.get("prompt", {})
379
- prompt_d = {}
380
- for prompt_name, prompt_obj in prompt_data.items():
381
- prompt_d[prompt_name] = Prompt.from_dict(prompt_obj)
382
-
383
- result = Result(
384
- agent=Agent.from_dict(json_dict["agent"]),
385
- scenario=Scenario.from_dict(json_dict["scenario"]),
386
- model=LanguageModel.from_dict(json_dict["model"]),
387
- iteration=json_dict["iteration"],
388
- answer=json_dict["answer"],
389
- prompt=prompt_d, # json_dict["prompt"],
390
- raw_model_response=json_dict.get(
391
- "raw_model_response", {"raw_model_response": "No raw model response"}
392
- ),
393
- question_to_attributes=json_dict.get("question_to_attributes", None),
394
- generated_tokens=json_dict.get("generated_tokens", {}),
395
- comments_dict=json_dict.get("comments_dict", {}),
396
- )
397
- return result
398
-
399
- def rich_print(self) -> None:
400
- """Display an object as a table."""
401
- # from edsl.utilities import print_dict_with_rich
402
- from rich import print
403
- from rich.table import Table
404
-
405
- table = Table(title="Result")
406
- table.add_column("Attribute", style="bold")
407
- table.add_column("Value")
408
-
409
- to_display = self.__dict__.copy()
410
- data = to_display.pop("data", None)
411
- for attr_name, attr_value in to_display.items():
412
- if hasattr(attr_value, "rich_print"):
413
- table.add_row(attr_name, attr_value.rich_print())
414
- elif isinstance(attr_value, dict):
415
- a = PromptDict(attr_value)
416
- table.add_row(attr_name, a.rich_print())
417
- else:
418
- table.add_row(attr_name, repr(attr_value))
419
- return table
420
-
421
- def __repr__(self):
422
- """Return a string representation of the Result object."""
423
- return f"Result(agent={repr(self.agent)}, scenario={repr(self.scenario)}, model={repr(self.model)}, iteration={self.iteration}, answer={repr(self.answer)}, prompt={repr(self.prompt)})"
424
-
425
- @classmethod
426
- def example(cls):
427
- """Return an example Result object."""
428
- from edsl.results.Results import Results
429
-
430
- return Results.example()[0]
431
-
432
- def score(self, scoring_function: Callable) -> Any:
433
- """Score the result using a passed-in scoring function.
434
-
435
- >>> def f(status): return 1 if status == 'Joyful' else 0
436
- >>> Result.example().score(f)
437
- 1
438
- """
439
- import inspect
440
-
441
- signature = inspect.signature(scoring_function)
442
- params = {}
443
- for k, v in signature.parameters.items():
444
- if k in self.combined_dict:
445
- params[k] = self.combined_dict[k]
446
- elif v.default is not v.empty:
447
- params[k] = v.default
448
- else:
449
- raise ValueError(f"Parameter {k} not found in Result object")
450
- return scoring_function(**params)
451
-
452
-
453
- if __name__ == "__main__":
454
- import doctest
455
-
456
- doctest.testmod(optionflags=doctest.ELLIPSIS)
1
+ # """This module contains the Result class, which captures the result of one interview."""
2
+ from __future__ import annotations
3
+ from collections import UserDict
4
+ from typing import Any, Type, Callable, Optional
5
+ from collections import UserDict
6
+ from edsl.Base import Base
7
+ from edsl.utilities.decorators import add_edsl_version, remove_edsl_version
8
+
9
+
10
+ class PromptDict(UserDict):
11
+ """A dictionary that is used to store the prompt for a given result."""
12
+
13
+ def rich_print(self):
14
+ """Display an object as a table."""
15
+ from rich.table import Table
16
+
17
+ table = Table(title="")
18
+ table.add_column("Attribute", style="bold")
19
+ table.add_column("Value")
20
+
21
+ to_display = self
22
+ for attr_name, attr_value in to_display.items():
23
+ table.add_row(attr_name, repr(attr_value))
24
+
25
+ return table
26
+
27
+
28
+ def agent_namer_closure():
29
+ """Return a function that can be used to name an agent."""
30
+ agent_dict = {}
31
+
32
+ def agent_namer(agent):
33
+ """Return a name for an agent. If the agent has been named before, return the same name. Otherwise, return a new name."""
34
+ nonlocal agent_dict
35
+ agent_count = len(agent_dict)
36
+ if id(agent) in agent_dict:
37
+ return agent_dict[id(agent)]
38
+ else:
39
+ agent_dict[id(agent)] = f"Agent_{agent_count}"
40
+ return agent_dict[id(agent)]
41
+
42
+ return agent_namer
43
+
44
+
45
+ agent_namer = agent_namer_closure()
46
+
47
+
48
+ class Result(Base, UserDict):
49
+ """
50
+ This class captures the result of one interview.
51
+
52
+ The answer dictionary has the structure:
53
+
54
+ >>> import warnings
55
+ >>> warnings.simplefilter("ignore", UserWarning)
56
+ >>> Result.example().answer == {'how_feeling_yesterday': 'Great', 'how_feeling': 'OK'}
57
+ True
58
+
59
+ Its main data is an Agent, a Scenario, a Model, an Iteration, and an Answer.
60
+ These are stored both in the UserDict and as attributes.
61
+
62
+
63
+ """
64
+
65
+ def __init__(
66
+ self,
67
+ agent: "Agent",
68
+ scenario: "Scenario",
69
+ model: Type["LanguageModel"],
70
+ iteration: int,
71
+ answer: str,
72
+ prompt: dict[str, str] = None,
73
+ raw_model_response=None,
74
+ survey: Optional["Survey"] = None,
75
+ question_to_attributes: Optional[dict] = None,
76
+ generated_tokens: Optional[dict] = None,
77
+ comments_dict: Optional[dict] = None,
78
+ cache_used_dict: Optional[dict] = None,
79
+ ):
80
+ """Initialize a Result object.
81
+
82
+ :param agent: The Agent object.
83
+ :param scenario: The Scenario object.
84
+ :param model: The LanguageModel object.
85
+ :param iteration: The iteration number.
86
+ :param answer: The answer string.
87
+ :param prompt: A dictionary of prompts.
88
+ :param raw_model_response: The raw model response.
89
+
90
+ """
91
+ if question_to_attributes is not None:
92
+ question_to_attributes = question_to_attributes
93
+ else:
94
+ question_to_attributes = {}
95
+
96
+ if survey is not None:
97
+ question_to_attributes = {
98
+ q.question_name: {
99
+ "question_text": q.question_text,
100
+ "question_type": q.question_type,
101
+ "question_options": (
102
+ None
103
+ if not hasattr(q, "question_options")
104
+ else q.question_options
105
+ ),
106
+ }
107
+ for q in survey.questions
108
+ }
109
+
110
+ data = {
111
+ "agent": agent,
112
+ "scenario": scenario,
113
+ "model": model,
114
+ "iteration": iteration,
115
+ "answer": answer,
116
+ "prompt": prompt or {},
117
+ "raw_model_response": raw_model_response or {},
118
+ "question_to_attributes": question_to_attributes,
119
+ "generated_tokens": generated_tokens or {},
120
+ "comments_dict": comments_dict or {},
121
+ }
122
+ super().__init__(**data)
123
+ # but also store the data as attributes
124
+ self.agent = agent
125
+ self.scenario = scenario
126
+ self.model = model
127
+ self.iteration = iteration
128
+ self.answer = answer
129
+ self.prompt = prompt or {}
130
+ self.raw_model_response = raw_model_response or {}
131
+ self.survey = survey
132
+ self.question_to_attributes = question_to_attributes
133
+ self.generated_tokens = generated_tokens
134
+ self.comments_dict = comments_dict or {}
135
+ self.cache_used_dict = cache_used_dict or {}
136
+
137
+ self._combined_dict = None
138
+ self._problem_keys = None
139
+
140
+ ###############
141
+ # Used in Results
142
+ ###############
143
+ @property
144
+ def sub_dicts(self) -> dict[str, dict]:
145
+ """Return a dictionary where keys are strings for each of the main class attributes/objects."""
146
+ if self.agent.name is None:
147
+ agent_name = agent_namer(self.agent)
148
+ else:
149
+ agent_name = self.agent.name
150
+
151
+ # comments_dict = {k: v for k, v in self.answer.items() if k.endswith("_comment")}
152
+ question_text_dict = {}
153
+ question_options_dict = {}
154
+ question_type_dict = {}
155
+ for key, _ in self.answer.items():
156
+ if key in self.question_to_attributes:
157
+ # You might be tempted to just use the naked key
158
+ # but this is a bad idea because it pollutes the namespace
159
+ question_text_dict[key + "_question_text"] = (
160
+ self.question_to_attributes[key]["question_text"]
161
+ )
162
+ question_options_dict[key + "_question_options"] = (
163
+ self.question_to_attributes[key]["question_options"]
164
+ )
165
+ question_type_dict[key + "_question_type"] = (
166
+ self.question_to_attributes[key]["question_type"]
167
+ )
168
+
169
+ return {
170
+ "agent": self.agent.traits
171
+ | {"agent_name": agent_name}
172
+ | {"agent_instruction": self.agent.instruction},
173
+ "scenario": self.scenario,
174
+ "model": self.model.parameters | {"model": self.model.model},
175
+ "answer": self.answer,
176
+ "prompt": self.prompt,
177
+ "raw_model_response": self.raw_model_response,
178
+ "iteration": {"iteration": self.iteration},
179
+ "question_text": question_text_dict,
180
+ "question_options": question_options_dict,
181
+ "question_type": question_type_dict,
182
+ "comment": self.comments_dict,
183
+ "generated_tokens": self.generated_tokens,
184
+ }
185
+
186
+ def check_expression(self, expression) -> None:
187
+ for key in self.problem_keys:
188
+ if key in expression and not key + "." in expression:
189
+ raise ValueError(
190
+ f"Key by iself {key} is problematic. Use the full key {key + '.' + key} name instead."
191
+ )
192
+ return None
193
+
194
+ def code(self):
195
+ """Return a string of code that can be used to recreate the Result object."""
196
+ raise NotImplementedError
197
+
198
+ @property
199
+ def problem_keys(self):
200
+ """Return a list of keys that are problematic."""
201
+ return self._problem_keys
202
+
203
+ def _compute_combined_dict_and_problem_keys(self) -> None:
204
+ combined = {}
205
+ problem_keys = []
206
+ for key, sub_dict in self.sub_dicts.items():
207
+ combined.update(sub_dict)
208
+ # in some cases, the sub_dict might have keys that conflict with the main dict
209
+ if key in combined:
210
+ # The key is already in the combined dict
211
+ problem_keys = problem_keys + [key]
212
+
213
+ combined.update({key: sub_dict})
214
+ # I *think* this allows us to do do things like "answer.how_feelling" i.e., that the evaluator can use
215
+ # dot notation to access the subdicts.
216
+ self._combined_dict = combined
217
+ self._problem_keys = problem_keys
218
+
219
+ @property
220
+ def combined_dict(self) -> dict[str, Any]:
221
+ """Return a dictionary that includes all sub_dicts, but also puts the key-value pairs in each sub_dict as a key_value pair in the combined dictionary.
222
+
223
+ >>> r = Result.example()
224
+ >>> r.combined_dict['how_feeling']
225
+ 'OK'
226
+ """
227
+ if self._combined_dict is None or self._problem_keys is None:
228
+ self._compute_combined_dict_and_problem_keys()
229
+ return self._combined_dict
230
+
231
+ @property
232
+ def problem_keys(self):
233
+ """Return a list of keys that are problematic."""
234
+ if self._combined_dict is None or self._problem_keys is None:
235
+ self._compute_combined_dict_and_problem_keys()
236
+ return self._problem_keys
237
+
238
+ def get_value(self, data_type: str, key: str) -> Any:
239
+ """Return the value for a given data type and key.
240
+
241
+ >>> r = Result.example()
242
+ >>> r.get_value("answer", "how_feeling")
243
+ 'OK'
244
+
245
+ - data types can be "agent", "scenario", "model", or "answer"
246
+ - keys are relevant attributes of the Objects the data types represent
247
+ """
248
+ return self.sub_dicts[data_type][key]
249
+
250
+ @property
251
+ def key_to_data_type(self) -> dict[str, str]:
252
+ """Return a dictionary where keys are object attributes and values are the data type (object) that the attribute is associated with.
253
+
254
+ >>> r = Result.example()
255
+ >>> r.key_to_data_type["how_feeling"]
256
+ 'answer'
257
+
258
+ """
259
+ d = {}
260
+ problem_keys = []
261
+ data_types = sorted(self.sub_dicts.keys())
262
+ for data_type in data_types:
263
+ for key in self.sub_dicts[data_type]:
264
+ if key in d:
265
+ import warnings
266
+
267
+ warnings.warn(
268
+ f"Key '{key}' of data type '{data_type}' is already in use. Renaming to {key}_{data_type}"
269
+ )
270
+ problem_keys.append((key, data_type))
271
+ key = f"{key}_{data_type}"
272
+ # raise ValueError(f"Key '{key}' is already in the dictionary")
273
+ d[key] = data_type
274
+
275
+ for key, data_type in problem_keys:
276
+ self.sub_dicts[data_type][f"{key}_{data_type}"] = self.sub_dicts[
277
+ data_type
278
+ ].pop(key)
279
+ return d
280
+
281
+ def rows(self, index) -> tuple[int, str, str, str]:
282
+ """Return a generator of rows for the Result object."""
283
+ for data_type, subdict in self.sub_dicts.items():
284
+ for key, value in subdict.items():
285
+ yield (index, data_type, key, str(value))
286
+
287
+ def leaves(self):
288
+ leaves = []
289
+ for question_name, answer in self.answer.items():
290
+ if not question_name.endswith("_comment"):
291
+ leaves.append(
292
+ {
293
+ "question": f"({question_name}): "
294
+ + str(
295
+ self.question_to_attributes[question_name]["question_text"]
296
+ ),
297
+ "answer": answer,
298
+ "comment": self.answer.get(question_name + "_comment", ""),
299
+ "scenario": repr(self.scenario),
300
+ "agent": repr(self.agent),
301
+ "model": repr(self.model),
302
+ "iteration": self.iteration,
303
+ }
304
+ )
305
+ return leaves
306
+
307
+ ###############
308
+ # Useful
309
+ ###############
310
+ def copy(self) -> Result:
311
+ """Return a copy of the Result object."""
312
+ return Result.from_dict(self.to_dict())
313
+
314
+ def __eq__(self, other) -> bool:
315
+ """Return True if the Result object is equal to another Result object.
316
+
317
+ >>> r = Result.example()
318
+ >>> r == r
319
+ True
320
+
321
+ """
322
+ return self.to_dict() == other.to_dict()
323
+
324
+ ###############
325
+ # Serialization
326
+ ###############
327
+ def to_dict(self, add_edsl_version=True) -> dict[str, Any]:
328
+ """Return a dictionary representation of the Result object.
329
+
330
+ >>> r = Result.example()
331
+ >>> r.to_dict()['scenario']
332
+ {'period': 'morning', 'edsl_version': '...', 'edsl_class_name': 'Scenario'}
333
+ """
334
+
335
+ def convert_value(value, add_edsl_version=True):
336
+ if hasattr(value, "to_dict"):
337
+ return value.to_dict(add_edsl_version=add_edsl_version)
338
+ else:
339
+ return value
340
+
341
+ d = {}
342
+ for key, value in self.items():
343
+ d[key] = convert_value(value, add_edsl_version=add_edsl_version)
344
+
345
+ if key == "prompt":
346
+ new_prompt_dict = {}
347
+ for prompt_name, prompt_obj in value.items():
348
+ new_prompt_dict[prompt_name] = (
349
+ prompt_obj
350
+ if not hasattr(prompt_obj, "to_dict")
351
+ else prompt_obj.to_dict()
352
+ )
353
+ d[key] = new_prompt_dict
354
+ if add_edsl_version:
355
+ from edsl import __version__
356
+
357
+ d["edsl_version"] = __version__
358
+ d["edsl_class_name"] = "Result"
359
+
360
+ return d
361
+
362
+ def __hash__(self):
363
+ """Return a hash of the Result object."""
364
+ from edsl.utilities.utilities import dict_hash
365
+
366
+ return dict_hash(self.to_dict(add_edsl_version=False))
367
+
368
+ @classmethod
369
+ @remove_edsl_version
370
+ def from_dict(self, json_dict: dict) -> Result:
371
+ """Return a Result object from a dictionary representation."""
372
+
373
+ from edsl import Agent
374
+ from edsl import Scenario
375
+ from edsl.language_models.LanguageModel import LanguageModel
376
+ from edsl.prompts.Prompt import Prompt
377
+
378
+ prompt_data = json_dict.get("prompt", {})
379
+ prompt_d = {}
380
+ for prompt_name, prompt_obj in prompt_data.items():
381
+ prompt_d[prompt_name] = Prompt.from_dict(prompt_obj)
382
+
383
+ result = Result(
384
+ agent=Agent.from_dict(json_dict["agent"]),
385
+ scenario=Scenario.from_dict(json_dict["scenario"]),
386
+ model=LanguageModel.from_dict(json_dict["model"]),
387
+ iteration=json_dict["iteration"],
388
+ answer=json_dict["answer"],
389
+ prompt=prompt_d, # json_dict["prompt"],
390
+ raw_model_response=json_dict.get(
391
+ "raw_model_response", {"raw_model_response": "No raw model response"}
392
+ ),
393
+ question_to_attributes=json_dict.get("question_to_attributes", None),
394
+ generated_tokens=json_dict.get("generated_tokens", {}),
395
+ comments_dict=json_dict.get("comments_dict", {}),
396
+ )
397
+ return result
398
+
399
+ def rich_print(self) -> None:
400
+ """Display an object as a table."""
401
+ # from edsl.utilities import print_dict_with_rich
402
+ from rich import print
403
+ from rich.table import Table
404
+
405
+ table = Table(title="Result")
406
+ table.add_column("Attribute", style="bold")
407
+ table.add_column("Value")
408
+
409
+ to_display = self.__dict__.copy()
410
+ data = to_display.pop("data", None)
411
+ for attr_name, attr_value in to_display.items():
412
+ if hasattr(attr_value, "rich_print"):
413
+ table.add_row(attr_name, attr_value.rich_print())
414
+ elif isinstance(attr_value, dict):
415
+ a = PromptDict(attr_value)
416
+ table.add_row(attr_name, a.rich_print())
417
+ else:
418
+ table.add_row(attr_name, repr(attr_value))
419
+ return table
420
+
421
+ def __repr__(self):
422
+ """Return a string representation of the Result object."""
423
+ return f"Result(agent={repr(self.agent)}, scenario={repr(self.scenario)}, model={repr(self.model)}, iteration={self.iteration}, answer={repr(self.answer)}, prompt={repr(self.prompt)})"
424
+
425
+ @classmethod
426
+ def example(cls):
427
+ """Return an example Result object."""
428
+ from edsl.results.Results import Results
429
+
430
+ return Results.example()[0]
431
+
432
+ def score(self, scoring_function: Callable) -> Any:
433
+ """Score the result using a passed-in scoring function.
434
+
435
+ >>> def f(status): return 1 if status == 'Joyful' else 0
436
+ >>> Result.example().score(f)
437
+ 1
438
+ """
439
+ import inspect
440
+
441
+ signature = inspect.signature(scoring_function)
442
+ params = {}
443
+ for k, v in signature.parameters.items():
444
+ if k in self.combined_dict:
445
+ params[k] = self.combined_dict[k]
446
+ elif v.default is not v.empty:
447
+ params[k] = v.default
448
+ else:
449
+ raise ValueError(f"Parameter {k} not found in Result object")
450
+ return scoring_function(**params)
451
+
452
+
453
+ if __name__ == "__main__":
454
+ import doctest
455
+
456
+ doctest.testmod(optionflags=doctest.ELLIPSIS)