[SYNPY-1544] Fixes docstring (#1155)

* fixes docstring

* protocol docstring

* fix imports

Author: BWMac

synapseclient/models/agent.py

@@ -647,7 +647,7 @@ async def get_async(self, *, synapse_client: Optional[Synapse] = None) -> "Agent
 
                 import asyncio
                 from synapseclient import Synapse
-                from synapseclient.models.agent import Agent
+                from synapseclient.models import Agent, AgentSessionAccessLevel
 
                 syn = Synapse()
                 syn.login()
@@ -838,6 +838,9 @@ async def prompt_async(
 
                 async def main():
                     my_agent = Agent()
+                    await my_agent.start_session_async(
+                        access_level=AgentSessionAccessLevel.WRITE_YOUR_PRIVATE_DATA
+                    )
                     await my_agent.prompt_async(
                         prompt="Add the annotation 'test' to the file 'syn123456789'",
                         enable_trace=True,

synapseclient/models/protocols/agent_protocol.py

@@ -332,11 +332,15 @@ def prompt(
             The baseline Synpase Agent can be used to add annotations to files.
 
             from synapseclient import Synapse
+            from synapseclient.models import Agent, AgentSessionAccessLevel
 
             syn = Synapse()
             syn.login()
 
             my_agent = Agent()
+            my_agent.start_session(
+                access_level=AgentSessionAccessLevel.WRITE_YOUR_PRIVATE_DATA
+            )
             my_agent.prompt(
                 prompt="Add the annotation 'test' to the file 'syn123456789'",
                 enable_trace=True,