diff --git a/parea/cookbook/parea_llm_proxy/dynamic_few_shot_injection.py b/parea/cookbook/parea_llm_proxy/dynamic_few_shot_injection.py index 7f98a1d6..a5f8ae96 100644 --- 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: