feat(datasets): new collection if not exists, plus cookbook

parea-ai · Jun 4, 2024 · 0195e3b · 0195e3b
1 parent 980c0c4
commit 0195e3b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/parea/cookbook/parea_llm_proxy/dynamic_few_shot_injection.py b/parea/cookbook/parea_llm_proxy/dynamic_few_shot_injection.py
@@ -14,6 +14,8 @@
 
 p = Parea(api_key=os.getenv("PAREA_API_KEY"))
 
+NUM_INTERACTIONS = 5
+
 
 class Person(BaseModel):
     name: str
@@ -26,7 +28,6 @@ class Email(BaseModel):
 
 
 mock_DB: dict[str, Email] = {}
-NUM_INTERACTIONS = 5
 
 
 def call_llm(messages: List[Message]) -> CompletionResponse: