+#hvddlZddlZddlmZmZddlmZddlZGddejZ GddZ y)N)ListOptional) BaseModelc~eZdZejdZejdZejdZy)DescriptionSignaturezname of a fielddesczan example value for the fieldz8a short text only description of what the field containsN) __name__ __module__ __qualname__dspy InputField field_nameexample OutputField description]/var/www/html/sandstorm/venv/lib/python3.12/site-packages/dspy/experimental/synthetic_data.pyrr s7 &78Jdoo#CDG"$""(bcKrrceZdZd deedeeejfdZde deejfdZ dZ d e deejfd Z de fd Zy) SyntheticDataGeneratorN schema_classexamplesc ||_||_y)N)rr)selfrrs r__init__zSyntheticDataGenerator.__init__s(  r sample_sizereturnc|js|js td|jr LL@ A==+. .$/3t}}3E\]$^!!??@YZ59]]t}}11ZHZZrc|jr!|jj}|d}|S|jr~|jdjd}t j t }|jDcic].}|dt||t||ji0}}|Si}|Scc}w)zDefine fields to generate if a schema class is provided. Infer fields to generate if an inital sample of examples is provided. Returns: dict: dictionary of fields to generate propertiesr_storer)rr) rmodel_json_schemar__dict__r Predictrkeysstrr)r data_schemar)inferred_schema descriptorfields r_define_or_infer_fieldsz.SyntheticDataGenerator._define_or_infer_fields*s   ++==?K$\2J]]"mmA.77AO&:;J'6';';'=?# -jE[^_not_u[v6w6E6E2F"GG?J?J ?s3B<r%c|j}|jr|jjndd}|j|}t |t j f|}t j||}|ttjd|}|jD cgc]>}t j|jD cic]} | t|| c} @c} }Scc} wcc} }w)aGenerate additional examples if needed. Args: additional_samples_needed (int): the difference between the desired number of examples and the current number of examples Returns: List[dspy.Example]: list of synthetic examples Inferred Signature)n)sindex)r4rr _prepare_fieldstyper r7r-r/randomrandint completionsExampler.getattr) rr%r) class_namefieldssignature_class generatorresponse completionrs rr$z4SyntheticDataGenerator._generate_additional_examples=s113 6:6G6G))22ZXXab %%j1zDNN+[*&8s8tTXT`T`Oa8( T rr) r"r=typingrrpydanticrr r7rrrrrrPs2 ! d4>>d H H r