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

parea-ai · Jun 4, 2024 · 980c0c4 · 980c0c4
1 parent 43c7267
commit 980c0c4
Showing 1 changed file with 1 addition and 2 deletions.
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,8 +14,6 @@
 
 p = Parea(api_key=os.getenv("PAREA_API_KEY"))
 
-NUM_INTERACTIONS = 5
-
 
 class Person(BaseModel):
     name: str
@@ -28,6 +26,7 @@ class Email(BaseModel):
 
 
 mock_DB: dict[str, Email] = {}
+NUM_INTERACTIONS = 5
 
 
 def call_llm(messages: List[Message]) -> CompletionResponse: