Skip to content

Commit

Permalink
are u kidding
Browse files Browse the repository at this point in the history
  • Loading branch information
MadcowD committed Aug 22, 2024
1 parent 5f9741b commit 396eb58
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 1 deletion.
7 changes: 6 additions & 1 deletion src/ell/util/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,4 +139,9 @@ def _call(
for _, choice_deltas in sorted(choices_progress.items(), key= lambda x: x[0],)
]

return tracked_results[0] if n_choices == 1 else tracked_results, metadata
return tracked_results[0] if n_choices == 1 else tracked_results, metadata

# Todo: for implementation of lmp.text, & lmp.multimodal, we need to do the following:
# 1. implement a _call fn that creates a generic multimodal output input and output type irrespective of the client for a model
# 2. modify lm to use _call then process it as a ttacke dlstr so tis very clean to use
# 3. implement a multimodal decorator that can handle the multimodal input and output types & does not restructure shit.
2 changes: 2 additions & 0 deletions src/ell/util/closure.py
Original file line number Diff line number Diff line change
Expand Up @@ -505,3 +505,5 @@ def globalvars(func, recurse=True, builtin=False):
#NOTE: if name not in __globals__, then we skip it...
return dict((name,globs[name]) for name in func if name in globs)


# XXX: This is a mess. COuld probably be about 100 lines of code max.

0 comments on commit 396eb58

Please sign in to comment.