organization_index added along README section

Author: st1vms

README.md

@@ -11,6 +11,7 @@
   - [Faster Loading](#faster-loading-avoiding-selenium)
   - [Proxies](#proxies)
   - [Changing model version](#changing-claude-model)
+  - [Changing Organization](#changing-organization)
 - [Troubleshooting](#troubleshooting)
 - [Donating](#donating)
 
@@ -272,6 +273,24 @@ client = ClaudeAPIClient(session, model_name="claude-2.0")
 
 You can retrieve the `model_name` strings from the [official API docs](https://docs.anthropic.com/claude/docs/models-overview#model-comparison)
 
+__________
+
+### Changing Organization
+
+As reported in issue [#23](https://github.com/st1vms/unofficial-claude-api/issues/23)
+if you're encountering 403 errors when using Selenium to auto retrieve a `SessionData` class and your account has multiple organizations,
+you may want to override the default organization retrieved.
+
+By default `get_session_data` retrieves the last organization from the result array found [here](https://claude.ai/api/organizations).
+You can override the index to fetch by using parameter `organization_index`:
+
+```py
+from claude_api.session import get_session_data
+
+# Defaults to -1 (last entry)
+session = get_session_data(organization_index=-1)
+```
+
 ## TROUBLESHOOTING
 
 Some common errors that may arise during the usage of this API:

claude_api/session.py

@@ -38,13 +38,16 @@ class SessionData:
     """
 
 
-def get_session_data(profile: str = "", quiet: bool = False) -> SessionData | None:
+def get_session_data(profile: str = "", quiet: bool = False, organization_index:int=-1) -> SessionData | None:
     """
     Retrieves Claude session data
 
     This function requires a profile with Claude login and geckodriver installed!
 
     The default Firefox profile will be used, if the profile argument was not overwrited.
+
+    Parameter `organization_index` is by default -1
+    (last entry from https://claude.ai/api/organizations)
     """
 
     json_tab_id = 'a[id="rawdata-tab"]'
@@ -81,8 +84,8 @@ def get_session_data(profile: str = "", quiet: bool = False) -> SessionData | No
         if pre.text:
             j = json_loads(pre.text)
             try:
-                if j and len(j) >= 1 and "uuid" in j[0]:
-                    org_id = j[0]["uuid"]
+                if j and len(j) > organization_index and "uuid" in j[organization_index]:
+                    org_id = j[organization_index]["uuid"]
             except KeyError:
                 print(
                     f"\nUnable to retrieve organization_id from profile: {profile}\n"