o RTh@sdZddlmZddlmZddlmZmZmZddl m Z m Z ddl m Z mZmZddlmZddlmZdd lmZeed d d ed d ddeddddeddZeed d ddeddddeddZdS)z>Magic to generate code cells for notebooks using OpenAI's API.) get_ipython) cell_magic)argumentmagic_argumentsparse_argstring) PastAssists build_context) GenaiMarkdownStagecan_handle_display_updates)generate_next_from_history) PromptStore) trim_messages_to_fit_token_limitz--fresh store_true)actionz --verbosez.Show additional information in the cell output)rhelpz--modelzgpt-3.5-turbo-0301zthe model to use)defaultrc Cst}tt|}ttjd}|t|j || }|j }t }g}|j sBt|j dd} |j } t|j| | d} t| j|d}|jrTtd|td|td||t|||d tj|_d S) u Get help based on the current notebook session. Outputs markdown. `genai`'s `assist` magic asks ChatGPT to respond based on: * Your previous code `In[*]` * The output of your previous code `Out[*]` * Past exceptions * Past `genai` suggestions Usage: # Cell 1 ```python import pandas as pd # Berkeley Restaurant Inspections Data df = pd.read_json("https://data.cityofberkeley.info/resource/iuea-7eac.json") df ``` # Cell 2 ```python %%assist Let's plot the number of restaurants inspected per day. ``` Will output markdown like this: ``````markdown Sure! First we'll need to make sure the `inspection_date` column in your DataFrame is in a datetime format. You can do this with the `pd.to_datetime()` method. Then we can group the data by day and count the number of unique restaurants inspected each day. This can be achieved using the `groupby()` and `nunique()` methods, respectively. Finally, we can plot the results using `plot()` with the `bar` style. Here's some code to get you started: ```python import matplotlib.pyplot as plt # Convert inspection_date to datetime format df['inspection_date'] = pd.to_datetime(df['inspection_date']) # Group by day and count number of unique restaurants daily_count = df.groupby(df['inspection_date'].dt.date)['doing_business_as'].nunique() # Plot the results as a bar chart daily_count.plot(kind='bar') plt.show() ``` Give this a try and let me know if you have any questions! `````` If you've run previous cells in the notebook, the generated code will be based on the history of the current notebook session. That even means that, for example, running `df.columns` in a previous cell will cause generated code to use column names in the future! Caveats: - Only the last 5 cell executions are provided as context. - The generated code is not guaranteed to be correct, idiomatic, efficient, readable, or useful. - The generated code is not guaranteed to be syntactically correct or even something to write home about. There are several options you can use to control the behavior of the magic. --fresh: Ignore the history of the current notebook session and generate code from scratch. --verbose: Show additional information in the cell output. Example: ```python %%assist --fresh --verbose # how can I query for pokemon via the PokéAPI? ``` )stage)startstop)modelmagic arguments:z submission:z messages:)streamN)rrassistr r STARTINGdisplayraddexecution_countstriprr freshmaxrhistory_managerrmessagesverboseprintconsumer FINISHEDr) linecellipargsgm cell_textrrr$rrcontextr0O/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/genai/magics.pyrs,^     rz--modifyz"the magic to modify the prompt forcCstt|}|jdvrtd|j|jr4tj}|jdks"|jdkr%tj}td|td|td||jdkr>|t_d S|jdksH|jdkrM|t_d Sd S) aReplace the default prompt for genai By default it modifies the prompt for the `assist` magic, but you can also use it to modify the prompt for exception handling. Example: ```python %%prompt You are a pirate. You are in a tavern. You are cursed with the knowledge of programming. Deep learning brought you here and only deep teaching can get you out. The only way to lift the curse is to help a programmer with their code. Respond in markdown. Talk like a pirate. ``` Example changing exceptions: ```python %%prompt --modify exceptions You are a valley girl. You are on a walk. You are one of the best programmers in the world. People come to you looking for help. Respond in markdown. Talk like a valley girl. )r exceptions exceptionzUnknown prompt to modify: r2r3rz old prompt:z new prompt:rN) rpromptmodify ValueErrorr%r assist_promptexception_promptr&)r)r*r, old_promptr0r0r1r4s $       r4N)__doc__IPythonrIPython.core.magicrIPython.core.magic_argumentsrrr genai.contextrr genai.displayr r r genai.generater genai.promptsr genai.tokensrrr4r0r0r0r1sH      x