add patch function

agentops/session.py

@@ -1,4 +1,5 @@
 import copy
+import functools
 import json
 import threading
 import time
@@ -241,3 +242,11 @@ def create_agent(self, name, agent_id):
         )
 
         return agent_id
+
+    def patch(self, func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            kwargs["session"] = self
+            return func(*args, **kwargs)
+
+        return wrapper

tests/core_manual_tests/api_server/server.py

@@ -22,7 +22,6 @@ def completion():
         model="gpt-3.5-turbo",
         messages=messages,
         temperature=0.5,
-        session=session,  # <-- add the agentops session to the create function
     )
 
     session.record(

tests/core_manual_tests/multi_session_llm.py

@@ -15,20 +15,18 @@
 
 messages = [{"role": "user", "content": "Hello"}]
 
-response = openai.chat.completions.create(
+# option 1: use session.patch
+response = session_1.patch(openai.chat.completions.create)(
     model="gpt-3.5-turbo",
     messages=messages,
     temperature=0.5,
-    session=session_1,  # <-- add the agentops session_id to the create function
 )
 
 session_1.record(ActionEvent(action_type="test event"))
 
-response = openai.chat.completions.create(
-    model="gpt-3.5-turbo",
-    messages=messages,
-    temperature=0.5,
-    session=session_2,  # <-- add the agentops session_id to the create function
+# option 2: add session as a keyword argument
+response2 = openai.chat.completions.create(
+    model="gpt-3.5-turbo", messages=messages, temperature=0.5, session=session_2
 )
 
 session_1.end_session(end_state="Success")