CultriX commited on
Commit
e153a12
·
verified ·
1 Parent(s): d11af12

Update yall.py

Browse files
Files changed (1) hide show
  1. yall.py +42 -0
yall.py CHANGED
@@ -53,6 +53,48 @@ def update_gist(content, gist_id, access_token):
53
 
54
 
55
  @st.cache_data
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
56
  def create_yall():
57
  # Get token
58
  GITHUB_API_TOKEN = os.environ.get("github")
 
53
 
54
 
55
  @st.cache_data
56
+
57
+ def get_gist_info(gist):
58
+ """
59
+ Extract information from a GitHub Gist and return a GistInfo object.
60
+
61
+ Args:
62
+ gist (dict): A dictionary representing a GitHub Gist.
63
+
64
+ Returns:
65
+ GistInfo: An object containing information about the gist.
66
+ """
67
+ # Extract information from the gist
68
+ gist_id = gist['id']
69
+ filename = gist['files'][0]['filename']
70
+ url = gist['html_url']
71
+ model_name = 'Unknown' # Replace with actual logic
72
+ model_id = 'Unknown' # Replace with actual logic
73
+ model = 'Unknown' # Replace with actual logic
74
+ agieval = 0.0 # Replace with actual logic
75
+ gpt4all = 0.0 # Replace with actual logic
76
+ truthfulqa = 0.0 # Replace with actual logic
77
+ bigbench = 0.0 # Replace with actual logic
78
+ average = 0.0 # Replace with actual logic
79
+
80
+ # Create a GistInfo object
81
+ gist_info = GistInfo(
82
+ gist_id=gist_id,
83
+ filename=filename,
84
+ url=url,
85
+ model_name=model_name,
86
+ model_id=model_id,
87
+ model=model,
88
+ agieval=agieval,
89
+ gpt4all=gpt4all,
90
+ truthfulqa=truthfulqa,
91
+ bigbench=bigbench,
92
+ average=average,
93
+ )
94
+
95
+ return gist_info
96
+
97
+
98
  def create_yall():
99
  # Get token
100
  GITHUB_API_TOKEN = os.environ.get("github")