edsl 0.1.33__py3-none-any.whl → 0.1.33.dev1__py3-none-any.whl

edsl/surveys/SurveyExportMixin.py

@@ -2,27 +2,6 @@
 
 from typing import Union, Optional
 
-import subprocess
-import platform
-import os
-import tempfile
-
-
-def open_docx(file_path):
-    """
-    Open a docx file using the default application in a cross-platform manner.
-
-    :param file_path: str, path to the docx file
-    """
-    file_path = os.path.abspath(file_path)
-
-    if platform.system() == "Darwin":  # macOS
-        subprocess.call(("open", file_path))
-    elif platform.system() == "Windows":  # Windows
-        os.startfile(file_path)
-    else:  # linux variants
-        subprocess.call(("xdg-open", file_path))
-
 
 class SurveyExportMixin:
     """A mixin class for exporting surveys to different formats."""
@@ -46,12 +25,7 @@ class SurveyExportMixin:
         )
         return q.run().select("description").first()
 
-    def docx(
-        self,
-        return_document_object: bool = False,
-        filename: Optional[str] = None,
-        open_file: bool = False,
-    ) -> Union["Document", None]:
+    def docx(self, filename=None) -> Union["Document", None]:
         """Generate a docx document for the survey."""
         from docx import Document
 
@@ -71,55 +45,19 @@ class SurveyExportMixin:
                 if hasattr(question, "question_options"):
                     for option in getattr(question, "question_options", []):
                         doc.add_paragraph(str(option), style="ListBullet")
+        if filename:
+            doc.save(filename)
+            print("The survey has been saved to", filename)
+            return
+        return doc
 
-        if return_document_object and filename is None:
-            return doc
-
-        if filename is None:
-            with tempfile.NamedTemporaryFile(
-                "w", delete=False, suffix=".docx", dir=os.getcwd()
-            ) as f:
-                filename = f.name
-
-        doc.save(filename)
-        print("The survey has been saved to", filename)
-        if open_file:
-            open_docx(filename)
-        return
-
-    def show(self):
-        self.to_scenario_list(questions_only=False, rename=True).print(format="rich")
-
-    def to_scenario_list(
-        self, questions_only: bool = True, rename=False
-    ) -> "ScenarioList":
+    def to_scenario_list(self) -> "ScenarioList":
         from edsl import ScenarioList, Scenario
 
-        # from edsl.questions import QuestionBase
-
-        if questions_only:
-            to_iterate_over = self._questions
-        else:
-            to_iterate_over = self.recombined_questions_and_instructions()
-
-        if rename:
-            renaming_dict = {
-                "name": "identifier",
-                "question_name": "identifier",
-                "question_text": "text",
-            }
-        else:
-            renaming_dict = {}
-
         all_keys = set([])
         scenarios = ScenarioList()
-        for item in to_iterate_over:
-            d = item.to_dict()
-            if item.__class__.__name__ == "Instruction":
-                d["question_type"] = "NA / instruction"
-            for key in renaming_dict:
-                if key in d:
-                    d[renaming_dict[key]] = d.pop(key)
+        for q in self._questions:
+            d = q.to_dict()
             all_keys.update(d.keys())
             scenarios.append(Scenario(d))
 

edsl/surveys/SurveyFlowVisualizationMixin.py

@@ -1,6 +1,5 @@
 """A mixin for visualizing the flow of a survey."""
 
-from typing import Optional
 from edsl.surveys.base import RulePriority, EndOfSurvey
 import tempfile
 
@@ -8,7 +7,7 @@ import tempfile
 class SurveyFlowVisualizationMixin:
     """A mixin for visualizing the flow of a survey."""
 
-    def show_flow(self, filename: Optional[str] = None):
+    def show_flow(self, filename: str = None):
         """Create an image showing the flow of users through the survey."""
         # Create a graph object
         import pydot

edsl/surveys/SurveyQualtricsImport.py

@@ -9,7 +9,7 @@ qualtrics_codes = {
     "TE": "free_text",
     "MC": "multiple_choice",
     "Matrix": "matrix",
-    "DB": "instruction",  # not quite right, but for now
+    "DB": "free_text",  # not quite right, but for now
     "Timing": "free_text",  # not quite right, but for now
 }
 # TE (Text Entry): Allows respondents to input a text response.
@@ -84,11 +84,6 @@ class QualtricsQuestion:
         return None
 
     def to_edsl(self):
-        if self.question_type == "instruction":
-            from edsl import Instruction
-
-            return [Instruction(text=self.question_text, name=self.question_name)]
-
         if self.question_type == "free_text":
             try:
                 q = Question(
@@ -192,14 +187,11 @@ class SurveyQualtricsImport:
             questions.extend(qualtrics_questions.to_edsl())
         return Survey(questions)
 
-    @property
-    def survey_data(self):
+    def extract_questions_from_json(self):
         with open(self.qsf_file_name, "r") as f:
             survey_data = json.load(f)
-        return survey_data
 
-    def extract_questions_from_json(self):
-        questions = self.survey_data["SurveyElements"]
+        questions = survey_data["SurveyElements"]
 
         extracted_questions = []
 
@@ -209,74 +201,11 @@ class SurveyQualtricsImport:
 
         return extracted_questions
 
-    def extract_blocks_from_json(self):
-        blocks = []
-
-        for element in self.survey_data["SurveyElements"]:
-            if element["Element"] == "BL":
-                for block_payload in element["Payload"]:
-                    block_elements = [
-                        BlockElement(be["Type"], be["QuestionID"])
-                        for be in block_payload["BlockElements"]
-                    ]
-                    options_data = block_payload.get("Options", {})
-                    options = BlockOptions(
-                        options_data.get("BlockLocking", "false"),
-                        options_data.get("RandomizeQuestions", "false"),
-                        options_data.get("BlockVisibility", "Collapsed"),
-                    )
-
-                    block = SurveyBlock(
-                        block_payload["Type"],
-                        block_payload["Description"],
-                        block_payload["ID"],
-                        block_elements,
-                        options,
-                    )
-                    blocks.append(block)
-
-        return blocks
-
-
-class SurveyBlock:
-    def __init__(self, block_type, description, block_id, block_elements, options):
-        self.block_type = block_type
-        self.description = description
-        self.block_id = block_id
-        self.block_elements = block_elements
-        self.options = options
-
-    def __repr__(self):
-        return f"SurveyBlock(type={self.block_type}, description={self.description}, id={self.block_id})"
-
-
-class BlockElement:
-    def __init__(self, element_type, question_id):
-        self.element_type = element_type
-        self.question_id = question_id
-
-    def __repr__(self):
-        return f"BlockElement(type={self.element_type}, question_id={self.question_id})"
-
-
-class BlockOptions:
-    def __init__(self, block_locking, randomize_questions, block_visibility):
-        self.block_locking = block_locking
-        self.randomize_questions = randomize_questions
-        self.block_visibility = block_visibility
-
-    def __repr__(self):
-        return (
-            f"BlockOptions(block_locking={self.block_locking}, "
-            f"randomize_questions={self.randomize_questions}, "
-            f"block_visibility={self.block_visibility})"
-        )
-
 
 if __name__ == "__main__":
     survey_creator = SurveyQualtricsImport("example.qsf")
     # print(survey_creator.question_data)
-    # survey = survey_creator.create_survey()
+    survey = survey_creator.create_survey()
     # info = survey.push()
     # print(info)
     # questions = survey.extract_questions_from_json()

edsl/utilities/gcp_bucket/simple_example.py

@@ -0,0 +1,9 @@
+from edsl.utilities.gcp_bucket.cloud_storage import CloudStorageManager
+
+manager = CloudStorageManager()
+
+# Download Process
+file_name = "GSS2022.dta"  # Name for the downloaded file
+save_name = "test.dta"
+
+manager.download_file(file_name, save_name)

edsl/utilities/utilities.py

@@ -20,14 +20,6 @@ from html import escape
 from typing import Callable, Union
 
 
-class CustomEncoder(json.JSONEncoder):
-    def default(self, obj):
-        try:
-            return json.JSONEncoder.default(self, obj)
-        except TypeError:
-            return str(obj)
-
-
 def time_it(func):
     @wraps(func)
     def wrapper(*args, **kwargs):
@@ -132,7 +124,7 @@ def data_to_html(data, replace_new_lines=False):
     from pygments.formatters import HtmlFormatter
     from IPython.display import HTML
 
-    json_str = json.dumps(data, indent=4, cls=CustomEncoder)
+    json_str = json.dumps(data, indent=4)
     formatted_json = highlight(
         json_str,
         JsonLexer(),

{edsl-0.1.33.dist-info → edsl-0.1.33.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: edsl
-Version: 0.1.33
+Version: 0.1.33.dev1
 Summary: Create and analyze LLM-based surveys
 Home-page: https://www.expectedparrot.com/
 License: MIT
@@ -23,11 +23,9 @@ Requires-Dist: black[jupyter] (>=24.4.2,<25.0.0)
 Requires-Dist: boto3 (>=1.34.161,<2.0.0)
 Requires-Dist: groq (>=0.9.0,<0.10.0)
 Requires-Dist: jinja2 (>=3.1.2,<4.0.0)
-Requires-Dist: json-repair (>=0.28.4,<0.29.0)
 Requires-Dist: jupyter (>=1.0.0,<2.0.0)
 Requires-Dist: markdown2 (>=2.4.11,<3.0.0)
 Requires-Dist: matplotlib (>=3.8,<3.9)
-Requires-Dist: mistralai (>=1.0.2,<2.0.0)
 Requires-Dist: nest-asyncio (>=1.5.9,<2.0.0)
 Requires-Dist: numpy (>=1.22,<2.0)
 Requires-Dist: openai (>=1.4.0,<2.0.0)
@@ -45,7 +43,6 @@ Requires-Dist: setuptools (<72.0)
 Requires-Dist: simpleeval (>=0.9.13,<0.10.0)
 Requires-Dist: sqlalchemy (>=2.0.23,<3.0.0)
 Requires-Dist: tenacity (>=8.2.3,<9.0.0)
-Requires-Dist: urllib3 (>=1.25.4,<1.27)
 Project-URL: Documentation, https://docs.expectedparrot.com
 Description-Content-Type: text/markdown
 
@@ -70,7 +67,7 @@ A quick example:
 
 ```python
 # Import a question type
-from edsl import QuestionMultipleChoice
+from edsl.questions import QuestionMultipleChoice
 
 # Construct a question using the question type template
 q = QuestionMultipleChoice(

edsl-0.1.33.dev1.dist-info/RECORD

@@ -0,0 +1,209 @@
+edsl/Base.py,sha256=ttNxUotSd9LSEJl2w6LdMtT78d0nMQvYDJ0q4JkqBfg,8945
+edsl/BaseDiff.py,sha256=RoVEh52UJs22yMa7k7jv8se01G62jJNWnBzaZngo-Ug,8260
+edsl/__init__.py,sha256=E6PkWI_owu8AUc4uJs2XWDVozqSbcRWzsIqf8_Kskho,1631
+edsl/__version__.py,sha256=X6lFB08-Y7atQgG3fk8iymIvdiZo3YA8OnNvAv2d2Zc,28
+edsl/agents/Agent.py,sha256=qNJsQkN6HuTKqJrQbuUEgRX3Wo7Dwukle0oNWPi0UIE,27191
+edsl/agents/AgentList.py,sha256=_MsdeOEgaANAceLIXwuLC22mwlBn0ruGX4GEqz8_SSY,9467
+edsl/agents/Invigilator.py,sha256=8nv98bjfal6Q-GVmsxj5Isqn-GLqXDXEijSai3KAbgQ,10923
+edsl/agents/InvigilatorBase.py,sha256=ncha1HF2V1Dz4f50Gekg6AzUXCD2Af82ztfSJZbgOHY,7469
+edsl/agents/PromptConstructionMixin.py,sha256=rC86neoY1-u5cIBozQalvVR8GUS6AMx76sSia8oz2Eg,17320
+edsl/agents/__init__.py,sha256=a3H1lxDwu9HR8fwh79C5DgxPSFv_bE2rzQ6y1D8Ba5c,80
+edsl/agents/descriptors.py,sha256=m8ND3-2-JbgNX1HGakBNLIeemwsgYa1mQxYO9GW33hw,2934
+edsl/base/Base.py,sha256=DShsfI6A2ojg42muPFpVtUgTX33pnqT5vtN0SRlr-9Q,8866
+edsl/config.py,sha256=CVdmS8L7nNAas_N9kLOVDUXqsKKLSUwvS62eFHd4xbw,6245
+edsl/conjure/AgentConstructionMixin.py,sha256=qLLYJH02HotVwBUYzaHs4QW0t5nsC3NX15qxQLzEwLI,5702
+edsl/conjure/Conjure.py,sha256=JaCuAm3rmqjh11_X8PXgvPsVHGql3yTn9JEzVlzOUVU,2019
+edsl/conjure/InputData.py,sha256=RdFgkzzayddSTd6vcwBB8LC796YBj94pciwrQx9nBtg,22003
+edsl/conjure/InputDataCSV.py,sha256=m4hHSQogxt7LNcGNxcGRkrbdGy8pIQu5KvXhIEt9i6k,1684
+edsl/conjure/InputDataMixinQuestionStats.py,sha256=Jav5pqYCLSQ1pAbGV8O9VCBu0vL8Q6pteSEhRKrONuw,6256
+edsl/conjure/InputDataPyRead.py,sha256=phGmzVi06jdbcxgngAp_lEawGkJr5dAC2B4ho5IrAy4,3064
+edsl/conjure/InputDataSPSS.py,sha256=lv7NK8lpUfjsWh8h9MpqQrCjJX6Oyg3tIOrXcQYcS_4,223
+edsl/conjure/InputDataStata.py,sha256=WbA7oUGHgHJgugKjakXzpFfzWxTI4bXGGSZLxM1-tEM,224
+edsl/conjure/QuestionOptionMixin.py,sha256=RSLZ7AmA65MfwyxKKjqprR9FdkLP6MsCcmTbJoqu-MM,2870
+edsl/conjure/QuestionTypeMixin.py,sha256=L2vU207aeezlI1OuJ4I4SdtFCWBZzE-mBk7VwnTghO8,892
+edsl/conjure/RawQuestion.py,sha256=lo8k1aln4RfnfeS6OukzwC7P_cRe2LVZxcL6uwmim6A,2116
+edsl/conjure/SurveyResponses.py,sha256=fdJNFasSXVf7S5KCsYE9WB6w17q7VRylzw0aD5zGO_M,191
+edsl/conjure/__init__.py,sha256=1lPYFjV73GzYYSXiTyxopM4nKcXVHumEEo0fe06DbMo,535
+edsl/conjure/examples/placeholder.txt,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edsl/conjure/naming_utilities.py,sha256=8uoGaCPZQKLwz2HudtsFSovivTGumQrFYxXxck5WUZQ,4964
+edsl/conjure/utilities.py,sha256=yRdOJx9KIpWXMx41Bbfysx7Zd4v2ROwca5L4T1rmtQM,5539
+edsl/conversation/Conversation.py,sha256=NdWH62XpcF6hoaG0ScMho_c3TO7PfBnbdlppUN-j07k,7627
+edsl/conversation/car_buying.py,sha256=Quh2Q8O9YoCyTKJUy3li376QFIOcL1gX0y89w3wlSl4,1950
+edsl/conversation/mug_negotiation.py,sha256=mjvAqErD4AjN3G2za2c-X-3axOShW-zAJUeiJqTxVPA,2616
+edsl/conversation/next_speaker_utilities.py,sha256=bqr5JglCd6bdLc9IZ5zGOAsmN2F4ERiubSMYvZIG7qk,3629
+edsl/coop/__init__.py,sha256=4iZCwJSzJVyjBYk8ggGxY2kZjq9dXVT1jhyPDNyew4I,115
+edsl/coop/coop.py,sha256=p2JvhzUpjXEaiFWafCpkfsPIsH2BqV9ePAJCKsKfFOo,27138
+edsl/coop/utils.py,sha256=UZwljKYW_Yjw7RYcjOg3SW7fn1pyHQfJ1fM48TBNoss,3601
+edsl/data/Cache.py,sha256=aR9ydTFvUWiRJZd2G7SLFFtZaS2At8ArbQ1SAhYZBDs,15629
+edsl/data/CacheEntry.py,sha256=_5UiFaJQu_U-Z1_lEPt-h6Gaidp2Eunk02wOd3Ni3MQ,7252
+edsl/data/CacheHandler.py,sha256=DxbfeT2nZGRu8yQkbWr2tyEnhNiClevMsd5KZMCq2f0,4793
+edsl/data/SQLiteDict.py,sha256=V5Nfnxctgh4Iblqcw1KmbnkjtfmWrrombROSQ3mvg6A,8979
+edsl/data/__init__.py,sha256=KBNGGEuGHq--D-TlpAQmvv_If906dJc1Gsy028zOx78,170
+edsl/data/orm.py,sha256=Jz6rvw5SrlxwysTL0QI9r68EflKxeEBmf6j6himHDS8,238
+edsl/data_transfer_models.py,sha256=ORcI-g_oUE8EofLvHQKnWC5ar-IgZiXGY2bIpTGAZyQ,1157
+edsl/enums.py,sha256=Ef-3pvC-uCMPYwdUrxeCo4Xx3zoOQiimU3W_vhKCir4,5077
+edsl/exceptions/__init__.py,sha256=HVg-U-rJ0fRoG9Rws6gnK5S9B68SkPWDPsoD6KpMZ-A,1370
+edsl/exceptions/agents.py,sha256=3SORFwFbMGrF6-vAL2GrKEVdPcXo7md_k2oYufnVXHA,673
+edsl/exceptions/configuration.py,sha256=qH2sInNTndKlCLAaNgaXHyRFdKQHL7-dElB_j8wz9g4,351
+edsl/exceptions/coop.py,sha256=xDr7k_Tir6L5AxO6GMmoFyUjZ3DIenPQflpUkaTqJl0,38
+edsl/exceptions/data.py,sha256=K24CjgwFiMWxrF1Z2dF6F7Vfrge_y9kMK_wsYYSaroU,209
+edsl/exceptions/general.py,sha256=zAyJnppPjjxQAn6X3A5fetmv5FUR7kQDU58vwBKvAks,1114
+edsl/exceptions/jobs.py,sha256=sSUATmzBIN1oINWuwPExxPqIWmfCo0XYj_yR4dJzVjo,803
+edsl/exceptions/language_models.py,sha256=KDChmr2rVGa1lj57i9-QGSLYTOeguRYyF2z8FyXEplA,1017
+edsl/exceptions/prompts.py,sha256=vD_reI-RVKWYHYozenEmhmB7Rb1sIiXghgNUtbVGBUo,247
+edsl/exceptions/questions.py,sha256=9z6RI7HRH-UMrnIlt28VTj9deCSbD1Frmpqi0H6YtF0,534
+edsl/exceptions/results.py,sha256=cn6Zb86Y648ulN3RYXb52HGv1NT3ykBfE6g2Xu8TeIw,325
+edsl/exceptions/surveys.py,sha256=lADtr-tvPmUYSfRy3TdkTV5SzZfShlMgCzm-ZGYRlGk,557
+edsl/inference_services/AnthropicService.py,sha256=tjYRJRIvQ7Z6uCYdqxm5ZlVjZdVZCnHtQ6QGHT59PXs,2822
+edsl/inference_services/AwsBedrock.py,sha256=shSh0QXyQC3c0KPfUi-y74TGFm1NxTSK6x9FtXi5ptI,3971
+edsl/inference_services/AzureAI.py,sha256=9D_uZ_OakJY3JLIioCdpvX4jwerrfizLu86KDWnV_2g,8041
+edsl/inference_services/DeepInfraService.py,sha256=fWlH5sCNxf8eHPHxPPxJMEVWpCM9sDenkC8IZYqtXfA,515
+edsl/inference_services/GoogleService.py,sha256=IwSwXr7khvHjpDzgiW5PRVKMhI2wQ9D1_H1MRnHjefU,2732
+edsl/inference_services/GroqService.py,sha256=vcHiMqdo4eryg2pOk2SfTNpTnvCrx8qkIwtJjVPDnZU,433
+edsl/inference_services/InferenceServiceABC.py,sha256=H6jW2gDKTLC3xgmqiSBdX4pGY1oauEO8VqGZoB3qvnQ,1790
+edsl/inference_services/InferenceServicesCollection.py,sha256=EDyxnoSjGXhWob_ost7U8WUYjn1jgL_noB0-VlXBnOo,2810
+edsl/inference_services/OllamaService.py,sha256=oro9CRl8IUE2Ro-zE69Cr4Zaf6Gdw29XW5CFU-46E0k,498
+edsl/inference_services/OpenAIService.py,sha256=_ikUPzyG3oe57NKfcCC7W20Ktis4kK1FWZ01zBbPa28,7719
+edsl/inference_services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edsl/inference_services/models_available_cache.py,sha256=aTJAYKjntQtb95jwmEB_QXjvwclydqZIqRhrkBw9CAM,3483
+edsl/inference_services/rate_limits_cache.py,sha256=HYslviz7mxF9U4CUTPAkoyBsiXjSju-YCp4HHir6e34,1398
+edsl/inference_services/registry.py,sha256=9bPIn6EJsStsNN_3bzkBDVoOLgwAtmUKle18mZskFJs,865
+edsl/inference_services/write_available.py,sha256=NNwhATlaMp8IYY635MSx-oYxt5X15acjAfaqYCo_I1Y,285
+edsl/jobs/Answers.py,sha256=z4TADN-iHIrbMtI1jVyiaetv0OkTv768dFBpREIQC6c,1799
+edsl/jobs/Jobs.py,sha256=jnfMbz4VHdDJifGmn3wxEm1_NvYGT6L7LlD41pT9Rsk,33282
+edsl/jobs/__init__.py,sha256=aKuAyd_GoalGj-k7djOoVwEbFUE2XLPlikXaA1_8yAg,32
+edsl/jobs/buckets/BucketCollection.py,sha256=LA8DBVwMdeTFCbSDI0S2cDzfi_Qo6kRizwrG64tE8S4,1844
+edsl/jobs/buckets/ModelBuckets.py,sha256=hxw_tzc0V42CiB7mh5jIxlgwDVJ-zFZhlLtKrHEg8ho,2419
+edsl/jobs/buckets/TokenBucket.py,sha256=A4ZHDztlswRXlsdOmtAoyy9gBj3Y5rgOeKsy_U62sp4,6320
+edsl/jobs/interviews/Interview.py,sha256=6CeWoHvhDqIbhBCrNubDvinBSa7K-Rks5ilGmJNFMg4,12028
+edsl/jobs/interviews/InterviewExceptionEntry.py,sha256=N_-3_KwXTZLhGNwZSyTgUMS0lE-hikWRy2AzLOduTDQ,2772
+edsl/jobs/interviews/InterviewStatistic.py,sha256=hY5d2EkIJ96NilPpZAvZZzZoxLXM7ss3xx5MIcKtTPs,1856
+edsl/jobs/interviews/InterviewStatisticsCollection.py,sha256=_ZZ0fnZBQiIywP9Q_wWjpWhlfcPe2cn32GKut10t5RI,788
+edsl/jobs/interviews/InterviewStatusDictionary.py,sha256=MSyys4hOWe1d8gfsUvAPbcKrs8YiPnz8jpufBSJL7SU,2485
+edsl/jobs/interviews/InterviewStatusLog.py,sha256=6u0F8gf5tha39VQL-IK_QPkCsQAYVOx_IesX7TDDX_A,3252
+edsl/jobs/interviews/InterviewStatusMixin.py,sha256=VV0Pel-crUsLoGpTifeIIkXsLGj0bfuO--UtpRnH-dU,1251
+edsl/jobs/interviews/InterviewTaskBuildingMixin.py,sha256=O1XwX6QqhQ-M-b4fqy0XzopLna2A8BTrdjKcz3N0XHs,11505
+edsl/jobs/interviews/ReportErrors.py,sha256=RSzDU2rWwtjfztj7sqaMab0quCiY-X2bG3AEOxhTim8,1745
+edsl/jobs/interviews/interview_exception_tracking.py,sha256=GF5PIj601rZPa5XWx1vTbFRt662S0CEoHYpG4Wulxj8,5066
+edsl/jobs/interviews/interview_status_enum.py,sha256=KJ-1yLAHdX-p8TiFnM0M3v1tnBwkq4aMCuBX6-ytrI8,229
+edsl/jobs/interviews/retry_management.py,sha256=9Efn4B3aV45vbocnF6J5WQt88i2FgFjoi5ALzGUukEE,1375
+edsl/jobs/runners/JobsRunnerAsyncio.py,sha256=U6_tJyTIxZmHNtJw7Gn4cpAU8Pzvfq3n9s39fVjsGkc,12802
+edsl/jobs/runners/JobsRunnerStatusData.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edsl/jobs/runners/JobsRunnerStatusMixin.py,sha256=MmVK2udusAbQ3KCkQPiQYjWVDfKprBk9Ecelgui__0k,13424
+edsl/jobs/tasks/QuestionTaskCreator.py,sha256=f-hF7TMdMMjQ5AQMpMJXrFdBUNSLrLHHl1sun774f_U,10394
+edsl/jobs/tasks/TaskCreators.py,sha256=XqAbNU33378Z4PQncokbfJwnKt3KHR9aqa5fKYRDpfg,2694
+edsl/jobs/tasks/TaskHistory.py,sha256=PV1G1VdaZNV3GQ_m1L4a_7uPmrEGZw4K-hn8gJe_aDE,16220
+edsl/jobs/tasks/TaskStatusLog.py,sha256=bqH36a32F12fjX-M-4lNOhHaK2-WLFzKE-r0PxZPRjI,546
+edsl/jobs/tasks/task_management.py,sha256=KMToZuXzMlnHRHUF_VHL0-lHMTGhklf2GHVuwEEHtzA,301
+edsl/jobs/tasks/task_status_enum.py,sha256=DOyrz61YlIS8R1W7izJNphcLrJ7I_ReUlfdRmk23h0Q,5333
+edsl/jobs/tokens/InterviewTokenUsage.py,sha256=u_6-IHpGFwZ6qMEXr24-jyLVUSSp4dSs_4iAZsBv7O4,1100
+edsl/jobs/tokens/TokenUsage.py,sha256=odj2-wDNEbHl9noyFAQ0DSKV0D9cv3aDOpmXufKZ8O4,1323
+edsl/language_models/LanguageModel.py,sha256=klXlrI3Xty7EpT3MB1OUzbue38zRkvzA0vrZOLuQLmo,21569
+edsl/language_models/ModelList.py,sha256=G8tzHqzz4exc28BGvGgVRk1Xwu8EDCiVWxMC5l8VnvI,2862
+edsl/language_models/RegisterLanguageModelsMeta.py,sha256=2bvWrVau2BRo-Bb1aO-QATH8xxuW_tF7NmqBMGDOfSg,8191
+edsl/language_models/__init__.py,sha256=bvY7Gy6VkX1gSbNkRbGPS-M1kUnb0EohL0FSagaEaTs,109
+edsl/language_models/registry.py,sha256=h5eLN7lsqpK5vcEsxn3gsARb9Qyx-k3jq2MMnyuATcY,3727
+edsl/language_models/repair.py,sha256=xaNunBRpMWgceSPOzk-ugp5WXtgLuuhj23TgXUeOobw,5963
+edsl/language_models/unused/ReplicateBase.py,sha256=J1oqf7mEyyKhRwNUomnptVqAsVFYCbS3iTW0EXpKtXo,3331
+edsl/notebooks/Notebook.py,sha256=xi9xkxmkQ6-DwhqbjjMLpYKB0VJV20AtwEonJ6mnqjo,7739
+edsl/notebooks/__init__.py,sha256=VNUA3nNq04slWNbYaNrzOhQJu3AZANpvBniyCJSzJ7U,45
+edsl/prompts/Prompt.py,sha256=12cbeQTKqfVQGpd1urqKZeXiDtKz2RAJqftoXS3q-DE,10070
+edsl/prompts/QuestionInstructionsBase.py,sha256=xico6ob4cqWsHl-txj2RbY_4Ny5u9UqvIjmoTVgjUhk,348
+edsl/prompts/__init__.py,sha256=Z0ywyJfnV0i-sR1SCdK4mHhgECT5cXpHVA-pKuBTifc,85
+edsl/prompts/library/agent_instructions.py,sha256=_2kCDYvh3ZOF72VvcIH5jhmHjM6AAgQdkWrHvR52Yhg,1100
+edsl/prompts/library/agent_persona.py,sha256=jEl1LjlOP67vOPiRW0S_-TRMz3n6Tp3mPRvltM2r2_k,516
+edsl/prompts/library/question_budget.py,sha256=RAc7pmQRbs_-xksU52yYl8iqJ7HejPY4sQOlCSQLGWs,1190
+edsl/prompts/library/question_checkbox.py,sha256=NIGcdnfkJ8_XbGsdq9toeWA_phQVx5FIBOAe0mReXJw,1462
+edsl/prompts/library/question_extract.py,sha256=QxNiNBjUk25Ss_Pax0iDgmgTYXEycpWJk2z0evmqsP0,775
+edsl/prompts/library/question_freetext.py,sha256=_I7hcniRfVwmZr2wXAasfbZ0GTu41ZIKfcoixGhzZUI,510
+edsl/prompts/library/question_linear_scale.py,sha256=sQpgjSvqJU-uQti-DCxOzLzj05ENkpyxmH-qTFq39x0,778
+edsl/prompts/library/question_list.py,sha256=KYi3gtcWyDzRLyAb2-k7C-QJ9TAfbLGPkWVvVmdT6xg,731
+edsl/prompts/library/question_multiple_choice.py,sha256=_2LUM9bOInODoFyaTrKfMkcb5Z7RDj_odq5iAnD7KVQ,1617
+edsl/prompts/library/question_numerical.py,sha256=ZdWnDbTU0gQdMMqcWHw_eIHEZkJ_kuE-XWrnCYFR07w,1456
+edsl/prompts/library/question_rank.py,sha256=WDgXyph0EKWJrSgsW2eqcx3xdU-WA1LEvB2jvQFOQ8s,882
+edsl/prompts/prompt_config.py,sha256=O3Y5EvBsCeKs9m9IjXiRXOcHWlWvQV0yqsNb2oSR1ow,974
+edsl/prompts/registry.py,sha256=XOsqGsvNOmIG83jqoszqI72yNZkkszKR4FrEhwSzj_Q,8093
+edsl/questions/AnswerValidatorMixin.py,sha256=U5i79HoEHpSoevgtx68TSg8a9tELq3R8xMtYyK1L7DQ,12106
+edsl/questions/QuestionBase.py,sha256=XVB_G6ksX-rYq_mJlWlNHqaD68cJrG7iFymo7rWKGaU,19066
+edsl/questions/QuestionBudget.py,sha256=aaRvSxBDzrQS8o4RBGsALZkSnKqGVJcHkDeEPAS2pd0,5974
+edsl/questions/QuestionCheckBox.py,sha256=d-79SmqZgfPEHCMCzIdjDQJ2JSKJbFxEez05iguY3Ro,5958
+edsl/questions/QuestionExtract.py,sha256=QxVGhgJHtxaPCOS0yqPdO4lsF5SIjjrIBlaiGXA0KjE,3818
+edsl/questions/QuestionFreeText.py,sha256=b4V2lbj6L9AeIOMNlqrrUtZbuhxuUkbMum3NXaBOPZU,3004
+edsl/questions/QuestionFunctional.py,sha256=jlC1eNE-kpp9o5CXKo-c3Re4PIq1_WmdJ66u9nD-W7w,4967
+edsl/questions/QuestionList.py,sha256=jEOEnhlAz1rNc9uHzx8kTO5EkpSU3SDWcXtvoEmLIj8,3884
+edsl/questions/QuestionMultipleChoice.py,sha256=7Lmy2gQ4ejeFenNQcaVXjE_TVyNAs3nizvn0zIS_d38,6455
+edsl/questions/QuestionNumerical.py,sha256=dQkobQTXkmMUY4f-lSVAiTS33009cDy5qFCM1gJeCbs,3459
+edsl/questions/QuestionRank.py,sha256=d_r9KE0v3PQs00ujIaloZZIb1FdcN6Wuak1og_0yNEQ,5709
+edsl/questions/RegisterQuestionsMeta.py,sha256=unON0CKpW-mveyhg9V3_BF_GYYwytMYP9h2ZZPetVNM,1994
+edsl/questions/SimpleAskMixin.py,sha256=YRLiHA6TPMKyLWvdAi7Mfnxcqtw7Kem6tH28YNb8n4U,2227
+edsl/questions/__init__.py,sha256=U7t2dk_dHSaFRciHETXaGaJRnCobaik3mIKCpGRUuJo,1160
+edsl/questions/compose_questions.py,sha256=ZcLkwNaofk-o0-skGXEDGZAj6DumOhVhyIUjqEnKrbg,3380
+edsl/questions/derived/QuestionLikertFive.py,sha256=zKkjKI3HSt9ZGyBBNgXNsfXZ7gOoOknidLDPgMN_PcI,2475
+edsl/questions/derived/QuestionLinearScale.py,sha256=7tybIaMaDTMkTFC6CymusW69so-zf5jUn8Aqf5pw__Y,2673
+edsl/questions/derived/QuestionTopK.py,sha256=TouXKZt_h6Jd-2rAjkEOCJOzzei4Wa-3hjYq9CLxWws,2744
+edsl/questions/derived/QuestionYesNo.py,sha256=KtMGuAPYWv-7-9WGa0fIS2UBxf6KKjFpuTk_h8FPZbg,2076
+edsl/questions/derived/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edsl/questions/descriptors.py,sha256=W0_mGZKKiXv2E2BKIy8-4n_QDILh9sun85Oev1YRoLs,14791
+edsl/questions/question_registry.py,sha256=ZD7Y_towDdlnnmLq12vVewgQ3fEk9Ur0tCTWK8-WqeQ,5241
+edsl/questions/settings.py,sha256=er_z0ZW_dgmC5CHLWkaqBJiuWgAYzIund85M5YZFQAI,291
+edsl/results/Dataset.py,sha256=DZgb3vIj69ON7APQ6DimjBwAS1xZvZiXOg68CjW9E3I,8662
+edsl/results/DatasetExportMixin.py,sha256=6Ky0CPxOOj6Og5nMMBQ-bhgJuzC6Ch_vHIquBbc_IdQ,25981
+edsl/results/Result.py,sha256=53lz1mGqmnt2wHl-Ccimbla7cNlg_mLUgOBQa6Qd19k,14433
+edsl/results/Results.py,sha256=IUHfmcBVtZ3me4VvVBQsdKKtuMXaZtLkzIrH5US7aUY,38155
+edsl/results/ResultsDBMixin.py,sha256=Vs95zbSB4G7ENY4lU7OBdekg9evwTrtPH0IIL2NAFTk,7936
+edsl/results/ResultsExportMixin.py,sha256=XizBsPNxziyffirMA4kS7UHpYM1WIE4s1K-B7TqTfDw,1266
+edsl/results/ResultsFetchMixin.py,sha256=VEa0TKDcXbnTinSKs9YaE4WjOSLmlp9Po1_9kklFvSo,848
+edsl/results/ResultsGGMixin.py,sha256=SAYz8p4wb1g8x6KhBVz9NHOGib2c2XsqtTclpADrFeM,4344
+edsl/results/ResultsToolsMixin.py,sha256=mseEFxJCf9sjXdIxpjITt_UZBwdXxw2o2VLg5jMrALA,3017
+edsl/results/__init__.py,sha256=2YcyiVtXi-3vIV0ZzOy1PqBLm2gaziufJVi4fdNrAt8,80
+edsl/scenarios/FileStore.py,sha256=AevvU1qdLSmL4Q7cb1PhUiaJk1i5T0sgkTKBYf0KDN4,8722
+edsl/scenarios/Scenario.py,sha256=VpQzY3169Z18ZoNOu_pjIf66TXngIVFZHvQXUJeRdQg,15254
+edsl/scenarios/ScenarioHtmlMixin.py,sha256=EmugmbPJYW5eZS30rM6pDMDQD9yrrvHjmgZWB1qBfq4,1882
+edsl/scenarios/ScenarioImageMixin.py,sha256=VJ5FqyPrL5-ieORlWMpnjmOAFIau8QFZCIZyEBKgb6I,3530
+edsl/scenarios/ScenarioList.py,sha256=Atqjcrvhgvw0r25AimK2HEsuv7kCRjutZQwIb2DQkwI,27214
+edsl/scenarios/ScenarioListExportMixin.py,sha256=nuBrSlkNogfVZzIV2vJai216ebWH3iJxiLJAJ_n0xg8,1410
+edsl/scenarios/ScenarioListPdfMixin.py,sha256=dt7GzC2UZZpQjjilS7_eBm0BmqU6bDWWbBP7ejkhgHM,3830
+edsl/scenarios/__init__.py,sha256=KRwZCLf2R0qyJvv1NGbd8M51Bt6Ia6Iylg-Xq_2Fa6M,98
+edsl/shared.py,sha256=lgLa-mCk2flIhxXarXLtfXZjXG_6XHhC2A3O8yRTjXc,20
+edsl/study/ObjectEntry.py,sha256=e3xRPH8wCN8Pum5HZsQRYgnSoauSvjXunIEH79wu5A8,5788
+edsl/study/ProofOfWork.py,sha256=FaqYtLgeiTEQXWKukPgPUTWMcIN5t1FR7h7Re8QEtgc,3433
+edsl/study/SnapShot.py,sha256=-5zoP4uTvnqtu3zRNMD-fKsNAVYX9psoKRADfotsF9E,2439
+edsl/study/Study.py,sha256=zLy2mMvsX_QgZ6D4dcgYJoEacyajkRnARYjIFvyCO1o,16939
+edsl/study/__init__.py,sha256=YAvPLTPG3hK_eN9Ar3d1_d-E3laXpSya879A25-JAxU,170
+edsl/surveys/DAG.py,sha256=ozQuHo9ZQ8Eet5nDXtp7rFpiSocvvfxIHtyTnztvodg,2380
+edsl/surveys/Memory.py,sha256=-ikOtkkQldGB_BkPCW3o7AYwV5B_pIwlREw7aVCSHaQ,1113
+edsl/surveys/MemoryPlan.py,sha256=BeLuqS5Q8G2jSluHYFCAxVmj7cNPK-rDQ3mUsuDjikQ,7979
+edsl/surveys/Rule.py,sha256=FnTCEWlbHDH_wOmnN1fiuqwrJzEEhPG4jibLXhRq03M,11104
+edsl/surveys/RuleCollection.py,sha256=sN7aYDQJG3HmE-WxohgpctcQbHewjwE6NAqEVTxvFP8,13359
+edsl/surveys/Survey.py,sha256=USU3cxJtflOAmDyxAljUUd45vU3j9XWtgSsBTZYkCaY,51355
+edsl/surveys/SurveyCSS.py,sha256=NjJezs2sTlgFprN6IukjGKwNYmNdXnLjzV2w5K4z4RI,8415
+edsl/surveys/SurveyExportMixin.py,sha256=vj9bZReHx0wBK9sVuS0alzPIUDdg6AFFMd7bl1RKWKI,6555
+edsl/surveys/SurveyFlowVisualizationMixin.py,sha256=Z-YqeedMqWOtCFy003YJ9aneJ1n4bn70lDoILwLtTc0,3966
+edsl/surveys/SurveyQualtricsImport.py,sha256=GG8usoPtG2_Ef-aIpwIEfCzOsjCFZHE6mtvof0cb7Z0,7238
+edsl/surveys/__init__.py,sha256=vjMYVlP95fHVqqw2FfKXRuYbTArZkZr1nK4FnXzZWzs,129
+edsl/surveys/base.py,sha256=n5PBx0BF0powzBXCsufpUekfNK_9huf3rohtU1mMCq0,1001
+edsl/surveys/descriptors.py,sha256=3B-hBVvGpLlVBCyOnPuxkLjesvpr0QIuATbggp_MJ7o,2076
+edsl/tools/__init__.py,sha256=4iRiX1K0Yh8RGwlUBuzipvFfRrofKqCRQ0SzNK_2oiQ,41
+edsl/tools/clusters.py,sha256=uvDN76bfHIHS-ykB2iioXu0gKeP_UyD7Q9ee67w_fV4,6132
+edsl/tools/embeddings.py,sha256=-mHqApiFbGzj_2Cr_VVl_7XiBBDyPB5-6ZO7IsXvaig,677
+edsl/tools/embeddings_plotting.py,sha256=fznAqLnjF_K8PkJy1C4hBRObm3f8ebDIxQzrqRXlEJM,3564
+edsl/tools/plotting.py,sha256=DbVNlm8rNCnWHXi5wDPnOviZ1Qxz-rHvtQM1zHWw1f8,3120
+edsl/tools/summarize.py,sha256=YcdB0IjZn92qv2zVJuxsHfXou810APWYKeaHknsATqM,656
+edsl/utilities/SystemInfo.py,sha256=qTP_aKwECPxtTnbEjJ7F1QqKU9U3rcEEbppg2ilQGuY,792
+edsl/utilities/__init__.py,sha256=oUWx_h-8OFb1Of2SgIQZuIu7hX_bhDuwCSwksKGWZ6k,544
+edsl/utilities/ast_utilities.py,sha256=49KDDu-PHYpzDN5cCaDf-ReQH-1dzjT5EG3WVSa8THk,868
+edsl/utilities/data/Registry.py,sha256=q2DKc7CpG_7l47MxxsSi6DJOs4p1Z3qNx7PV-v8CUOE,176
+edsl/utilities/data/__init__.py,sha256=pDjGnzC11q4Za8qX5zcg6plcQ_8Qjpb-sAKPwOlKmCY,62
+edsl/utilities/data/scooter_results.json,sha256=tRtVAI5haLMh_-wjz9it_uk_I1bGe5qdFHsIRQMi_ks,250944
+edsl/utilities/decorators.py,sha256=rlTSMItwmWUxHQBIEUDxX3lFzgtiT8PnfNavakuf-2M,2319
+edsl/utilities/gcp_bucket/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edsl/utilities/gcp_bucket/cloud_storage.py,sha256=dStHsG181YP9ALW-bkyO-sgMWuLV5aaLsnsCOVD8jJw,3472
+edsl/utilities/gcp_bucket/simple_example.py,sha256=pR_IH_Y640_-YnEyNpE7V_1MtBFC9nD3dg2NdSVcuXY,251
+edsl/utilities/interface.py,sha256=AaKpWiwWBwP2swNXmnFlIf3ZFsjfsR5bjXQAW47tD-8,19656
+edsl/utilities/repair_functions.py,sha256=tftmklAqam6LOQQu_-9U44N-llycffhW8LfO63vBmNw,929
+edsl/utilities/restricted_python.py,sha256=5-_zUhrNbos7pLhDl9nr8d24auRlquR6w-vKkmNjPiA,2060
+edsl/utilities/utilities.py,sha256=2KwKc1J0lu6_CjIAKaufvfLOJ4j3XGKyClwJ6TjKLxM,11157
+edsl-0.1.33.dev1.dist-info/LICENSE,sha256=_qszBDs8KHShVYcYzdMz3HNMtH-fKN_p5zjoVAVumFc,1111
+edsl-0.1.33.dev1.dist-info/METADATA,sha256=7-OQHBeU2oBSZ54VSVS8JDM3n4iMWSO0eoE5s_Ajq2s,4306
+edsl-0.1.33.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+edsl-0.1.33.dev1.dist-info/RECORD,,

edsl/TemplateLoader.py

@@ -1,24 +0,0 @@
-from importlib import resources
-from jinja2 import BaseLoader, TemplateNotFound
-import os
-
-
-class TemplateLoader(BaseLoader):
-    def __init__(self, package_name, templates_dir):
-        self.package_name = package_name
-        self.templates_dir = templates_dir
-
-    def get_source(self, environment, template):
-        try:
-            parts = [self.templates_dir] + template.split("/")
-            template_path = os.path.join(*parts)
-
-            # Use resources.files() to get a Traversable object
-            templates = resources.files(self.package_name).joinpath(self.templates_dir)
-
-            # Use the read_text() method of the Traversable object
-            content = templates.joinpath(template).read_text()
-
-            return content, None, lambda: True
-        except FileNotFoundError:
-            raise TemplateNotFound(template)

edsl/auto/AutoStudy.py

@@ -1,117 +0,0 @@
-from typing import Optional
-
-from edsl import Model
-from edsl.auto.StageQuestions import StageQuestions
-from edsl.auto.StagePersona import StagePersona
-from edsl.auto.StagePersonaDimensions import StagePersonaDimensions
-from edsl.auto.StagePersonaDimensionValues import StagePersonaDimensionValues
-from edsl.auto.StagePersonaDimensionValueRanges import (
-    StagePersonaDimensionValueRanges,
-)
-from edsl.auto.StageLabelQuestions import StageLabelQuestions
-from edsl.auto.StageGenerateSurvey import StageGenerateSurvey
-
-# from edsl.auto.StageBase import gen_pipeline
-
-from edsl.auto.utilities import agent_generator, create_agents, gen_pipeline
-
-
-class AutoStudy:
-    def __init__(
-        self,
-        overall_question: str,
-        population: str,
-        model: Optional["Model"] = None,
-        survey: Optional["Survey"] = None,
-        agent_list: Optional["AgentList"] = None,
-        default_num_agents=11,
-    ):
-        self.overall_question = overall_question
-        self.population = population
-        self._survey = survey
-        self._agent_list = agent_list
-        self._agent_list_generator = None
-        self._persona_mapping = None
-        self._results = None
-        self.default_num_agents = default_num_agents
-        self.model = model or Model()
-
-    @property
-    def survey(self):
-        if self._survey is None:
-            self._survey = self._create_survey()
-        return self._survey
-
-    @property
-    def persona_mapping(self):
-        if self._persona_mapping is None:
-            self._persona_mapping = self._create_persona_mapping()
-        return self._persona_mapping
-
-    @property
-    def agent_list_generator(self):
-        if self._agent_list_generator is None:
-            self._agent_list_generator = self._create_agent_list_generator()
-        return self._agent_list_generator
-
-    @property
-    def results(self):
-        if self._results is None:
-            self._results = self._create_results()
-        return self._results
-
-    def _create_survey(self):
-        survey_pipline_stages = [
-            StageQuestions,
-            StageLabelQuestions,
-            StageGenerateSurvey,
-        ]
-        survey_pipeline = gen_pipeline(survey_pipline_stages)
-        return survey_pipeline.process(
-            data=survey_pipeline.input(
-                overall_question=self.overall_question, population=self.population
-            )
-        ).survey
-
-    def _create_persona_mapping(self):
-        persona_pipeline_stages = [
-            StageQuestions,
-            StagePersona,
-            StagePersonaDimensions,
-            StagePersonaDimensionValues,
-            StagePersonaDimensionValueRanges,
-        ]
-
-        persona_pipeline = gen_pipeline(persona_pipeline_stages)
-        sample_agent_results = persona_pipeline.process(
-            persona_pipeline.input(
-                overall_question=overall_question, population=self.population
-            )
-        )
-        return sample_agent_results
-
-    def _create_agent_list_generator(self):
-        return agent_generator(
-            persona=self.persona_mapping.persona,
-            dimension_dict=self.persona_mapping.mapping,
-        )
-
-    def agent_list(self, num_agents):
-        return create_agents(
-            agent_generator=self.agent_list_generator,
-            survey=self.survey,
-            num_agents=num_agents,
-        )
-
-    def _create_results(self, num_agents=None):
-        if num_agents is None:
-            num_agents = self.default_num_agents
-        agent_list = self.agent_list(num_agents)
-        return self.survey.by(agent_list).by(self.model).run()
-
-
-if __name__ == "__main__":
-    overall_question = "Should online platforms be regulated with respect to selling electric scooters?"
-    auto_study = AutoStudy(overall_question, population="US Adults")
-
-    results = auto_study.results