update python sdk to have optimize and ape schedule

README.md

@@ -23,7 +23,7 @@ You can find our full documentation [here](https://weavel.ai/docs/python-sdk).
 
 ## How to use
 
-### Basic Usage
+### Option 1: Using OpenAI wrapper
 
 ```python
 from weavel import WeavelOpenAI as OpenAI
@@ -42,7 +42,40 @@ response = openai.chat.completions.create(
 
 ```
 
-### Advanced Usage
+### Option 2: Logging inputs/outputs of LLM calls
+
+```python
+from weavel import Weavel
+from openai import OpenAI
+from pydantic import BaseModel
+
+openai = OpenAI()
+# initialize Weavel
+weavel = Weavel()
+
+class Answer(BaseModel):
+    reasoning: str
+    answer: str
+
+question = "What is x if x + 2 = 4?"
+response = openai.beta.chat.completions.parse(
+    model="gpt-4o-2024-08-06",
+    messages=[
+        {"role": "system", "content": "You are a math teacher."},
+        {"role": "user", "content": question}
+    ],
+    response_format=Answer
+).choices[0].message.parsed
+
+# log the generation
+weavel.generation(
+    name="solve-math", # optional
+    inputs={"question": question},
+    outputs=response.model_dump()
+)
+```
+
+### Option 3 (Advanced Usage): OTEL-compatible trace logging
 
 ```python
 from weavel import Weavel

setup.py

@@ -10,7 +10,7 @@
 
 setup(
     name="weavel",
-    version="1.7.1",
+    version="1.9.4",
     packages=find_namespace_packages(),
     entry_points={},
     description="Weavel, Prompt Optimization and Evaluation for LLM Applications",
@@ -32,8 +32,6 @@
         "termcolor",
         "watchdog",
         "readerwriterlock",
-        "pendulum",
-        "httpx[http2]",
         "nest_asyncio",
     ],
     python_requires=">=3.8.10",
@@ -47,6 +45,6 @@
         "dataset curation",
         "prompt engineering",
         "prompt optimization",
-        "AI Prompt Engineer"
+        "AI Prompt Engineer",
     ],
 )

weavel/__init__.py

@@ -4,4 +4,4 @@
 
 from .utils import *
 
-__version___ = "1.7.1"
+__version___ = "1.9.4"

weavel/_body.py

@@ -132,7 +132,10 @@ class CaptureRecordBody(BaseRecordBody):
 
 
 class CaptureObservationBody(BaseObservationBody):
-    name: str
+    name: Optional[str] = Field(
+        default=None,
+        description="The name of the observation. Optional.",
+    )
 
 
 class CaptureMessageBody(CaptureRecordBody):
@@ -262,6 +265,22 @@ class CaptureGenerationBody(CaptureObservationBody):
         default=None,
         description="The outputs of the generation. Optional.",
     )
+    messages: Optional[List[Dict[str, str]]] = Field(
+        default=None,
+        description="The messages of the generation. Optional.",
+    )
+    model: Optional[str] = Field(
+        default=None,
+        description="The model of the generation. Optional.",
+    )
+    latency: Optional[float] = Field(
+        default=None,
+        description="The latency of the generation. Optional.",
+    )
+    cost: Optional[float] = Field(
+        default=None,
+        description="The cost of the generation. Optional.",
+    )
     prompt_name: Optional[str] = Field(
         default=None,
         description="The name of the prompt. Optional.",