add a test

sotopia-lab · Nov 1, 2023 · 66f58df · 66f58df
1 parent 9686111
commit 66f58df
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/llm_generate/test_redis2.py b/llm_generate/test_redis2.py
@@ -0,0 +1,18 @@
+from redis_om import JsonModel, get_redis_connection
+
+class Person(JsonModel):
+    name: str
+    age: int
+
+# Create an instance of your model
+person = Person(name="John", age=30)
+
+# Save to Redis
+person.save()
+
+# Retrieve from Redis
+retrieved_person = Person.load(person.id)
+
+# Print the retrieved data
+print(retrieved_person.name)  # Output: John
+print(retrieved_person.age)   # Output: 30