Add entries to in-memory store on Put() (#7085)

python/ray/includes/unique_ids.pxd

@@ -152,6 +152,8 @@ cdef extern from "ray/common/id.h" namespace "ray" nogil:
 
         CObjectID WithPlasmaTransportType()
 
+        CObjectID WithDirectTransportType()
+
         int64_t ObjectIndex() const
 
         CTaskID TaskId() const

python/ray/includes/unique_ids.pxi

@@ -185,7 +185,7 @@ cdef class ObjectID(BaseID):
 
     @classmethod
     def from_random(cls):
-        return cls(CObjectID.FromRandom().Binary())
+        return cls(CObjectID.FromRandom().WithDirectTransportType().Binary())
 
     def __await__(self):
         # Delayed import because this can only be imported in py3.