PyPI - edsl - Versions diffs - 0.1.31.dev4__py3-none-any.whl → 0.1.32__py3-none-any.whl - Mend

edsl 0.1.31.dev4py3-none-any.whl → 0.1.32py3-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 (50) hide show

edsl/__version__.py +1 -1
edsl/agents/Invigilator.py +3 -4
edsl/agents/PromptConstructionMixin.py +35 -15
edsl/config.py +11 -1
edsl/conjure/Conjure.py +6 -0
edsl/data/CacheHandler.py +3 -4
edsl/enums.py +4 -0
edsl/exceptions/general.py +10 -8
edsl/inference_services/AwsBedrock.py +110 -0
edsl/inference_services/AzureAI.py +197 -0
edsl/inference_services/DeepInfraService.py +4 -3
edsl/inference_services/GroqService.py +3 -4
edsl/inference_services/InferenceServicesCollection.py +13 -8
edsl/inference_services/OllamaService.py +18 -0
edsl/inference_services/OpenAIService.py +23 -18
edsl/inference_services/models_available_cache.py +31 -0
edsl/inference_services/registry.py +13 -1
edsl/jobs/Jobs.py +100 -19
edsl/jobs/buckets/TokenBucket.py +12 -4
edsl/jobs/interviews/Interview.py +31 -9
edsl/jobs/interviews/InterviewExceptionEntry.py +101 -0
edsl/jobs/interviews/InterviewTaskBuildingMixin.py +49 -34
edsl/jobs/interviews/interview_exception_tracking.py +68 -10
edsl/jobs/runners/JobsRunnerAsyncio.py +36 -15
edsl/jobs/runners/JobsRunnerStatusMixin.py +81 -51
edsl/jobs/tasks/TaskCreators.py +1 -1
edsl/jobs/tasks/TaskHistory.py +145 -1
edsl/language_models/LanguageModel.py +58 -43
edsl/language_models/registry.py +2 -2
edsl/questions/QuestionBudget.py +0 -1
edsl/questions/QuestionCheckBox.py +0 -1
edsl/questions/QuestionExtract.py +0 -1
edsl/questions/QuestionFreeText.py +2 -9
edsl/questions/QuestionList.py +0 -1
edsl/questions/QuestionMultipleChoice.py +1 -2
edsl/questions/QuestionNumerical.py +0 -1
edsl/questions/QuestionRank.py +0 -1
edsl/results/DatasetExportMixin.py +33 -3
edsl/scenarios/Scenario.py +14 -0
edsl/scenarios/ScenarioList.py +216 -13
edsl/scenarios/ScenarioListExportMixin.py +15 -4
edsl/scenarios/ScenarioListPdfMixin.py +3 -0
edsl/surveys/Rule.py +5 -2
edsl/surveys/Survey.py +84 -1
edsl/surveys/SurveyQualtricsImport.py +213 -0
edsl/utilities/utilities.py +31 -0
{edsl-0.1.31.dev4.dist-info → edsl-0.1.32.dist-info}/METADATA +4 -1
{edsl-0.1.31.dev4.dist-info → edsl-0.1.32.dist-info}/RECORD +50 -45
{edsl-0.1.31.dev4.dist-info → edsl-0.1.32.dist-info}/LICENSE +0 -0
{edsl-0.1.31.dev4.dist-info → edsl-0.1.32.dist-info}/WHEEL +0 -0

edsl/jobs/tasks/TaskHistory.py CHANGED Viewed

@@ -11,6 +11,8 @@ class TaskHistory:
         [Interview.exceptions, Interview.exceptions, Interview.exceptions, ...]
+        >>> _ = TaskHistory.example()
+        ...
         """
         self.total_interviews = interviews
@@ -18,8 +20,26 @@ class TaskHistory:
         self._interviews = {index: i for index, i in enumerate(self.total_interviews)}
+    @classmethod
+    def example(cls):
+        from edsl.jobs.interviews.Interview import Interview
+        from edsl.jobs.Jobs import Jobs
+        j = Jobs.example(throw_exception_probability=1, test_model=True)
+        from edsl.config import CONFIG
+        results = j.run(print_exceptions=False, skip_retry=True, cache=False)
+        return cls(results.task_history.total_interviews)
     @property
     def exceptions(self):
+        """
+        >>> len(TaskHistory.example().exceptions)
+        4
+        """
         return [i.exceptions for k, i in self._interviews.items() if i.exceptions != {}]
     @property
@@ -42,7 +62,12 @@ class TaskHistory:
     @property
     def has_exceptions(self) -> bool:
-        """Return True if there are any exceptions."""
+        """Return True if there are any exceptions.
+        >>> TaskHistory.example().has_exceptions
+        True
+        """
         return len(self.exceptions) > 0
     def _repr_html_(self):
@@ -216,6 +241,47 @@ class TaskHistory:
         }
         """
+    @property
+    def exceptions_by_type(self) -> dict:
+        """Return a dictionary of exceptions by type."""
+        exceptions_by_type = {}
+        for interview in self.total_interviews:
+            for question_name, exceptions in interview.exceptions.items():
+                for exception in exceptions:
+                    exception_type = exception["exception"]
+                    if exception_type in exceptions_by_type:
+                        exceptions_by_type[exception_type] += 1
+                    else:
+                        exceptions_by_type[exception_type] = 1
+        return exceptions_by_type
+    @property
+    def exceptions_by_question_name(self) -> dict:
+        """Return a dictionary of exceptions tallied by question name."""
+        exceptions_by_question_name = {}
+        for interview in self.total_interviews:
+            for question_name, exceptions in interview.exceptions.items():
+                if question_name not in exceptions_by_question_name:
+                    exceptions_by_question_name[question_name] = 0
+                exceptions_by_question_name[question_name] += len(exceptions)
+        for question in self.total_interviews[0].survey.questions:
+            if question.question_name not in exceptions_by_question_name:
+                exceptions_by_question_name[question.question_name] = 0
+        return exceptions_by_question_name
+    @property
+    def exceptions_by_model(self) -> dict:
+        """Return a dictionary of exceptions tallied by model and question name."""
+        exceptions_by_model = {}
+        for interview in self.total_interviews:
+            model = interview.model
+            if model not in exceptions_by_model:
+                exceptions_by_model[model.model] = 0
+            if interview.exceptions != {}:
+                exceptions_by_model[model.model] += len(interview.exceptions)
+        return exceptions_by_model
     def html(
         self,
         filename: Optional[str] = None,
@@ -236,6 +302,8 @@ class TaskHistory:
         if css is None:
             css = self.css()
+        models_used = set([i.model for index, i in self._interviews.items()])
         template = Template(
             """
         <!DOCTYPE html>
@@ -249,6 +317,69 @@ class TaskHistory:
         </style>
         </head>
         <body>
+            <h1>Overview</h1>
+            <p>There were {{ interviews|length }} total interviews. The number of interviews with exceptions was {{ num_exceptions }}.</p>
+            <p>The models used were: {{ models_used }}.</p>
+            <p>For documentation on dealing with exceptions on Expected Parrot,
+            see <a href="https://docs.expectedparrot.com/en/latest/exceptions.html">here</a>.</p>
+            <h2>Exceptions by Type</h2>
+            <table>
+                <thead>
+                    <tr>
+                        <th>Exception Type</th>
+                        <th>Number</th>
+                    </tr>
+                </thead>
+                <tbody>
+                    {% for exception_type, exceptions in exceptions_by_type.items() %}
+                        <tr>
+                            <td>{{ exception_type }}</td>
+                            <td>{{ exceptions }}</td>
+                        </tr>
+                    {% endfor %}
+                </tbody>
+            </table>
+            <h2>Exceptions by Model</h2>
+            <table>
+                <thead>
+                    <tr>
+                        <th>Model</th>
+                        <th>Number</th>
+                    </tr>
+                </thead>
+                <tbody>
+                    {% for model, exceptions in exceptions_by_model.items() %}
+                        <tr>
+                            <td>{{ model }}</td>
+                            <td>{{ exceptions }}</td>
+                        </tr>
+                    {% endfor %}
+                </tbody>
+            </table>
+            <h2>Exceptions by Question Name</h2>
+            <table>
+                <thead>
+                    <tr>
+                        <th>Question Name</th>
+                        <th>Number of Exceptions</th>
+                    </tr>
+                </thead>
+                <tbody>
+                    {% for question_name, exception_count in exceptions_by_question_name.items() %}
+                        <tr>
+                            <td>{{ question_name }}</td>
+                            <td>{{ exception_count }}</td>
+                        </tr>
+                    {% endfor %}
+                </tbody>
+            </table>
             {% for index, interview in interviews.items() %}
                 {% if interview.exceptions != {} %}
                    <div class="interview">Interview: {{ index }} </div>
@@ -296,11 +427,18 @@ class TaskHistory:
         """
         )
+        # breakpoint()
         # Render the template with data
         output = template.render(
             interviews=self._interviews,
             css=css,
+            num_exceptions=len(self.exceptions),
             performance_plot_html=performance_plot_html,
+            exceptions_by_type=self.exceptions_by_type,
+            exceptions_by_question_name=self.exceptions_by_question_name,
+            exceptions_by_model=self.exceptions_by_model,
+            models_used=models_used,
         )
         # Save the rendered output to a file
@@ -344,3 +482,9 @@ class TaskHistory:
         if return_link:
             return filename
+if __name__ == "__main__":
+    import doctest
+    doctest.testmod(optionflags=doctest.ELLIPSIS)

edsl/language_models/LanguageModel.py CHANGED Viewed

@@ -11,6 +11,7 @@ import hashlib
 from typing import Coroutine, Any, Callable, Type, List, get_type_hints
 from abc import ABC, abstractmethod
 class IntendedModelCallOutcome:
     "This is a tuple-like class that holds the response, cache_used, and cache_key."
@@ -21,7 +22,7 @@ class IntendedModelCallOutcome:
     def __iter__(self):
         """Iterate over the class attributes.
         >>> a, b, c = IntendedModelCallOutcome({'answer': "yes"}, True, 'x1289')
         >>> a
         {'answer': 'yes'}
@@ -32,10 +33,11 @@ class IntendedModelCallOutcome:
     def __len__(self):
         return 3
     def __repr__(self):
         return f"IntendedModelCallOutcome(response = {self.response}, cache_used = {self.cache_used}, cache_key = '{self.cache_key}')"
 from edsl.config import CONFIG
 from edsl.utilities.decorators import sync_wrapper, jupyter_nb_handler
@@ -131,18 +133,22 @@ class LanguageModel(
     def api_token(self) -> str:
         if not hasattr(self, "_api_token"):
             key_name = service_to_api_keyname.get(self._inference_service_, "NOT FOUND")
-            self._api_token = os.getenv(key_name)
-            if (
-                self._api_token is None
-                and self._inference_service_ != "test"
-                and not self.remote
-            ):
+            if self._inference_service_ == "bedrock":
+                self._api_token = [os.getenv(key_name[0]), os.getenv(key_name[1])]
+                # Check if any of the tokens are None
+                missing_token = any(token is None for token in self._api_token)
+            else:
+                self._api_token = os.getenv(key_name)
+                missing_token = self._api_token is None
+            if missing_token and self._inference_service_ != "test" and not self.remote:
+                print("rainsing error")
                 raise MissingAPIKeyError(
                     f"""The key for service: `{self._inference_service_}` is not set.
-                    Need a key with name {key_name} in your .env file.
-                    """
+                        Need a key with name {key_name} in your .env file."""
                 )
-        return self._api_token
+            return self._api_token
     def __getitem__(self, key):
         return getattr(self, key)
@@ -159,8 +165,7 @@ class LanguageModel(
         if verbose:
             print(f"Current key is {masked}")
         return self.execute_model_call(
-            user_prompt="Hello, model!",
-            system_prompt="You are a helpful agent."
+            user_prompt="Hello, model!", system_prompt="You are a helpful agent."
         )
     def has_valid_api_key(self) -> bool:
@@ -208,8 +213,6 @@ class LanguageModel(
         """
         self._set_rate_limits(rpm=rpm, tpm=tpm)
     def _set_rate_limits(self, rpm=None, tpm=None) -> None:
         """Set the rate limits for the model.
@@ -250,14 +253,16 @@ class LanguageModel(
         >>> LanguageModel._overide_default_parameters(passed_parameter_dict={"temperature": 0.5}, default_parameter_dict={"temperature":0.9, "max_tokens": 1000})
         {'temperature': 0.5, 'max_tokens': 1000}
         """
-        #parameters = dict({})
+        # parameters = dict({})
-        return {parameter_name: passed_parameter_dict.get(parameter_name, default_value)
-                for parameter_name, default_value in default_parameter_dict.items()}
-    def __call__(self, user_prompt:str, system_prompt:str):
+        return {
+            parameter_name: passed_parameter_dict.get(parameter_name, default_value)
+            for parameter_name, default_value in default_parameter_dict.items()
+        }
+    def __call__(self, user_prompt: str, system_prompt: str):
         return self.execute_model_call(user_prompt, system_prompt)
     @abstractmethod
     async def async_execute_model_call(user_prompt: str, system_prompt: str):
         """Execute the model call and returns a coroutine.
@@ -316,8 +321,10 @@ class LanguageModel(
         data["choices[0]"]["message"]["content"].
         """
         raise NotImplementedError
-    async def _async_prepare_response(self, model_call_outcome: IntendedModelCallOutcome, cache: "Cache") -> dict:
+    async def _async_prepare_response(
+        self, model_call_outcome: IntendedModelCallOutcome, cache: "Cache"
+    ) -> dict:
         """Prepare the response for return."""
         model_response = {
@@ -327,21 +334,19 @@ class LanguageModel(
             "raw_model_response": model_call_outcome.response,
         }
-        answer_portion = self.parse_response(model_call_outcome.response)
+        answer_portion = self.parse_response(model_call_outcome.response)
         try:
             answer_dict = json.loads(answer_portion)
         except json.JSONDecodeError as e:
             # TODO: Turn into logs to generate issues
             answer_dict, success = await repair(
-                bad_json=answer_portion,
-                error_message=str(e),
-                cache=cache
+                bad_json=answer_portion, error_message=str(e), cache=cache
             )
             if not success:
                 raise Exception(
                     f"""Even the repair failed. The error was: {e}. The response was: {answer_portion}."""
                 )
         return {**model_response, **answer_dict}
     async def async_get_raw_response(
@@ -353,16 +358,18 @@ class LanguageModel(
         encoded_image=None,
     ) -> IntendedModelCallOutcome:
         import warnings
-        warnings.warn("This method is deprecated. Use async_get_intended_model_call_outcome.")
+        warnings.warn(
+            "This method is deprecated. Use async_get_intended_model_call_outcome."
+        )
         return await self._async_get_intended_model_call_outcome(
             user_prompt=user_prompt,
             system_prompt=system_prompt,
             cache=cache,
             iteration=iteration,
-            encoded_image=encoded_image
+            encoded_image=encoded_image,
         )
     async def _async_get_intended_model_call_outcome(
         self,
         user_prompt: str,
@@ -404,8 +411,8 @@ class LanguageModel(
             "iteration": iteration,
         }
         cached_response, cache_key = cache.fetch(**cache_call_params)
-        if (cache_used := cached_response is not None):
+        if cache_used := cached_response is not None:
             response = json.loads(cached_response)
         else:
             f = (
@@ -413,16 +420,24 @@ class LanguageModel(
                 if hasattr(self, "remote") and self.remote
                 else self.async_execute_model_call
             )
-            params = {"user_prompt": user_prompt, "system_prompt": system_prompt,
-            **({"encoded_image": encoded_image} if encoded_image else {})
+            params = {
+                "user_prompt": user_prompt,
+                "system_prompt": system_prompt,
+                **({"encoded_image": encoded_image} if encoded_image else {}),
             }
             response = await f(**params)
-            new_cache_key = cache.store(**cache_call_params, response=response) # store the response in the cache
-            assert new_cache_key == cache_key # should be the same
-        return IntendedModelCallOutcome(response = response, cache_used = cache_used, cache_key = cache_key)
+            new_cache_key = cache.store(
+                **cache_call_params, response=response
+            )  # store the response in the cache
+            assert new_cache_key == cache_key  # should be the same
+        return IntendedModelCallOutcome(
+            response=response, cache_used=cache_used, cache_key=cache_key
+        )
-    _get_intended_model_call_outcome = sync_wrapper(_async_get_intended_model_call_outcome)
+    _get_intended_model_call_outcome = sync_wrapper(
+        _async_get_intended_model_call_outcome
+    )
     get_raw_response = sync_wrapper(async_get_raw_response)
@@ -443,7 +458,7 @@ class LanguageModel(
         self,
         user_prompt: str,
         system_prompt: str,
-        cache: 'Cache',
+        cache: "Cache",
         iteration: int = 1,
         encoded_image=None,
     ) -> dict:
@@ -461,8 +476,8 @@ class LanguageModel(
             "system_prompt": system_prompt,
             "iteration": iteration,
             "cache": cache,
-            **({"encoded_image": encoded_image} if encoded_image else {})
-        }
+            **({"encoded_image": encoded_image} if encoded_image else {}),
+        }
         model_call_outcome = await self._async_get_intended_model_call_outcome(**params)
         return await self._async_prepare_response(model_call_outcome, cache=cache)

edsl/language_models/registry.py CHANGED Viewed

@@ -36,10 +36,10 @@ class Model(metaclass=Meta):
         from edsl.inference_services.registry import default
         registry = registry or default
         if isinstance(model_name, int):
             model_name = cls.available(name_only=True)[model_name]
         factory = registry.create_model_factory(model_name)
         return factory(*args, **kwargs)

edsl/questions/QuestionBudget.py CHANGED Viewed

@@ -25,7 +25,6 @@ class QuestionBudget(QuestionBase):
         :param question_text: The text of the question.
         :param question_options: The options for allocation of the budget sum.
         :param budget_sum: The total amount of the budget to be allocated among the options.
-        :param instructions: Instructions for the question. If not provided, the default instructions are used. To view them, run `QuestionBudget.default_instructions`.
         """
         self.question_name = question_name
         self.question_text = question_text

edsl/questions/QuestionCheckBox.py CHANGED Viewed

@@ -33,7 +33,6 @@ class QuestionCheckBox(QuestionBase):
         :param question_name: The name of the question.
         :param question_text: The text of the question.
         :param question_options: The options the respondent should select from.
-        :param instructions: Instructions for the question. If not provided, the default instructions are used. To view them, run `QuestionCheckBox.default_instructions`.
         :param min_selections: The minimum number of options that must be selected.
         :param max_selections: The maximum number of options that must be selected.
         """

edsl/questions/QuestionExtract.py CHANGED Viewed

@@ -22,7 +22,6 @@ class QuestionExtract(QuestionBase):
         :param question_text: The text of the question.
         :param question_options: The options the respondent should select from.
         :param answer_template: The template for the answer.
-        :param instructions: Instructions for the question. If not provided, the default instructions are used. To view them, run `QuestionExtract.default_instructions`.
         """
         self.question_name = question_name
         self.question_text = question_text

edsl/questions/QuestionFreeText.py CHANGED Viewed

@@ -17,21 +17,14 @@ class QuestionFreeText(QuestionBase):
         """
     )
-    def __init__(
-        self,
-        question_name: str,
-        question_text: str,
-        instructions: Optional[str] = None,
-    ):
+    def __init__(self, question_name: str, question_text: str):
         """Instantiate a new QuestionFreeText.
         :param question_name: The name of the question.
         :param question_text: The text of the question.
-        :param instructions: Instructions for the question. If not provided, the default instructions are used. To view them, run `QuestionFreeText.default_instructions`.
         """
         self.question_name = question_name
         self.question_text = question_text
-        self.instructions = instructions
     ################
     # Answer methods
@@ -79,7 +72,7 @@ def main():
     q = QuestionFreeText.example()
     q.question_text
     q.question_name
-    q.instructions
+    # q.instructions
     # validate an answer
     q._validate_answer({"answer": "I like custard"})
     # translate answer code

edsl/questions/QuestionList.py CHANGED Viewed

@@ -22,7 +22,6 @@ class QuestionList(QuestionBase):
         :param question_name: The name of the question.
         :param question_text: The text of the question.
-        :param instructions: Instructions for the question. If not provided, the default instructions are used. To view them, run `QuestionList.default_instructions`.
         :param max_list_items: The maximum number of items that can be in the answer list.
         """
         self.question_name = question_name

edsl/questions/QuestionMultipleChoice.py CHANGED Viewed

@@ -33,7 +33,6 @@ class QuestionMultipleChoice(QuestionBase):
         :param question_name: The name of the question.
         :param question_text: The text of the question.
         :param question_options: The options the agent should select from.
-        :param instructions: Instructions for the question. If not provided, the default instructions are used. To view them, run `QuestionMultipleChoice.default_instructions`.
         """
         self.question_name = question_name
         self.question_text = question_text
@@ -96,7 +95,7 @@ class QuestionMultipleChoice(QuestionBase):
             question_option_key = list(meta.find_undeclared_variables(parsed_content))[
                 0
             ]
-            #breakpoint()
+            # breakpoint()
             translated_options = scenario.get(question_option_key)
         else:
             translated_options = [

edsl/questions/QuestionNumerical.py CHANGED Viewed

@@ -26,7 +26,6 @@ class QuestionNumerical(QuestionBase):
         :param question_name: The name of the question.
         :param question_text: The text of the question.
-        :param instructions: Instructions for the question. If not provided, the default instructions are used. To view them, run `QuestionNumerical.default_instructions`.
         :param min_value: The minimum value of the answer.
         :param max_value: The maximum value of the answer.
         """

edsl/questions/QuestionRank.py CHANGED Viewed

@@ -30,7 +30,6 @@ class QuestionRank(QuestionBase):
         :param question_name: The name of the question.
         :param question_text: The text of the question.
         :param question_options: The options the respondent should select from.
-        :param instructions: Instructions for the question. If not provided, the default instructions are used. To view them, run `QuestionRank.default_instructions`.
         :param min_selections: The minimum number of options that must be selected.
         :param max_selections: The maximum number of options that must be selected.
         """

edsl/results/DatasetExportMixin.py CHANGED Viewed

@@ -27,6 +27,10 @@ class DatasetExportMixin:
         >>> d.relevant_columns(remove_prefix=True)
         ['b']
+        >>> d = Dataset([{'a':[1,2,3,4]}, {'b':[5,6,7,8]}])
+        >>> d.relevant_columns()
+        ['a', 'b']
         >>> from edsl.results import Results; Results.example().select('how_feeling', 'how_feeling_yesterday').relevant_columns()
         ['answer.how_feeling', 'answer.how_feeling_yesterday']
@@ -592,8 +596,29 @@ class DatasetExportMixin:
         if return_link:
             return filename
+    def to_docx(self, filename: Optional[str] = None, separator: str = "\n"):
+        """Export the results to a Word document.
+        :param filename: The filename to save the Word document to.
+        """
+        from docx import Document
+        doc = Document()
+        for entry in self:
+            key, values = list(entry.items())[0]
+            doc.add_paragraph(key)
+            line = separator.join(values)
+            doc.add_paragraph(line)
+        if filename is not None:
+            doc.save(filename)
+        else:
+            return doc
     def tally(
-        self, *fields: Optional[str], top_n: Optional[int] = None, output="dict"
+        self, *fields: Optional[str], top_n: Optional[int] = None, output="Dataset"
     ) -> Union[dict, "Dataset"]:
         """Tally the values of a field or perform a cross-tab of multiple fields.
@@ -601,9 +626,11 @@ class DatasetExportMixin:
         >>> from edsl.results import Results
         >>> r = Results.example()
-        >>> r.select('how_feeling').tally('answer.how_feeling')
+        >>> r.select('how_feeling').tally('answer.how_feeling', output = "dict")
         {'OK': 2, 'Great': 1, 'Terrible': 1}
-        >>> r.select('how_feeling', 'period').tally('how_feeling', 'period')
+        >>> r.select('how_feeling').tally('answer.how_feeling', output = "Dataset")
+        Dataset([{'value': ['OK', 'Great', 'Terrible']}, {'count': [2, 1, 1]}])
+        >>> r.select('how_feeling', 'period').tally('how_feeling', 'period', output = "dict")
         {('OK', 'morning'): 1, ('Great', 'afternoon'): 1, ('Terrible', 'morning'): 1, ('OK', 'afternoon'): 1}
         """
         from collections import Counter
@@ -615,6 +642,8 @@ class DatasetExportMixin:
             column.split(".")[-1] for column in self.relevant_columns()
         ]
+        # breakpoint()
         if not all(
             f in self.relevant_columns() or f in relevant_columns_without_prefix
             for f in fields
@@ -641,6 +670,7 @@ class DatasetExportMixin:
         from edsl.results.Dataset import Dataset
         if output == "dict":
+            # why did I do this?
             warnings.warn(
                 textwrap.dedent(
                     """\

edsl/scenarios/Scenario.py CHANGED Viewed

@@ -182,6 +182,19 @@ class Scenario(Base, UserDict, ScenarioImageMixin, ScenarioHtmlMixin):
                 new_scenario[key] = self[key]
         return new_scenario
+    @classmethod
+    def from_url(cls, url: str, field_name: Optional[str] = "text") -> "Scenario":
+        """Creates a scenario from a URL.
+        :param url: The URL to create the scenario from.
+        :param field_name: The field name to use for the text.
+        """
+        import requests
+        text = requests.get(url).text
+        return cls({"url": url, field_name: text})
     @classmethod
     def from_image(cls, image_path: str) -> str:
         """Creates a scenario with a base64 encoding of an image.
@@ -207,6 +220,7 @@ class Scenario(Base, UserDict, ScenarioImageMixin, ScenarioHtmlMixin):
     @classmethod
     def from_pdf(cls, pdf_path):
         import fitz  # PyMuPDF
+        from edsl import Scenario
         # Ensure the file exists
         if not os.path.exists(pdf_path):

edsl 0.1.31.dev4__py3-none-any.whl → 0.1.32__py3-none-any.whl

edsl 0.1.31.dev4py3-none-any.whl → 0.1.32py3-none-any.whl