search

sotopia-lab / sotopia

Sotopia: an Open-ended Social Learning Environment (ICLR 2024 spotlight)
https://docs.sotopia.world
MIT License
164 stars 20 forks source link

Add 1.2 Download and browse Sotopia(-pi) data #71

Closed ProKil closed 6 months ago

XuhuiZhou commented 6 months ago

Adding some use cases for browsing Sotopia data:

  1. episode rendering (we have some functions for that now; just need better documentation)
  2. get a list of scenarioes by searching specific topics
  3. get a list of characters by searching specific personality/occupations
  4. get a list of current existing tags
ProKil commented 6 months ago
  1. is already in the colab demo,
  2. what are the keys and filtering algorithm?
  3. this is not hard, I could do this
  4. the latency of this could be large
ProKil commented 6 months ago

I have finished 3 and 4 in https://colab.research.google.com/drive/1Gi2U3cA0KG1nekw1A0N0DZpx6gKrj8Nc?usp=sharing.

I am not sure if 4 is interesting for most users?