Skip to content

Commit 17fa059

Browse files
andhreljaKernDerKernigeFeuerpfeilJWittmeyer
authored
LLM attribute calculation (#70)
* perf: add LLM_RESPONSE support * fix: overwriting default user prompt with ac.USER_PROMPT * perf: mustache align to requirements * perf: avoid variable overrides by end users * fix: mustache rendering * feat: implement asynchronous processing for LLM_RESPONSE data type * fix: update batch size calculation to use NUM_WORKERS_A2VYBG * ensure at least 1 worker exists * fix: update progress tracking in LLM record processing * fix: improve progress tracking in LLM record processing * perf: unprototype run_ac.py * perf: add llm ac caching * adds valid cache link check which fixes playground error * fix: clarify log message for cached record usage * refactor: streamline attribute calculation and caching logic * refactor: rename functions and variables for clarity and consistency + cache at end * chore: pr review * chore: pr review --------- Co-authored-by: Moritz Feuerpfeil <moritz.feuerpfeil@kern.ai> Co-authored-by: JWittmeyer <jens.wittmeyer@kern.ai>
1 parent dcb6372 commit 17fa059

File tree

5 files changed

+179
-44
lines changed

5 files changed

+179
-44
lines changed

mustache.py

+9
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from typing import Dict, Any
2+
import chevron
3+
4+
5+
def prepare_and_render_mustache(template_text: str, record_dict: Dict[str, Any]) -> str:
6+
return chevron.render(
7+
template=template_text,
8+
data=record_dict["data"],
9+
)

requirements.txt

+13-6
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,7 @@ aiohappyeyeballs==2.4.3
99
# -r requirements/exec-env-requirements.txt
1010
# aiohttp
1111
aiohttp==3.11.7
12-
# via
13-
# -r requirements/exec-env-requirements.txt
14-
# openai
12+
# via -r requirements/exec-env-requirements.txt
1513
aiosignal==1.3.1
1614
# via
1715
# -r requirements/exec-env-requirements.txt
@@ -23,6 +21,7 @@ anyio==4.6.2.post1
2321
# -r requirements/exec-env-requirements.txt
2422
# groq
2523
# httpx
24+
# openai
2625
async-timeout==5.0.1
2726
# via
2827
# -r requirements/exec-env-requirements.txt
@@ -68,6 +67,8 @@ charset-normalizer==3.4.0
6867
# via
6968
# -r requirements/exec-env-requirements.txt
7069
# requests
70+
chevron==0.14.0
71+
# via -r requirements/requirements.in
7172
click==8.1.7
7273
# via
7374
# -r requirements/exec-env-requirements.txt
@@ -99,6 +100,7 @@ distro==1.9.0
99100
# via
100101
# -r requirements/exec-env-requirements.txt
101102
# groq
103+
# openai
102104
docopt==0.6.2
103105
# via
104106
# -r requirements/exec-env-requirements.txt
@@ -144,6 +146,7 @@ httpx==0.27.2
144146
# via
145147
# -r requirements/exec-env-requirements.txt
146148
# groq
149+
# openai
147150
idna==3.10
148151
# via
149152
# -r requirements/exec-env-requirements.txt
@@ -167,6 +170,8 @@ jinja2==3.1.4
167170
# via
168171
# -r requirements/exec-env-requirements.txt
169172
# spacy
173+
jiter==0.8.2
174+
# via openai
170175
joblib==1.4.2
171176
# via
172177
# -r requirements/exec-env-requirements.txt
@@ -254,8 +259,8 @@ numpy==1.23.4
254259
# spacy
255260
# textacy
256261
# thinc
257-
openai==0.27.7
258-
# via -r requirements/exec-env-requirements.txt
262+
openai==1.59.7
263+
# via -r requirements/requirements.in
259264
packaging==24.2
260265
# via
261266
# -r requirements/exec-env-requirements.txt
@@ -297,6 +302,7 @@ pydantic==2.7.4
297302
# -r requirements/requirements.in
298303
# confection
299304
# groq
305+
# openai
300306
# spacy
301307
# thinc
302308
# weasel
@@ -343,7 +349,6 @@ requests==2.31.0
343349
# -r requirements/exec-env-requirements.txt
344350
# google-search-results
345351
# lexmo
346-
# openai
347352
# spacy
348353
# sumy
349354
# textacy
@@ -380,6 +385,7 @@ sniffio==1.3.1
380385
# anyio
381386
# groq
382387
# httpx
388+
# openai
383389
soupsieve==2.6
384390
# via
385391
# -r requirements/exec-env-requirements.txt
@@ -466,6 +472,7 @@ typing-extensions==4.12.2
466472
# cloudpathlib
467473
# groq
468474
# multidict
475+
# openai
469476
# pydantic
470477
# pydantic-core
471478
# sqlalchemy

requirements/exec-env-requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ numpy==1.23.4
171171
# spacy
172172
# textacy
173173
# thinc
174-
openai==0.27.7
174+
# openai==0.27.7
175175
# via -r exec-env-requirements.in
176176
packaging==24.2
177177
# via

requirements/requirements.in

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
11
-r exec-env-requirements.txt
22
pydantic==2.7.4
3-
spacy[ja]==3.7.5
3+
spacy[ja]==3.7.5
4+
chevron==0.14.0
5+
openai==1.59.7

0 commit comments

Comments
 (0)