Skip to content

Commit

Permalink
codestyle
Browse files Browse the repository at this point in the history
  • Loading branch information
@jalexanderII
jalexanderII committed Oct 18, 2023
1 parent 8666b72 commit 2fa1065
Showing 1 changed file with 68 additions and 68 deletions.
136 changes: 68 additions & 68 deletions parea/cookbook/tracing_with_Parea_sdk.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -117,40 +117,40 @@
"\n",
"\n",
"def argument_generator(query: str, additional_description: str = \"\") -> str:\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-tbFUZ5rRaXshj8o5Opfyr\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" },\n",
" )\n",
" ).content\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-tbFUZ5rRaXshj8o5Opfyr\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" },\n",
" )\n",
" ).content\n",
"\n",
"\n",
"def critic(argument: str) -> str:\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-iAuVLFHy6VypfGZxwAWW0\",\n",
" llm_inputs={\"argument\": argument},\n",
" )\n",
" ).content\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-iAuVLFHy6VypfGZxwAWW0\",\n",
" llm_inputs={\"argument\": argument},\n",
" )\n",
" ).content\n",
"\n",
"\n",
"def refiner(query: str, additional_description: str, current_arg: str, criticism: str) -> str:\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-rEjM4X10rJomOD8Rj9gzJ\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" \"argument\": current_arg,\n",
" \"criticism\": criticism,\n",
" },\n",
" )\n",
" ).content\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-rEjM4X10rJomOD8Rj9gzJ\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" \"argument\": current_arg,\n",
" \"criticism\": criticism,\n",
" },\n",
" )\n",
" ).content\n",
"\n",
"\n",
"# Non deployed version\n",
Expand Down Expand Up @@ -209,9 +209,9 @@
"\n",
"@trace\n",
"def argument_chain(query: str, additional_description: str = \"\") -> str:\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner(query, additional_description, argument, criticism)"
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner(query, additional_description, argument, criticism)"
]
},
{
Expand Down Expand Up @@ -251,8 +251,8 @@
],
"source": [
"result1 = argument_chain(\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
")\n",
"print(result1)"
]
Expand Down Expand Up @@ -302,10 +302,10 @@
"\n",
"@trace\n",
"def argument_chain2(query: str, additional_description: str = \"\") -> tuple[str, str]:\n",
" trace_id = get_current_trace_id() # get parent's trace_id\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner(query, additional_description, argument, criticism), trace_id"
" trace_id = get_current_trace_id() # get parent's trace_id\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner(query, additional_description, argument, criticism), trace_id"
]
},
{
Expand All @@ -331,8 +331,8 @@
],
"source": [
"result, trace_id = argument_chain2(\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
")\n",
"print(trace_id)"
]
Expand Down Expand Up @@ -424,30 +424,30 @@
"\n",
"# let's return the full CompletionResponse to see what other information is returned\n",
"def refiner2(query: str, additional_description: str, current_arg: str, criticism: str) -> CompletionResponse:\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-rEjM4X10rJomOD8Rj9gzJ\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" \"argument\": current_arg,\n",
" \"criticism\": criticism,\n",
" },\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-rEjM4X10rJomOD8Rj9gzJ\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" \"argument\": current_arg,\n",
" \"criticism\": criticism,\n",
" },\n",
" )\n",
" )\n",
" )\n",
"\n",
"\n",
"@trace\n",
"def argument_chain3(query: str, additional_description: str = \"\") -> CompletionResponse:\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner2(query, additional_description, argument, criticism)\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner2(query, additional_description, argument, criticism)\n",
"\n",
"\n",
"result2 = argument_chain3(\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
")\n",
"print(json.dumps(asdict(result2), indent=2))"
],
Expand Down Expand Up @@ -498,14 +498,14 @@
"\n",
"# you can also add metadata and tags via the decorator\n",
"@trace(\n",
" tags=[\"cookbook-example-deployed\", \"feedback_tracked-deployed\"],\n",
" metadata={\"source\": \"python-sdk\", \"deployed\": \"True\"},\n",
" tags=[\"cookbook-example-deployed\", \"feedback_tracked-deployed\"],\n",
" metadata={\"source\": \"python-sdk\", \"deployed\": \"True\"},\n",
")\n",
"def argument_chain_tags_metadata(query: str, additional_description: str = \"\") -> Tuple[CompletionResponse, str]:\n",
" trace_id = get_current_trace_id() # get parent's trace_id\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner2(query, additional_description, argument, criticism), trace_id"
" trace_id = get_current_trace_id() # get parent's trace_id\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner2(query, additional_description, argument, criticism), trace_id"
]
},
{
Expand All @@ -515,17 +515,17 @@
"from attrs import asdict\n",
"\n",
"result2, trace_id = argument_chain_tags_metadata(\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
")\n",
"print(json.dumps(asdict(result2), indent=2))\n",
"\n",
"p.record_feedback(\n",
" FeedbackRequest(\n",
" trace_id=trace_id,\n",
" score=0.7, # 0.0 (bad) to 1.0 (good)\n",
" target=\"Coffee is wonderful. End of story.\",\n",
" )\n",
" FeedbackRequest(\n",
" trace_id=trace_id,\n",
" score=0.7, # 0.0 (bad) to 1.0 (good)\n",
" target=\"Coffee is wonderful. End of story.\",\n",
" )\n",
")"
],
"metadata": {
Expand Down

0 comments on commit 2fa1065

Please sign in to comment.